麓野 发表于 2024-8-5 01:32:38

【EF Core】自动生成的字段值

自动生成字段值,咱们首先想到的是主键列(带 IDENTITY 的主键)。EF Core 默认的主键配置也是启用 Identity 自增长的,而且可以自动标识主键。前提是代表主键的实体属性名要符合以下规则:
1、名字叫 ID、id、或 Id,就是不分大小写;
2、名字由实体类名 + Id 构成。比如,Car 实体类,包含一个属性叫 CarID 或 CarId;
3、属性类型是整数类型(int、long、ushort 等,但不是 byte)或 GUID。
这些识别主键的规则是由一种叫“约定”(Convension)的东西实现的,具体来说,是一个叫 KeyDiscoveryConvention 的类。老周放一小段源代码给各位瞧瞧。
public class KeyDiscoveryConvention :
    IEntityTypeAddedConvention,
    IPropertyAddedConvention,
    IKeyRemovedConvention,
    IEntityTypeBaseTypeChangedConvention,
    IEntityTypeMemberIgnoredConvention,
    IForeignKeyAddedConvention,
    IForeignKeyRemovedConvention,
    IForeignKeyPropertiesChangedConvention,
    IForeignKeyUniquenessChangedConvention,
    IForeignKeyOwnershipChangedConvention,
    ISkipNavigationForeignKeyChangedConvention
{
    private const string KeySuffix = "Id";

    ……

    public static IEnumerable<IConventionProperty> DiscoverKeyProperties(
      IConventionEntityType entityType,
      IEnumerable<IConventionProperty> candidateProperties)
    {
      Check.NotNull(entityType, nameof(entityType));

      // ReSharper disable PossibleMultipleEnumeration
      var keyProperties = candidateProperties.Where(p => string.Equals(p.Name, KeySuffix, StringComparison.OrdinalIgnoreCase));
      if (!keyProperties.Any())
      {
            var entityTypeName = entityType.ShortName();
            keyProperties = candidateProperties.Where(
                p => p.Name.Length == entityTypeName.Length + KeySuffix.Length
                  && p.Name.StartsWith(entityTypeName, StringComparison.OrdinalIgnoreCase)
                  && p.Name.EndsWith(KeySuffix, StringComparison.OrdinalIgnoreCase));
      }

      return keyProperties;
      // ReSharper restore PossibleMultipleEnumeration
    }
   ……
}这几个逻辑 And 其实就是查找 Id 格式的属性名,如 StudentID、CarId、OrderID…… 外键的发现原理也跟主键一样。
用 Sqlite 数据举一个简单的例子。下面是实体类(假设它用来表示输入法信息):
public class InputMethod
{
    public ushort RecoId { get; set; }
    public string? MethodDisplay { get; set; }
    public string? Description { get; set; }
    public string? Culture { get; set; }
}如你所见,这个类作为主键的属性是 RecoId,但是,它的命名是无法被自动识别的,咱们必须明确地告诉 EF,它是主键。方法有二:
1、批注法。直接在属性上应用相关的特性类。如
public class InputMethod
{
   
    public ushort RecoId { get; set; }
    ……
}2、重写 DbContext 类的 OnModelCreating 方法。如
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    modelBuilder.Entity<InputMethod>().HasKey(e => e.RecoId);
}如果使用了上面重写 OnModelCreating 方法,那么,你的 DbContext 派生类已经能识别 InputMethod 实体类了。但如果你用的是在属性上应用 特性的方式,那么 DbContext 的派生类是识别不到实体类的,你需要将它的集合声明为 DbContext 的属性。
internal class TestDBContext : DbContext
{
    // 构造函数
    public TestDBContext(DbContextOptions<TestDBContext> opt)
      : base(opt)
    { }

