如何从 Entity Framework Core 获取实体保存顺序

Question

When Entity Framework Core bulk saves all your changes, it uses an internal priority computed by all your foreign keys to make sure everything is saved in the right order to prevent foreign key errors.

How can I get this priority list from Entity Framework?

I'm constructing a synchronizer that moves a lot of entities between servers. I need to send the entities in the right order because I cannot save everything in one go due to memory usage. I want to bulk save entities from the stream, but at the moment I got issues with foreign keys. It would be nice if I could use the work done by Entity Framework to get the correct save order.

Answer 1

I've done some digging inside EF Core, and found that the sorting is done inside CommandBatchPreparer. This class has a protected method called TopologicalSort that sorts all queued commands for execution order. I could inherit from CommandBatchPreparer, but this is very specialized for ModificationCommand. So in my case it's not possible.

I've prepared my own entity sorter based on that information. Keep in mind that entities that have internal self referencing properties with ForeignKey need extra handling to sort based on internal references.

public class DbContextEntityProperty
{
    public PropertyInfo Property { get; set; }
    public Type Type { get; set; }
    public bool SelfReferencing { get; set; }
    public IEntityType EntityType { get; set; }
}

public List<DbContextEntityProperty> SortedEntityProperties()
{
    var result = new List<DbContextEntityProperty>();
    var properties = GetType().GetProperties()
        .Where(o => o.PropertyType.IsGenericType && (o.PropertyType.GetGenericTypeDefinition() == typeof(DbSet<>)))
        .Select(o =>
        {
            var t = o.PropertyType.GetGenericArguments()[0];
            return new DbContextEntityProperty { Property = o, Type = null, SelfReferencing = false,EntityType = Model.FindEntityType(t) };
        })
        .ToDictionary(o => o.EntityType);
    Action<DbContextEntityProperty> recursiveDependencyTraverser = null;
    recursiveDependencyTraverser = (p) =>
    {
        if (p.Type == null)
        {
            p.Type = p.EntityType.ClrType;
            foreach (var et in p.EntityType.GetDeclaredForeignKeys().Select(o => o.PrincipalEntityType))
                if (et.ClrType == p.Type)
                    p.SelfReferencing = true;
                else if (properties.TryGetValue(et, out var nextP))
                    recursiveDependencyTraverser(nextP);
            result.Add(p);
        }
    };
    foreach (var p in properties.Values)
        recursiveDependencyTraverser(p);
    return result;
}

Answer 2

To programmatically determine the foreign keys in your context you can use something like

        Dictionary<string, Dictionary<string, IEnumerable<IForeignKey>>> foreignKeys = new Dictionary<string, Dictionary<string, IEnumerable<IForeignKey>>>();
        IEnumerable<IEntityType> ets = dbContext.Model.GetEntityTypes();
        foreach (var (et, prop) in from IEntityType et in ets
                                   let entityProps = et.GetProperties()
                                   from IProperty prop in entityProps
                                   select (et, prop))
        {
            IEnumerable<IForeignKey> fks = et.FindForeignKeys(prop);
            if(fks.Any())
            {
                if (!foreignKeys.Keys.Contains(et.DisplayName()))
                {
                    foreignKeys.Add(et.DisplayName(), new Dictionary<string, IEnumerable<IForeignKey>>());
                }

                foreignKeys[et.DisplayName()].Add(prop.Name, fks);
            }
        } 

This will create a dictionary of tables names, each dictionary entry will be a dictionary of fields and the foreign keys attached to the field ie

IEnumerable<IForeignKey> fks = foreignKeys["TableName"]["FieldName"];

From this you should be able to work out what data is required for you to insert your records into the database.