如何通过反射从EF Core 2反向生成的.dll中获取实体主键？

针对你需要从指定DLL中提取EF Core实体主键的需求，这里提供一个通用的反射方案——完全不需要实例化DbContext，特别适合你在代码生成工具里批量处理10个数据库项目的场景。

核心思路 #

EF Core的实体主键配置通常有两种形式：要么直接写在DbContext的OnModelCreating方法里，要么封装成实现IEntityTypeConfiguration<T>的独立配置类。我们可以通过反射加载目标程序集，收集这些配置信息并构建出实体模型，进而提取主键信息。

实现代码 #

下面是完整的C#代码示例，包含详细注释：

using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Metadata;

public static class EntityPrimaryKeyExtractor
{
    public static string GetEntityPrimaryKey(string dllPath, string entityTypeName)
    {
        // 1. 加载目标DLL对应的程序集
        var targetAssembly = Assembly.LoadFrom(dllPath);
        
        // 2. 定位目标实体类型
        var entityType = targetAssembly.GetType(entityTypeName);
        if (entityType == null)
            throw new ArgumentException($"未在指定DLL中找到实体类型：{entityTypeName}");

        // 3. 初始化ModelBuilder，用于构建EF Core模型
        var modelBuilder = new ModelBuilder();
        
        // 3.1 查找并应用所有独立的IEntityTypeConfiguration<T>配置类
        var configClasses = targetAssembly.GetTypes()
            .Where(type => type.GetInterfaces().Any(iface => 
                iface.IsGenericType && iface.GetGenericTypeDefinition() == typeof(IEntityTypeConfiguration<>)));
        
        foreach (var configClass in configClasses)
        {
            // 获取当前配置类对应的实体类型
            var configEntityType = configClass.GetInterfaces()
                .First(iface => iface.IsGenericType)
                .GetGenericArguments()[0];
            
            // 通过反射调用ModelBuilder的ApplyConfiguration方法加载配置
            var applyMethod = typeof(ModelBuilder).GetMethod("ApplyConfiguration")!
                .MakeGenericMethod(configEntityType);
            var configInstance = Activator.CreateInstance(configClass);
            applyMethod.Invoke(modelBuilder, new[] { configInstance });
        }

        // 3.2 查找所有DbContext派生类，合并它们的模型配置
        var dbContextTypes = targetAssembly.GetTypes()
            .Where(type => typeof(DbContext).IsAssignableFrom(type) && !type.IsAbstract);
        
        foreach (var dbContextType in dbContextTypes)
        {
            // 创建空的DbContextOptions（不需要真实数据库连接，格式合法即可）
            var dummyOptions = new DbContextOptionsBuilder()
                .UseSqlServer("Server=.;Database=Dummy;Trusted_Connection=True;")
                .Options;
            
            // 实例化DbContext（轻量级操作，仅用于触发模型构建）
            using var dbContext = (DbContext)Activator.CreateInstance(dbContextType, dummyOptions)!;
            dbContext.Model.GetEntityTypes(); // 强制触发模型构建
            modelBuilder.Model.Merge(dbContext.Model); // 合并模型配置
        }

        // 4. 从构建好的模型中提取目标实体的主键
        var entityInModel = modelBuilder.Model.FindEntityType(entityType);
        if (entityInModel == null)
            throw new InvalidOperationException($"实体{entityTypeName}未在EF Core模型中注册");

        var primaryKey = entityInModel.FindPrimaryKey();
        if (primaryKey == null || primaryKey.Properties.Count == 0)
            throw new InvalidOperationException($"实体{entityTypeName}未定义主键");

        // 如果是复合主键，返回逗号分隔的名称；否则返回单个主键名称
        return string.Join(", ", primaryKey.Properties.Select(p => p.Name));
    }
}

使用示例 #

// 调用示例，替换成你的实际路径和实体名称
var dllPath = @"C:\OnlineShoppingStore\bin\Debug\netcoreapp2.2\OnlineShoppingStore.dll";
var targetEntity = "Product";
var primaryKeyName = EntityPrimaryKeyExtractor.GetEntityPrimaryKey(dllPath, targetEntity);
Console.WriteLine($"实体{targetEntity}的主键是：{primaryKeyName}");
// 预期输出：实体Product的主键是：ProductInfoId

关键说明 #

• 无需真实数据库：我们用的是虚构的连接字符串，因为只需要构建模型，不需要访问实际数据库。
• 兼容两种配置方式：同时支持独立配置类和DbContext内联配置的场景，不会遗漏任何主键定义。
• 复合主键支持：如果实体使用复合主键，会自动返回逗号分隔的所有主键成员名称。
• 批量处理友好：可以轻松封装成循环逻辑，一次性处理多个数据库项目的DLL文件。

注意事项 #

1. 确保目标DLL引用的EF Core版本和你的代码一致（示例适配EF Core 2.2，和你的环境匹配）。
2. 如果DLL中有多个DbContext，代码会自动合并所有模型配置，保证实体信息完整。
3. 代码中实例化DbContext是轻量级操作，仅用于触发模型构建，不会建立数据库连接，这比手动调用受保护的OnModelCreating方法更可靠。

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