    // 将实体集合声明为属性
    public DbSet<InputMethod> InputMethods { get; set; }
}注意,数据记录的集合要用 DbSet,其他类型的集合是不行的哟。比如,你改成这样,就会报错。
public List<InputMethod> InputMethods { get; set; }
说明人家只认 DbSet 集合,其他集合无效。
这里老周选用服务容器来配置。
static void Main(string[] args)
{
    IServiceCollection services = new ServiceCollection();
    // 构建连接字符串
    SqliteConnectionStringBuilder constrbd = new();
    constrbd.DataSource = "abc.db";
    // 添加 Sqlite 功能
    services.AddSqlite<TestDBContext>(
            connectionString:    constrbd.ToString(),
            optionsAction:       dcopt =>
            {
                dcopt.LogTo(msg => Console.WriteLine(msg), LogLevel.Information);
            }
      );
    // 生成服务列表
    var svcProd = services.BuildServiceProvider();
    if(svcProd == null)
    {
      return;
    }

    // 访问数据上下文
    using TestDBContext dbc = svcProd.GetRequiredService<TestDBContext>();
    ……
}连接字符串你可以直接用字符串写,不用 ConnectionStringBuilder。默认的 SQLite 库是不支持密码的,所以老周就不设置密码了。在调用 AddSqlite 方法时,有一个名为 optionsAction 的参数,咱们可以用它配置日志输出。LogTo 方法配置简单,只要提供一个委托,它绑定的方法只要有一个 string 类型的输入参数就行,这个字符串参数就是日志文本。
配置日志功能后,运行程序时,控制台能看到执行的 SQL 语句。
下面咱们来创建数据库,然后插入两条 InputMethod 记录。
// 访问数据上下文
using TestDBContext dbc = svcProd.GetRequiredService<TestDBContext>();
// 删除数据库
dbc.Database.EnsureDeleted();
// 创建数据库
dbc.Database.EnsureCreated();

// 尝试插入两条记录
InputMethod[] ents = [
      new(){MethodDisplay = "双拼输入", Description="按两个键完成一个音节",Culture="zh-CN"},
      new() {MethodDisplay = "六指输入", Description="专供六个指头的人使用",Culture="zh-CN"}
    ];
dbc.Set<InputMethod>().AddRange(ents);
int result = dbc.SaveChanges();
Console.WriteLine($"更新记录数:{result}");

// 打印插入的记录
foreach(InputMethod im in dbc.Set<InputMethod>())
{
    Console.WriteLine($"ID={im.RecoId}, Display={im.MethodDisplay}, Culture={im.Culture}");
}这里是为了测试,调用了 EnsureDeleted 方法,实际应用时一般不要调用。因为这个方法的功能是把现存的数据库删除。如果调用了此方法,那应用程序每次启动都会删掉数据库,那用户肯定会投诉你的。EnsureCreated 方法可以使用,它的功能是如果数据库不存在,就创建新数据库;如果数据库存在,那啥也不做。所以,调用 EnsureCreated 方法不会造成数据丢失,放心用。
插入数据和调用 SaveChanges 方法保存到数据库的代码,相信大伙都很熟了,老周就不介绍了。
程序运行之后,将得到这样的日志:
info: 2024/8/4 12:48:11.517 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (10ms) , CommandType='Text', CommandTimeout='30']
      PRAGMA journal_mode = 'wal';
info: 2024/8/4 12:48:11.582 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (0ms) , CommandType='Text', CommandTimeout='30']
      CREATE TABLE "tb_ims" (
          "RecoId" INTEGER NOT NULL CONSTRAINT "PK_tb_ims""MethodDisplay""Description""Culture" TEXT NULL
      );
info: 2024/8/4 12:48:11.700 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (3ms) , CommandType='Text', CommandTimeout='30']
      INSERT INTO "tb_ims" ("Culture", "Description", "MethodDisplay")
      VALUES (@p0, @p1, @p2)
      RETURNING "RecoId";
info: 2024/8/4 12:48:11.712 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (0ms) , CommandType='Text', CommandTimeout='30']
      INSERT INTO "tb_ims" ("Culture", "Description", "MethodDisplay")
      VALUES (@p0, @p1, @p2)
      RETURNING "RecoId";
