asp.net - 如何在 Entity Framework Core 2.1 中播种枚举

Question

我是 EF Core 的新手，正在尝试播种枚举。

根据Data Seeding ，此功能是 EF Core 2.1 的新增功能。

我回顾了几个解决方案，包括这个 SO solution by Blake Mumford ，但这对我不起作用，因为枚举不是引用类型。

我的目标(在迁移的帮助下)是将 Category 枚举填充到一个名为 Category 的新 SQL 表中，并让我的 Payment 表包含一个引用 Category 表作为外键的列。

任何帮助将不胜感激。谢谢。

public partial class Payment
{
    public int PaymentId { get; set; }
    public DateTime PostedDate { get; set; }
    public string Vendor { get; set; }
    public decimal Amount { get; set; }

    public virtual Category Category { get; set; }
}

public enum Category
{
    Restaurants,
    Groceries,
    [Display(Name = "Home Goods")]
    HomeGoods,
    Entertainment
}

public partial class MyDbContext : DbContext
{
    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        // Error: The type Category must be a reference type in order to use it as parameter TEntity in the 
        // genric type or method ModelBuilder.Entity<TEntity>()

        modelBuilder.Entity<Category>().HasData(Category.Restaurants, 
                                                Category.Groceries, 
                                                Category.HomeGoods, 
                                                Category.Entertainment);
    }
}

Answer 1

这就是我所做的，希望它适用于您的环境。

环境

项目框架

.NetCore 3.0

核桃:

EFCore 3.1.2

EFCore.Design 3.1.2

EFCore.Tools 3.1.2

EFCore.Relational 3.1.2

EFCore.SqlServer 3.1.2

实现

添加 OnModelCreating

public partial class MyDbContext : DbContext
{
    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        // Error: The type Category must be a reference type in order to use it as parameter TEntity in the 
        // genric type or method ModelBuilder.Entity<TEntity>()

        modelBuilder.Entity<UserRole>().HasData(EnumFunctions.GetModelsFromEnum<UserRole, UserRoleEnum>());
    }
}

创建一个通用枚举函数

public static class EnumFunctions
{
    public static IEnumerable<TModel> GetModelsFromEnum<TModel, TEnum>() where TModel : IEnumModel<TModel, TEnum>, new()
    {
        var enums = new List<TModel>();
        foreach (var enumVar in (TEnum[])Enum.GetValues(typeof(TEnum)))
        {
            enums.Add(new TModel
            {
                Id = enumVar,
                Name = enumVar.ToString()
            });
        }

        return enums;
    }
}

创建接口(interface)

public interface IEnumModel<TModel, TModelIdType>
{
    TModelIdType Id { get; set; }
    string Name { get; set; }
}

将接口(interface)应用于模型

[Table("UserRoles")]
public class UserRole : IEnumModel<UserRole, UserRoleEnum>
{
    [Key]
    public UserRoleEnum Id { get; set; }
    public string Name { get; set; }       
}

添加您的迁移

PM> add-migration SeedUserRoleTable

您应该看到添加的迁移

使用种子信息更新数据库

PM> update-database