如何在 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

掘金：

1. EFCore 3.1.2
2. EFCore.Design 3.1.2
3. EFCore.Tools 3.1.2
4. EFCore.Relational 3.1.2
5. EFCore.SqlServer 3.1.2

实施

1. 添加 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>());
       }
   }
   

2. 创建一个通用枚举函数

   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;
       }
   }
   

3. 创建接口

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

4. 将接口应用到模型

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

5. 添加您的迁移

   PM> add-migration SeedUserRoleTable
   

您应该会看到添加的迁移

6. 使用种子信息更新数据库

   PM> update-database
   

Answer 2

在我的 Entities 文件夹下，我添加了 Enumerations 文件夹，我有这个：

public class UserStatus : Enumeration
    {

        public static readonly UserStatus New = new UserStatus(1, "New");
        public static readonly UserStatus UnderReview = new UserStatus(2, "UnderReview");
        public static readonly UserStatus Customer = new UserStatus(3, "Customer");
        public static readonly UserStatus Approved = new UserStatus(4, "Approved");
        public static readonly UserStatus Declined = new UserStatus(5, "Declined");

        public UserStatus(int id, string name)
                    : base(id, name)
        {
        }
    }

在我的 DataContext.cs 中，我只是使用它来播种数据：

 modelBuilder.Entity<UserStatus>().HasData(Enumeration.GetAll<UserStatus>());

UserStatus 未在 DataContext.cs 中声明为 DbSet

.NET Core 3.1

Answer 3

您不能将枚举类型的值植入数据库，因为枚举不是实体类型，因此首先不会进入数据库。换句话说，不会有Categories 表，因此将值播种到那个不存在的表中是没有意义的。

如果你想真正将你的类别持久化在你的数据库中，那么你需要创建一个类似的类：

public class Category
{
    public int Id { get; set; }
    public string Name { get; set; }
}