更新记录数:2
info: 2024/8/4 12:48:11.849 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (0ms) , CommandType='Text', CommandTimeout='30']
      SELECT "t"."RecoId", "t"."Culture", "t"."Description", "t"."MethodDisplay"
      FROM "tb_ims" AS "t"
ID=1, Display=双拼输入, Culture=zh-CN
ID=2, Display=六指输入, Culture=zh-CN这样你会发现,对于整数类型的主键,默认是自动生成递增ID的。注意,这个是由数据库生成的,而不是 EF Core 的生成器。不同数据库的 SQL 语句会有差异。
为了对比,咱们不防改为 SQL Server,看看输出的日志。
// 构建连接字符串
SqlConnectionStringBuilder constrbd = new();
constrbd.DataSource = ".\\SQLTEST";
constrbd.InitialCatalog = "CrazyDB";
constrbd.IntegratedSecurity = true;
// 不信任服务器证书有时候会连不上
constrbd.TrustServerCertificate = true;
// 可读可写
constrbd.ApplicationIntent = ApplicationIntent.ReadWrite;

// 添加 SQL Server 功能
services.AddSqlServer<TestDBContext>(
      connectionString: constrbd.ToString(),
      optionsAction: opt =>
      {
            opt.LogTo(logmsg => Console.WriteLine(logmsg), LogLevel.Information);
      });其他代码不变,再次运行。输出的日志如下:
info: 2024/8/4 13:01:06.087 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (115ms) , CommandType='Text', CommandTimeout='60']
      CREATE DATABASE ;
info: 2024/8/4 13:01:06.122 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (31ms) , CommandType='Text', CommandTimeout='60']
      IF SERVERPROPERTY('EngineEdition') <> 5
      BEGIN
          ALTER DATABASE SET READ_COMMITTED_SNAPSHOT ON;
      END;
info: 2024/8/4 13:01:06.137 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (5ms) , CommandType='Text', CommandTimeout='30']
      SELECT 1
info: 2024/8/4 13:01:06.181 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (10ms) , CommandType='Text', CommandTimeout='30']
      CREATE TABLE (
          int NOT NULL IDENTITY,
          nvarchar(12) NOT NULL,
          nvarchar(max) NULL,
          nvarchar(max) NULL,
          CONSTRAINT PRIMARY KEY ()
      );
info: 2024/8/4 13:01:06.317 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (30ms) , CommandType='Text', CommandTimeout='30']
      SET IMPLICIT_TRANSACTIONS OFF;
      SET NOCOUNT ON;
      MERGE USING (
      VALUES (@p0, @p1, @p2, 0),
      (@p3, @p4, @p5, 1)) AS i (, , , _Position) ON 1=0
      WHEN NOT MATCHED THEN
      INSERT (, , )
      VALUES (i., i., i.)
      OUTPUT INSERTED., i._Position;
更新记录数:2
info: 2024/8/4 13:01:06.438 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (2ms) , CommandType='Text', CommandTimeout='30']
      SELECT ., ., ., .
      FROM AS
ID=1, Display=双拼输入, Culture=zh-CN
ID=2, Display=六指输入, Culture=zh-CNA、使用 Sqlite 数据库时,生成的 CREATE TABLE 语句,主键列是 PRIMARY KEY AUTOINCREMENT;
B、使用 SQL Server 时,主键列使用的是 IDENTITY,默认以 1 为种子,增量是 1。所以插入记录的键值是1和2。
 
有时候我们并不希望主键列自动生成值,同样有两种配置方法:
1、通过特性类来批注。如
public class InputMethod
{
   
