EFCore重写SaveChanges给公共字段赋值

1 篇文章 0 订阅
1 篇文章 0 订阅

第一步,新建公共字段实体类EntityBase

    public class EntityBase
    {
        //主键id
        public Guid Id { get; set; }
        //创建时间
        public DateTime CreationTime { get; set; }
        //最后修改时间
        public DateTime LastModificationTime { get; set; }
 
    }

第二步,新建User实体类继承EntityBase

    /// <summary>
    /// 用户管理实体表
    /// </summary>
    [Table(nameof(User))]
    public class User : EntityBase
    {
        /// <summary>
        /// 用户名
        /// </summary>
        [StringLength(50)]
        public string? UserName { get; set; }

        public string? NickName { get; set; }

    }

第三步,在EF上下文类中重写SaveChanges方法

        public FaceContext(DbContextOptions<FaceContext> options) : base(options)
        {
        }
        public DbSet<User> User { get; set; }

        public override int SaveChanges()
        {
            SetSystemField();
            return base.SaveChanges();
        }

        public override Task<int> SaveChangesAsync(CancellationToken cancellationToken = default(CancellationToken))
        {
            SetSystemField();
            return base.SaveChangesAsync();
        }
 
        /// <summary>
        /// 系统字段赋值
        /// </summary>
        private void SetSystemField()
        {
            foreach (var item in ChangeTracker.Entries())
            {
                if (item.Entity is EntityBase)
                {
                    var entity = (EntityBase)item.Entity;
                    //添加操作
                    if (item.State == EntityState.Added)
                    {
                        if (entity.Id == Guid.Empty)
                        {
                            entity.Id = Guid.NewGuid();
                        }
                        entity.CreationTime = DateTime.Now;
                        entity.LastModificationTime = DateTime.Now;
                    }
                    //修改操作
                    else if (item.State == EntityState.Modified)
                    {
                        entity.LastModificationTime = DateTime.Now;
                    }
                }

            }
        }

再使用SaveChanges保存数据就可以不用管公共字段了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值