EFCore.BulkExtensions 中使用 TPH 继承模型的批量操作指南

理解问题背景

在使用 EFCore.BulkExtensions 进行批量操作时，当遇到 Table-Per-Hierarchy (TPH) 继承模型时，开发者可能会遇到一些特殊问题。TPH 是 Entity Framework Core 中处理继承关系的一种方式，它将整个类层次结构映射到单个数据库表中，并使用鉴别器列来区分不同类型。

常见问题分析

1. 鉴别器列缺失问题

当直接对基类集合进行批量操作时，可能会遇到"鉴别器列为空"的错误。这是因为批量操作需要明确知道每个实体的具体类型，以便正确设置鉴别器值。

2. 属性不存在异常

在对混合类型集合进行操作时，可能会遇到"属性不存在"的异常。这是因为某些子类特有的属性在其他子类中并不存在，批量操作框架需要正确处理这种情况。

解决方案

分类型处理策略

正确的做法是将不同类型的实体分开处理：

public async Task AddRange(IEnumerable<Entity> entities, CancellationToken cancellationToken)
{
    // 按具体类型分组
    var organizations = entities.OfType<Entity.Organization>();
    var globals = entities.OfType<Entity.Global>();

    // 分别处理每种类型
    if (organizations.Any())
    {
        await _dbContext.BulkInsertAsync(organizations, cancellationToken: cancellationToken);
    }

    if (globals.Any())
    {
        await _dbContext.BulkInsertAsync(globals, cancellationToken: cancellationToken);
    }
}

批量插入或更新操作

对于需要判断存在性的场景，可以使用 BulkInsertOrUpdateAsync，但仍需分类型处理：

public async Task BulkInsertIfNotExist(IEnumerable<Entity> entities, CancellationToken cancellationToken)
{
    var organizations = entities.OfType<Entity.Organization>();
    var globals = entities.OfType<Entity.Global>();

    if (organizations.Any())
    {
        await _dbContext.BulkInsertOrUpdateAsync(
            organizations,
            option =>
            {
                option.PropertiesToIncludeOnUpdate = new List<string>();
                option.SetOutputIdentity = true;
            },
            cancellationToken: cancellationToken);
    }

    if (globals.Any())
    {
        await _dbContext.BulkInsertOrUpdateAsync(
            globals,
            option =>
            {
                option.PropertiesToIncludeOnUpdate = new List<string>();
                option.SetOutputIdentity = true;
            },
            cancellationToken: cancellationToken);
    }
}

最佳实践建议

1. 避免混合使用传统EF和批量操作：不要在同一批数据上同时使用 AddRange 和 BulkInsert，这会导致重复操作。

2. 明确鉴别器配置：确保在DbContext中正确配置了鉴别器列：

builder.HasDiscriminator<string>("Discriminator")
       .HasValue<Entity.Global>("Global")
       .HasValue<Entity.Organization>("Organization");

3. 考虑性能与事务：对于大量数据操作，考虑将多个批量操作包裹在事务中，以保证数据一致性。

4. 输出标识处理：当需要获取插入后的ID时，使用 SetOutputIdentity = true 选项。

总结

在使用 EFCore.BulkExtensions 处理 TPH 继承模型时，关键在于理解批量操作需要明确的类型信息。通过将不同类型的实体分开处理，可以避免鉴别器问题和属性不存在异常。同时，遵循不混合使用不同操作模式的原则，可以确保数据操作的效率和正确性。