    public ushort RecoId { get; set; }
    public string? MethodDisplay { get; set; }
    public string? Description { get; set; }
    public string? Culture { get; set; }
}将 DatabaseGeneratedOption 设置为 None,就取消列的自动生成了。
2、通过模型配置,即重写 OnModelCreating 方法实现。
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    modelBuilder.Entity<InputMethod>().HasKey(e => e.RecoId);
    modelBuilder.Entity<InputMethod>()
      .Property(k => k.RecoId)
      .ValueGeneratedNever();
}这种情况下,插入数据时主键列就需要咱们手动赋值了。
======================================================================================
上面的是热身运动,是比较简单的应用方案。下面老周给大伙伴解决一个问题。老周看到在 GitHub 等平台上有人提问,但没有得到解决。如果你看到老周这篇水文并且你有此困惑,那你运气不错。好,F话不多说,咱们看问题。
需求:主键不变,但是我不想让它带有 IDENTITY,插入记录时用我自定义的方式生成主键的值。这个需要的本质就是:我不要数据库给我生成递增ID,我要在程序里生成。
前面老周提过,默认行为下主键列如果是整数类型或 GUID,就会产生自增长的列。所以,咱们有一个很关键的步骤——就是怎么禁止 EF 去产生 IDENTITY 列。如果你看到 EF Core SQL Server 的源代码,可能你会知道有个约定类叫 SqlServerValueGenerationStrategyConvention。这个约定类默认会设置主键列的自动生成策略为 IdentityColumn。
public virtual void ProcessModelInitialized(
   IConventionModelBuilder modelBuilder,
   IConventionContext<IConventionModelBuilder> context)
   => modelBuilder.HasValueGenerationStrategy(SqlServerValueGenerationStrategy.IdentityColumn);于是,有大伙伴可能会想到,那我从 SqlServerValueGenerationStrategyConvention 派生出一个类,重写 ProcessModelInitialized 方法,把自动生成策略改为 None,然后在约定集合中替换掉 SqlServerValueGenerationStrategyConvention。
这个思路不是不行,就是工作量大一些。你不仅要定义个新类,还要把它注册到服务容器中替换 SqlServerValueGenerationStrategyConvention 。毕竟 EF Core 框架内部也是使用了服务容器和依赖注入的方式来组织各种组件的。具体做法是在初始化 DbContext 类(包括你派生的类)时会传递一个 DbContextOptions 对象,它有一个 ReplaceService 方法,可以替换容器中的服务。在调用 AddSqlServer 方法时就可以配置。
public static IServiceCollection AddSqlServer<TContext>(
   this IServiceCollection serviceCollection,
   string? connectionString,
   Action<SqlServerDbContextOptionsBuilder>? sqlServerOptionsAction = null,
   Action<DbContextOptionsBuilder>? optionsAction = null)
   where TContext : DbContext上述方案太麻烦,故老周未采用。其实,就算服务初始化时设置了生成策略是 Identity,可我们可以在构建模型时修改它呀。做法就是重写 DbContext 类的 OnModelCreating 方法,然后通过 IConventionModelBuilder.HasValueGenerationStrategy 方法就能修改生成策略。当然,这里头是有点波折的,我们不能在 ModelBuilder 实例上调用,因为这货并不是直接实现 IConventionModelBuilder 接口的,它是这么搞的:
public class ModelBuilder : IInfrastructure<IConventionModelBuilder>IInfrastructure 接口的作用是把 T 隐藏,不希望程序代码访问类型T。DbContext 类也实现这个接口,但它隐藏的是 IServiceProvider 对象,不想让咱们访问里面注册的服务。也就是说,IConventionModelBuilder 的实现者被隐藏了。不过,EF Core 并没有把事情做得太绝,好歹给了一个扩展方法 GetInfrastructure。用这个扩展方法我们能得到 IConventionModelBuilder 类型的引用。
弄清楚这个原理,代码就好写了。
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    IConventionModelBuilder cvbd = modelBuilder.GetInfrastructure();
    if (cvbd.CanSetValueGenerationStrategy(Microsoft.EntityFrameworkCore.Metadata.SqlServerValueGenerationStrategy.None))
    {
      cvbd.HasValueGenerationStrategy(Microsoft.EntityFrameworkCore.Metadata.SqlServerValueGenerationStrategy.None);
    }

    ……
}把生成策略改为 None 后,生成主键列时就不会有 IDENTITY 了。
如果你乐意,可以在插入记录时手动给主键列赋值也行的。不过,为了能自动生成值,我们应该写一个自己的生成类。
public class MyValueGenerator : ValueGenerator<int>
{
    // 返回false表示这个生成的值不是临时,它最终要存入数据库的
    public override bool GeneratesTemporaryValues => false;

