csharp 更新相关子实体

Posted 2021-05-11

public override async Task<UserCategoryDto> Update(UserCategoryDto input)
        {
            CheckUpdatePermission();
            var existingEntity = await Repository.GetAll().Include(x => x.Items).FirstOrDefaultAsync(x => x.Id == input.Id);
            if (existingEntity != null)
            {
                // Update parent
                Repository.GetDbContext().Entry(existingEntity).CurrentValues.SetValues(ObjectMapper.Map<UserCategory>(input));

                // Delete children
                foreach (var existingChild in existingEntity.Items.ToList())
                {
                    if (input.Items.All(c => c.Id != existingChild.Id))
                        existingEntity.Items.Remove(existingChild);
                }

                // Update and Insert children
                foreach (var itemDto in input.Items)
                {
                    var existingChild = existingEntity.Items.SingleOrDefault(c => itemDto.Id > 0 && c.Id == itemDto.Id);

                    if (existingChild != null)
                    {
                        // Update child
                        var item = ObjectMapper.Map<UserCategoryItem>(itemDto);
                        Repository.GetDbContext().Entry(existingChild).CurrentValues.SetValues(item);
                    }
                    else
                    {
                        // Insert child
                        var item = ObjectMapper.Map<UserCategoryItem>(itemDto);
                        existingEntity.Items.Add(item);
                    }
                }

                await CurrentUnitOfWork.SaveChangesAsync();
            }

            return MapToEntityDto(existingEntity);
        }