快捷導(dǎo)航

Entity?Framework?Core種子數(shù)據(jù)Data-Seeding

更新時(shí)間：2022年03月25日 10:09:28 作者：.NET開發(fā)菜鳥

這篇文章介紹了Entity?Framework?Core種子數(shù)據(jù)Data-Seeding的用法，文中通過示例代碼介紹的非常詳細(xì)。對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

一、什么是Data-Seeding

Data-Seeding是EntityFrameworkCore 2.1以上版本新增加的特性。在項(xiàng)目剛開始的時(shí)候，我們往往是需要初始化一些基礎(chǔ)數(shù)據(jù)到數(shù)據(jù)庫(kù)中，通過Data-Seeding特性就可以實(shí)現(xiàn)這一功能。本篇文章我們將講解如何進(jìn)行數(shù)據(jù)初始化。

二、初始化方法

具體的數(shù)據(jù)初始化方法分為如下三種：

模型中配置。這種是通過調(diào)用HasData()方法。
手動(dòng)遷移時(shí)添加。
自定義初始化邏輯。

下面我們分別來(lái)講解如何使用這三種方式進(jìn)行數(shù)據(jù)遷移。

1、模型中配置

這種方式是通過調(diào)用HasData()方法實(shí)現(xiàn)的。這種也是我在項(xiàng)目開發(fā)過程中，經(jīng)常使用的。這種方式是在數(shù)據(jù)上下文類中重寫OnModelCreating()方法，我們先看HasData()方法的定義：

可以看到，方法的參數(shù)可以是Blog類型的數(shù)組，具體代碼如下：

using EFCore.Model;
using Microsoft.EntityFrameworkCore;

namespace EFCore.Data
{
    /// <summary>
    /// 數(shù)據(jù)上下文
    /// </summary>
    public class EFDbContext:DbContext
    {
        protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
        {
            optionsBuilder.UseSqlServer("Data Source=.;Initial Catalog=EFTest;User ID=sa;Password=123456;");
        }

        public DbSet<Blog> Blogs { get; set; }


        /// <summary>
        /// 重寫OnModelCreating方法
        /// </summary>
        /// <param name="modelBuilder"></param>
        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            // 針對(duì)Blog實(shí)體添加種子數(shù)據(jù)
            modelBuilder.Entity<Blog>().HasData(
                new Blog()
                {
                    // Id字段要賦值，否則會(huì)報(bào)錯(cuò)
                    Id=1,
                    Name="ef core"
                },
                new Blog()
                {
                    Id=2,
                    Name="ASP.NET Core"
                },
                new Blog()
                {
                    Id=3,
                    Name="圖解數(shù)據(jù)結(jié)構(gòu)"
                }
                );
            base.OnModelCreating(modelBuilder);
        }
    }
}

我們注意到：默認(rèn)情況下會(huì)自動(dòng)設(shè)置Id列為主鍵，并且是自動(dòng)增長(zhǎng)的。但是這里要設(shè)置Id的值，即使Id是自動(dòng)生成的主鍵，否則會(huì)報(bào)下圖所示的錯(cuò)誤：

添加完種子數(shù)據(jù)以后，我們運(yùn)行程序，查看輸出結(jié)果：

查看數(shù)據(jù)庫(kù)：

這樣就生成了數(shù)據(jù)庫(kù)和表，而且表里面也有了初始化數(shù)據(jù)。

假如這時(shí)候我們想增加一條數(shù)據(jù)，代碼如下：

using EFCore.Model;
using Microsoft.EntityFrameworkCore;

namespace EFCore.Data
{
    /// <summary>
    /// 數(shù)據(jù)上下文
    /// </summary>
    public class EFDbContext:DbContext
    {
        protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
        {
            optionsBuilder.UseSqlServer("Data Source=.;Initial Catalog=EFTest;User ID=sa;Password=123456;");
        }

        public DbSet<Blog> Blogs { get; set; }


        /// <summary>
        /// 重寫OnModelCreating方法
        /// </summary>
        /// <param name="modelBuilder"></param>
        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            // 針對(duì)Blog實(shí)體添加種子數(shù)據(jù)
            modelBuilder.Entity<Blog>().HasData(
                new Blog()
                {
                    // Id字段要賦值，否則會(huì)報(bào)錯(cuò)
                    Id=1,
                    Name="ef core"
                },
                new Blog()
                {
                    Id=2,
                    Name="ASP.NET Core"
                },
                new Blog()
                {
                    Id=3,
                    Name="圖解數(shù)據(jù)結(jié)構(gòu)"
                },
                // 新增加一條數(shù)據(jù)
                new Blog()
                {
                    Id=4,
                    Name="C#高級(jí)編程"
                }
                
                );
            base.OnModelCreating(modelBuilder);
        }
    }
}

這時(shí)候還能不能用剛才的方法呢？我們這時(shí)在運(yùn)行程序，查看結(jié)果：

這時(shí)候程序執(zhí)行失敗了，而且表里面的數(shù)據(jù)也沒有增加。這說(shuō)明context.Database.EnsureCreated()方法只有在第一次執(zhí)行的時(shí)候才會(huì)有效，以后數(shù)據(jù)進(jìn)行更改后就無(wú)效了。那么有什么方式可以實(shí)現(xiàn)呢？這時(shí)只有通過命令行進(jìn)行遷移或者通過context.Database.Migrate方法調(diào)用生成的遷移類才能對(duì)數(shù)據(jù)的更改有效。

我們把剛才生成的數(shù)據(jù)庫(kù)刪掉，新增加的那條數(shù)據(jù)注釋掉，然后使用命令行遷移的方式生成數(shù)據(jù)庫(kù)表，首先添加遷移：