    private static readonly Random rand = new((int)DateTime.Now.Ticks);

    public override int Next(EntityEntry entry)
    {
      // 获取所有实体
      DbSet<InputMethod> ents = entry.Context.Set<InputMethod>();
      int newID = default;
      do
      {
            // 生成随机ID
            newID = rand.Next();
      }
      // 保证不重复
      while (ents.Any(x => x.RecoId == newID));
      // 返回新值
      return newID;
    }
}我这里的逻辑是这样的,值是随机生成的,但要用一个循环去检查这个值是不是已存在数据库中,如果存在,继续生成,直到数值不重复。
实现自定义生成器,有两个抽象类可供选择:
1、如果你生成的值,类型不确定(可能是int,可能是 long,可能是……),那就实现 ValueGenerator 类;
2、如果要生成的值是明确类型的,比如这里是 int,那就实现带泛型参数的 ValueGenerator 类。
这两个类有继承关系,ValueGenerator 派生自 ValueGenerator 类。需要实现的抽象成员:
A、GeneratesTemporaryValues 属性:只读,返回 bool 值。如果你生成的值是临时的,返回 true,不是临时的,返回 false。啥意思呢。临时的值表示暂时赋值给属性/字段,但 INSERT、UPDATE 时,这个值不会存入数据库;如果不是临时的值,最终会存进数据库。上面例子中,老周让它返回 false,就说明生成的这个值,要写入数据库的。
B、如果继承 ValueGenerator 类,请实现 NextValue 抽象方法,返回类型是 object,就是生成的值;如果继承的是 ValueGenerator,请实现 Next 方法,此方法返回的类型由泛型参数决定。上面例子中是 int。
写好生成类后,要把它应用到实体模型中,同样是重写 DbContext 类的 OnModelCreating 方法。
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    IConventionModelBuilder cvbd = modelBuilder.GetInfrastructure();
    if (cvbd.CanSetValueGenerationStrategy(Microsoft.EntityFrameworkCore.Metadata.SqlServerValueGenerationStrategy.None))
    {
      cvbd.HasValueGenerationStrategy(Microsoft.EntityFrameworkCore.Metadata.SqlServerValueGenerationStrategy.None);
    }

    modelBuilder.Entity<InputMethod>().HasKey(e => e.RecoId);
    modelBuilder.Entity<InputMethod>()
      .Property(k => k.RecoId)
      .HasValueGenerator<MyValueGenerator>()
      .ValueGeneratedOnAdd();
    modelBuilder.Entity<InputMethod>().ToTable("tb_ims")
      .Property(x => x.MethodDisplay)
      .IsRequired()
      .HasMaxLength(12);
}ValueGeneratedOnAdd 方法表示在记录插入数据库时自动生成值,HasValueGenerator 方法设置你自定义的生成器。
现在,有了自定义生成规则,在插入数据时,主键不能赋值。一旦赋值,生成器就无效了。
// 尝试插入两条记录
InputMethod[] ents = [
      new(){ MethodDisplay = "双拼输入", Description="按两个键完成一个音节",Culture="zh-CN"},
      new() { MethodDisplay = "六指输入", Description="专供六个指头的人使用",Culture="zh-CN"}
    ];
dbc.Set<InputMethod>().AddRange(ents);
int result = dbc.SaveChanges();运行应用程序,你会发现,这次生成的 CREATE TABLE 语句中,RecoId 列已经没有 IDENTITY 关键字了。
info: 2024/8/4 18:41:24.956 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (12ms) , CommandType='Text', CommandTimeout='30']
      SELECT 1
