EF Core 为 dbContext 传递 TEntity 以执行更新或添加

Question

使用 EF Core，我有 10 种方法基本上做同样的事情。更新或添加，然后 SaveChanges()

即

  public void UpdateUser(User model)
    {
        if(... some condition.. to determine to add or update)
        {
            db.Update(model);
            db.Entry(model).State = EntityState.Modified;
        }
        else
        {        
           db.User.Add(model);
        } 
        db.SaveChanges(); 
    }

  public void UpdateCompany(Company model)
       {
        if(... some condition.. to determine to add or update)
        {
            db.Update(model);
            db.Entry(model).State = EntityState.Modified;
        }
        else
        {        
           db.Company.Add(model);
        } 
        db.SaveChanges(); 
    }

是否可以创建一种方法并只传递要更新或添加的实体？

Answer 1

您可以使用通用方法。如果您需要访问实体的 DbSet，请使用 db.Set<TEntity>() 而不是声明的 DbSet 属性 db.Users 或 db.Comanies。

    public void Update<TEntity>(TEntity entity) where TEntity: class
    {

        if (1==1 /*whatever*/)
        {
            db.Update(entity);
        }
        else
        {
            db.Set<TEntity>().Add(entity);
        }
        db.SaveChanges();
    }

Answer 2

Update 和 Add 之间的真正区别是 PrimaryKey 值。当您尝试 Update 包含 null PrimaryKey 的数据时，它不会更新数据，它会添加该数据。你可以控制model 来了解它是否更新。如果PrimaryKey 为空，它可以添加该数据。如果PrimaryKey 不为空，你可以Update 数据：

    public void UpdateOrAdd(Company model)
    {
        if(model.CompanyID !=0) //The primary key
        {
            db.Update(model);
            db.Entry(model).State = EntityState.Modified;
        }
        else
        {        
           db.Company.Add(model);
        } 
        db.SaveChanges(); 
    }

EF 在db.SaveChanges(); 方法之后创建一个PrimaryKey 值。

我建议你最后尝试使用 db.SaveChanges(); 方法。您可以收到错误消息并通过 SweetAlert 向自己显示，您可以给出正确的消息是操作 %100 是否成功。