然后更新數(shù)據(jù)庫(kù)：

更新完數(shù)據(jù)庫(kù)以后，我們?cè)贖asData()方法里面添加一條數(shù)據(jù)，然后再次執(zhí)行上面的添加遷移和更新數(shù)據(jù)庫(kù)的命令，發(fā)現(xiàn)這時(shí)候數(shù)據(jù)庫(kù)里面會(huì)添加新增加的數(shù)據(jù)。

在執(zhí)行完第二次添加遷移命令后，如果不使用更新數(shù)據(jù)庫(kù)命令，也可以通過代碼的方式進(jìn)行遷移，這就是調(diào)用context.Database.Migrate命令，代碼如下：

using EFCore.Data;
using Microsoft.EntityFrameworkCore;
using System;

namespace EFCore.Con
{
    class Program
    {
        static void Main(string[] args)
        {
            Console.WriteLine("Hello World!");

            EFDbContext dbContext = new EFDbContext();
            // 遷移
            dbContext.Database.Migrate();
            //bool tfTrue = dbContext.Database.EnsureCreated();
            //if(tfTrue)
            //{
            //    Console.WriteLine("數(shù)據(jù)庫(kù)創(chuàng)建成功!");
            //}
            //else
            //{
            //    Console.WriteLine("數(shù)據(jù)庫(kù)創(chuàng)建失敗!");
            //}

            Console.ReadKey();
        }
    }
}

這時(shí)會(huì)自動(dòng)調(diào)用最新的遷移文件去更新數(shù)據(jù)庫(kù)。

注意：調(diào)用該方法對(duì)數(shù)據(jù)的更改只有在遷移時(shí)才能生效，也就是說(shuō)只有通過命令進(jìn)行遷移或者通過context.Database.Migrate方法調(diào)用生成的遷移類才能對(duì)數(shù)據(jù)更改有效。而調(diào)用context.Database.EnsureCreated()方法只有在第一次執(zhí)行的時(shí)候才有效，但數(shù)據(jù)進(jìn)行更改后將無(wú)效。

這種方式有兩種限制：

必須指定主鍵的值（即使主鍵由數(shù)據(jù)庫(kù)自動(dòng)生成也要指定值）。
添加的必須是靜態(tài)數(shù)據(jù)，沒有任何的依賴。比如添加的Id主鍵的值在其它表里面有引用就不可以。

2、手動(dòng)遷移時(shí)添加

這種方式在這里不進(jìn)行講解，有興趣的可以參考微軟的官方文檔。

3、自定義初始化邏輯

執(zhí)行數(shù)據(jù)種子設(shè)定的一種簡(jiǎn)單而有效的方法是在主應(yīng)用程序邏輯開始執(zhí)行之前使用使用DbContext.SaveChanges()。代碼如下：

using EFCore.Data;
using EFCore.Model;
using Microsoft.EntityFrameworkCore;
using System;
using System.Linq;

namespace EFCore.Con
{
    class Program
    {
        static void Main(string[] args)
        {
            Console.WriteLine("Hello World!");

            //EFDbContext dbContext = new EFDbContext();
            //// 遷移
            //dbContext.Database.Migrate();
            ////bool tfTrue = dbContext.Database.EnsureCreated();
            ////if(tfTrue)
            ////{
            ////    Console.WriteLine("數(shù)據(jù)庫(kù)創(chuàng)建成功!");
            ////}
            ////else
            ////{
            ////    Console.WriteLine("數(shù)據(jù)庫(kù)創(chuàng)建失敗!");
            ////}

            #region 使用自定義初始化邏輯
            using(EFDbContext context=new EFDbContext())
            {
                context.Database.EnsureCreated();

                var testBlog = context.Blogs.FirstOrDefault(p => p.Name == "C#");
                if(testBlog == null)
                {
                    // 添加數(shù)據(jù)
                    context.Blogs.Add(new Blog()
                    {
                        Name = "C#"
                    });
                }

                // 保存數(shù)據(jù)
                context.SaveChanges();
            }
            #endregion
            Console.ReadKey();
        }
    }
}

通過這種自定義邏輯的方式也可以添加種子數(shù)據(jù)。如果HasData()方法里面添加了種子數(shù)據(jù)，那么會(huì)先把HasData()方法里面的種子數(shù)據(jù)添加到數(shù)據(jù)庫(kù)中。如果沒有名稱為C#的數(shù)據(jù)，則還會(huì)在添加一條數(shù)據(jù)。

如果有了該數(shù)據(jù)，就不會(huì)再添加了。

注意：這種方式新增數(shù)據(jù)的時(shí)候就不能再給主鍵Id賦值了，因?yàn)槭窍壬蓴?shù)據(jù)庫(kù)，自動(dòng)設(shè)置Id為主鍵，在添加數(shù)據(jù)的時(shí)候會(huì)自動(dòng)賦值。

三、Data-Seeding本質(zhì)

當(dāng)調(diào)用HasData()方法首次遷移時(shí)，實(shí)質(zhì)上是調(diào)用MigrationBuilder類中InsertData方法進(jìn)行插入。
當(dāng)調(diào)用HasData()方法更改數(shù)據(jù)（未更改主鍵）時(shí)，實(shí)質(zhì)上是調(diào)用MigrationBuilder類中UpdateData方法進(jìn)行更新操作。
當(dāng)調(diào)用HasData()方法移除數(shù)據(jù)或更改主鍵時(shí)，實(shí)質(zhì)上是調(diào)用MigrationBuilder類中DeleteData方法進(jìn)行刪除操作或者刪除和更新操作。