info: 2024/8/4 18:41:24.982 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (4ms) , CommandType='Text', CommandTimeout='60']
      IF SERVERPROPERTY('EngineEdition') <> 5
      BEGIN
          ALTER DATABASE SET SINGLE_USER WITH ROLLBACK IMMEDIATE;
      END;
info: 2024/8/4 18:41:25.003 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (21ms) , CommandType='Text', CommandTimeout='60']
      DROP DATABASE ;
info: 2024/8/4 18:41:25.104 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (82ms) , CommandType='Text', CommandTimeout='60']
      CREATE DATABASE ;
info: 2024/8/4 18:41:25.137 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (32ms) , CommandType='Text', CommandTimeout='60']
      IF SERVERPROPERTY('EngineEdition') <> 5
      BEGIN
          ALTER DATABASE SET READ_COMMITTED_SNAPSHOT ON;
      END;
info: 2024/8/4 18:41:25.142 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (0ms) , CommandType='Text', CommandTimeout='30']
      SELECT 1
info: 2024/8/4 18:41:25.194 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (6ms) , CommandType='Text', CommandTimeout='30']
      <em><strong>CREATE TABLE (
          </strong></em><em><strong>int NOT NULL,
          nvarchar(12</strong></em><em><strong>) NOT NULL,
          nvarchar(max) NULL,
          nvarchar(max) NULL,
          CONSTRAINT PRIMARY KEY ()
      );</strong></em>
info: 2024/8/4 18:41:25.408 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (24ms) , CommandType='Text', CommandTimeout='30']
      SELECT CASE
          WHEN EXISTS (
            SELECT 1
            FROM AS
            WHERE . = @__newID_0) THEN CAST(1 AS bit)
          ELSE CAST(0 AS bit)
      END
info: 2024/8/4 18:41:25.448 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (1ms) , CommandType='Text', CommandTimeout='30']
      SELECT CASE
          WHEN EXISTS (
            SELECT 1
            FROM AS
            WHERE . = @__newID_0) THEN CAST(1 AS bit)
          ELSE CAST(0 AS bit)
      END
info: 2024/8/4 18:41:25.488 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (2ms) , CommandType='Text', CommandTimeout='30']
      SET IMPLICIT_TRANSACTIONS OFF;
      SET NOCOUNT ON;
      INSERT INTO (, , , )
      VALUES (@p0, @p1, @p2, @p3),
      (@p4, @p5, @p6, @p7);
更新记录数:2
info: 2024/8/4 18:41:25.524 RelationalEventId.CommandExecuted (Microsoft.EntityFrameworkCore.Database.Command)
      Executed DbCommand (1ms) , CommandType='Text', CommandTimeout='30']
      SELECT ., ., ., .
      FROM AS
ID=427211935, Display=六指输入, Culture=zh-CN
ID=1993200136, Display=双拼输入, Culture=zh-CN怎么样,这玩法是不是很高端?当然,如果主键是字符串类型,你也可以生成字符串的值,一切看你需求,反正原理是相同的。
最后,咱们顺便聊聊如何自动更改日期时间的问题。这个在实际开发中也很常用,比如一个计划表,其实体如下:
public class Plan
{
    /// <summary>
    /// 计划ID
    /// </summary>
    public int ID { get; set; }
    /// <summary>
    /// 计划简述
    /// </summary>
    public string? PlanDesc { get; set; }
    /// <summary>
    /// 计划级别
    /// </summary>
    public int Level { get; set; }
    /// <summary>
    /// 计划创建时间
    /// </summary>
    public DateTime? CreateTime { get; set; }
    /// <summary>
    /// 总计划量
    /// </summary>
    public float TotalTask { get; set; }
    /// <summary>
    /// 完成量
    /// </summary>
    public float Completed { get; set; }
    /// <summary>
    /// 更新时间
    /// </summary>
    public DateTime? UpdateTime { get; set; }
}最后一个字段 UpdateTime 表示在插入后更新的时间,所以在插入时这个字段可以留 NULL。比如我修改计划完成数 Completed,在写入数据库时自动给 UpdateTime 字段赋当前时间。这个不能用值生成器来做,因为生成器只能在数据插入前或插入后产生一次值,后面更新数据时不会再生成新值,就做不到自动设置更新时间了。所以,这里咱们可以换个思路:重写 DbContext 类的 SaveChanges 方法,在命令发送到数据库之前找出哪些记录被修改过,然后设置 UpdateTime 属性,最后才发送 SQL 语句。这样也能达到自动记录更新时间的功能。
public class MyDBContext : DbContext
{
    ……

