EF Core 6新初始化上下文无法加载子实体问题求助

我正在开发一个解析JSON树并保存到数据库的Web API，使用EF Core 6.0.4。遇到以下问题：

• 保存树后，同一DbContext上下文内可以正常加载子实体
• 调用其他端点、使用新初始化的上下文加载时，子实体无法加载

尝试过的方法及问题：

• 使用_context.Entity(returnValue).Collection(x => x.Children)，但Lambda表达式中的x仅包含ICollection扩展方法，无法访问模型的Children字段
• 曾误以为.Include不属于EF Core，未正确使用它

控制器代码 #

public class CategoryTreeManagerController : ControllerBase
{
    private readonly CategoryTreeManagerService _service;
    public CategoryTreeManagerController(CategoryTreeManagerService service)
    {
        _service = service;
    }

    [HttpGet]
    public IEnumerable<CategoryTreeNode> GetTree()
    {
        return _service.GetTree(); // 仅返回根节点，无任何子节点
    }

    [HttpPost]
    public IEnumerable<CategoryTreeNode> SaveTree(CategoryTreeNode[] nodes)
    {
        _service.SaveTree(nodes[0]);

        return _service.GetTree(); // 此处能正确返回完整树形结构
    }
}

服务代码 #

public class CategoryTreeManagerService
{
    private readonly CategoryTreeManagerApiDbContext _context;
    public CategoryTreeManagerService(CategoryTreeManagerApiDbContext context)
    {
        _context = context;
    }

    public IEnumerable<CategoryTreeNode> GetTree()
    {
        CategoryTreeNode[] returnValue = _context.CategoryTreeNodes
            .Where(x => x.Parent == null) // 仅返回根节点
            .ToArray(); // 方便前端处理

        return returnValue;
    }

    public void SaveTree(CategoryTreeNode node)
    {
        if (_context.CategoryTreeNodes.Any(x => x.Id == node.Id))
        {
            _context.Update(node);
        }
        else
        {
            _context.CategoryTreeNodes.Add(node);
        }

        _context.SaveChanges();
    }
}

DbContext代码 #

public class CategoryTreeManagerApiDbContext : DbContext
{
    public CategoryTreeManagerApiDbContext() : base ()
    {
        Database.EnsureCreated();
    }
    
    public CategoryTreeManagerApiDbContext(DbContextOptions<CategoryTreeManagerApiDbContext> options) : base(options)
    {
        Database.EnsureCreated();
    }

    public DbSet<CategoryTreeNode> CategoryTreeNodes { get; set; }
    public DbSet<TreeNodeDetail> TreeNodeDetails { get; set; }

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        base.OnConfiguring(optionsBuilder);
    }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder.Entity<CategoryTreeNode>(entity =>
        {
            entity.HasKey(x => x.Id);
            entity.HasOne(x => x.Parent)
                .WithMany(x => x.Children)
                .IsRequired(false)
                .OnDelete(DeleteBehavior.Restrict);
        });
    }
}

模型类代码 #

public class CategoryTreeNode
{
    public int Id { get; set; }
    public string Label { get; set; }
    public string Story { get; set; }
    public string Icon { get; set; }
    public ICollection<TreeNodeDetail> Details { get; set; }
    public ICollection<CategoryTreeNode> Children { get; set; }

    [JsonIgnore]
    public CategoryTreeNode? Parent { get; set; }
}

public class TreeNodeDetail
{
    [JsonPropertyName("detailId")]
    public string Id { get; set; }
    [JsonPropertyName("detailTitle")]
    public string Title { get; set; }
    [JsonPropertyName("detailValue")]
    public string Value { get; set; }

    [JsonIgnore]
    [ForeignKey("CategoryTreeNode")]
    public int CategoryTreeNodeId { get; set; }
}

1. 用Include显式加载关联数据 #

EF Core完全支持.Include，之前的误解是错误的。对于树形结构，可通过Include加载直接子节点，结合ThenInclude加载嵌套关联：

public IEnumerable<CategoryTreeNode> GetTree()
{
    var returnValue = _context.CategoryTreeNodes
        .Where(x => x.Parent == null)
        .Include(x => x.Details) // 加载根节点的Details
        .Include(x => x.Children) // 加载根节点的直接子节点
            .ThenInclude(child => child.Details) // 加载子节点的Details
            .ThenInclude(child => child.Children) // 如需更深层级可继续嵌套（无限层级不推荐）
        .ToArray();

    return returnValue;
}

如果是无限层级的树，建议结合递归显式加载，或使用延迟加载。

2. 启用延迟加载（可选） #

若希望访问导航属性时自动加载关联数据，可启用延迟加载：

1. 安装Microsoft.EntityFrameworkCore.Proxies包
2. 在DbContext的OnConfiguring中配置：

protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
{
    optionsBuilder.UseLazyLoadingProxies();
    // 其他数据库连接配置，如UseSqlServer("连接字符串")
}

3. 将模型中的导航属性改为virtual：

public virtual ICollection<TreeNodeDetail> Details { get; set; }
public virtual ICollection<CategoryTreeNode> Children { get; set; }
public virtual CategoryTreeNode? Parent { get; set; }

注意：延迟加载可能引发N+1查询问题，需根据业务场景评估使用。

3. 修复显式加载的正确用法 #

若要手动显式加载子节点，需遍历每个被上下文跟踪的实体：

public IEnumerable<CategoryTreeNode> GetTree()
{
    var returnValue = _context.CategoryTreeNodes
        .Where(x => x.Parent == null)
        .ToArray();

    foreach(var node in returnValue)
    {
        _context.Entry(node).Collection(x => x.Children).Load(); // 显式加载当前节点的子节点
        // 如需加载更深层级，可在此处递归调用加载逻辑
    }

    return returnValue;
}

4. 优化SaveTree方法的潜在问题 #

直接使用_context.Update(node)会将整个实体图标记为修改，对于树形结构，可能导致未预期的数据覆盖。建议先查询数据库中的现有实体，再更新属性：

public void SaveTree(CategoryTreeNode node)
{
    var existingNode = _context.CategoryTreeNodes
        .Include(x => x.Children)
        .Include(x => x.Details)
        .FirstOrDefault(x => x.Id == node.Id);

    if (existingNode != null)
    {
        // 手动更新需要修改的属性
        existingNode.Label = node.Label;
        existingNode.Story = node.Story;
        existingNode.Icon = node.Icon;
        // 处理Children和Details的增删改逻辑
    }
    else
    {
        _context.CategoryTreeNodes.Add(node);
    }

    _context.SaveChanges();
}

内容的提问来源于stack exchange，提问作者lenny