在 EF Core 中为所有实体设置字符串枚举转换器的通用方法

Question

我正在尝试将字符串枚举值转换器设置为我的 EF Core Code-First 项目中所有实体的所有enum属性。

我可以像这样手动执行此操作：

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        // Do this for every single enum property in each of the entities
        modelBuilder.Entity<MyEntity>().Property(e => e.MyEnum).HasConversion<string>();
    }

但我正在寻找一种使用扩展方法为所有属性执行此操作的通用方法。 我尝试使用以下代码，但出现以下错误：

model 类型“对象”的转换器不能用于“MyEnum”

    public static void SetEnumStringConverter(this ModelBuilder modelBuilder)
    {
        var properties = modelBuilder?.Model.GetEntityTypes().SelectMany(x => x.GetProperties().Where(y => typeof(Enum).IsAssignableFrom(y.ClrType)));
        foreach (var property in properties)
        {
            property.SetValueConverter(new ValueConverter<object, string>(v => v.ToString(), v => (object)Enum.Parse(property.ClrType, v)));
        }
    }

还尝试了以下，给了我同样的错误

    property.SetValueConverter(new ValueConverter<Enum, string>(v => v.ToString(), v => Enum.Parse(property.ClrType, v)));

显然是因为枚举不是 object？

然而，有没有其他方法可以通用的方式做到这一点？

Answer 1

所以，经过一番挖掘，我找到了实现这一目标的方法。 虽然不简单，但得到了想要的结果。

    public static void SetEnumStringConverter(this ModelBuilder modelBuilder)
    {
        foreach (var entityType in modelBuilder?.Model.GetEntityTypes())
        {
            var entityBuilder = modelBuilder.Entity(entityType.ClrType);
            foreach (var property in entityType.GetProperties())
            {
                if (typeof(Enum).IsAssignableFrom(property.ClrType))
                {
                    entityBuilder.Property(property.ClrType, property.Name).HasConversion<string>();
                }
            }
        }
    }

Answer 2

您可以简单地使用SetProviderClrType扩展方法，例如

public static void SetEnumStringConverter(this ModelBuilder modelBuilder)
{
    var properties = modelBuilder.Model.GetEntityTypes()
        .SelectMany(e => e.GetProperties())
        .Where(p => (Nullable.GetUnderlyingType(p.ClrType) ?? p.ClrType).IsEnum);

    foreach (var property in properties)
       property.SetProviderClrType(typeof(string)); // <--
}