    public override int SaveChanges(bool acceptAllChangesOnSuccess)
    {
      var modifieds = from c in ChangeTracker.Entries()
                        where c.State == EntityState.Modified
                              && c.Entity is Plan
                        select c;
      foreach(var obj in modifieds)
      {
            obj.Property(nameof(Plan.UpdateTime)).CurrentValue = DateTime.Now;
      }
      return base.SaveChanges(acceptAllChangesOnSuccess);
    }
}Modified 表示实体被更改过的状态。修改属性值时,应赋值给 CurrentValue,它代表的是实体当前的值,不要改 OriginalValue 的值,它指的是从数据库中读到的值,多数情况下不用去改,除非你要把当前 DbContext 实例的数据复制到另一个 DbContext 实例。
这样当 Plan 对象被修改后,在提交前会自动设置更新时间。下面是测试代码:
// 创建上下文
using var ctx= new MyDBContext();
// 测试用,确定删除数据库
ctx.Database.EnsureDeleted();
// 确定创建数据库
ctx.Database.EnsureCreated();

// 创建三条记录
Plan p01 = new()
{
   PlanDesc = "装配电池",
   CreateTime = DateTime.Now,
   TotalTask = 100f,
   Completed = 0f,
};
Plan p02 = new Plan()
{
   PlanDesc = "更换底板",
   CreateTime = DateTime.Now,
   Level = 4,
   TotalTask = 12.0f,
   Completed = 0f
};
Plan p03 = new()
{
   PlanDesc = "清洗盖板",
   TotalTask = 20.5f,
   Completed = 0f,
   CreateTime = DateTime.Now
};
ctx.Plans.Add(p01);
ctx.Plans.Add(p02);
ctx.Plans.Add(p03);
// 更新到数据库
int n = ctx.SaveChanges();
Console.WriteLine($"已插入{n}条记录");

// 打印数据
Print(ctx.Plans);
MODIFY:   // 这是个标签
Console.Write("请输入要更新的记录ID:");
string? line = Console.ReadLine();
if(line == null)
{
   Console.WriteLine("你输入了吗?");
   goto MODIFY;    // 回到标签处
}
if(!int.TryParse(line, out int id))
{
   Console.WriteLine("你丫的输入的是整数吗?");
   goto MODIFY;    // 回到标签处
}
UPDATE:   // 标签
Console.Write("请输入计划完成数:");
line = Console.ReadLine();
if (line == null)
{
   Console.WriteLine("你确定你没敲错键盘?");
   goto UPDATE;
}
if(!float.TryParse(line, out float comp))
{
   Console.WriteLine("浮点数,浮点数,浮点数");
   goto UPDATE;
}
// 查找
Plan? curPlan = ctx.Plans.FirstOrDefault(x => x.ID == id);
if (curPlan == null)
{
   Console.WriteLine("找不到记录");
   goto MODIFY;
}
if(comp > curPlan.TotalTask)
{
   Console.WriteLine("你是在异空间工作吗?");
   goto UPDATE;
}
// 更新
curPlan.Completed = comp;
ctx.SaveChanges();

// 再次打印
Print(ctx.Plans);先插入三条数据,然后输入记录ID来修改 Completed 的值。更改后会看到更新时间。

 好了,今天咱们就水到这里了。
 

来源:https://www.cnblogs.com/tcjiaan/p/18340772
免责声明:由于采集信息均来自互联网,如果侵犯了您的权益,请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容,谢谢合作!
页: [1]
查看完整版本: 【EF Core】自动生成的字段值