天天看點

C# EntityFrameworkCore自動修改CreateTime,UpdateTime代碼

其他資料庫Model繼承BaseDbModel
using System;
using System.ComponentModel.DataAnnotations;
using Microsoft.EntityFrameworkCore;
using Newtonsoft.Json;

namespace Demo.Model.Base
{	
    public class BaseDbModel
    {
        [Key] public int Id { get; set; }
        public DateTime? CreateTime { get; set; }
        public DateTime? UpdateTime { get; set; }
        [JsonIgnore] public bool IsDeleted { get; set; }
    }
}

           

代碼

using System;
using System.Linq;
using Demo.Model;
using Demo.Model.Base;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;

namespace Demo.EFCore
{
    public class DefaultDbContext : DbContext
    {
        public DbSet<FrameWorkInfo> FrameWorkInfos { get; set; }
        public DbSet<Message> Messages { get; set; }

        protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
        {
            optionsBuilder.UseSqlite("Data Source=./Db/Sqlite.db");
            //列印sql日志
            optionsBuilder.LogTo((s) => { Console.WriteLine(s); }
                , LogLevel.Information);
        }
        
		// 重寫 SaveChanges
        public override int SaveChanges()
        {
            //自動修改 CreateTime,UpdateTime
            var entityEntries = ChangeTracker.Entries().ToList();
            foreach (var entry in entityEntries)
            {
                if (entry.State == EntityState.Added)
                    Entry(entry.Entity).Property(nameof(BaseDbModel.CreateTime)).CurrentValue = DateTime.Now;

                if (entry.State == EntityState.Modified)
                    Entry(entry.Entity).Property(nameof(BaseDbModel.UpdateTime)).CurrentValue = DateTime.Now;
            }

            return base.SaveChanges();
        }

        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
        	// 種子資料
            modelBuilder.Entity<FrameWorkInfo>()
                .HasData(
                    new FrameWorkInfo
                    {
                        Id = 1, Name = "React", Url = "https://github.com/facebook/react", CreateTime = DateTime.Now
                    },
                    new FrameWorkInfo
                    {
                        Id = 2, Name = "Node", Url = "https://github.com/nodejs/node", CreateTime = DateTime.Now
                    }
                );
        }
    }
}