C#实体框架,获取插入ID(泛型)

本文关键字:ID 泛型 插入 获取 实体 框架 | 更新日期: 2023-09-27 18:21:16

我有一个名为AddFoo(Foo-Foo)的方法

private int AddFoo(Foo foo)
{
  Using (BarContext db = new BarContext())
  {
    db.Foos.Add(foo);
    db.SaveChanges();
  }
  return foo.FooID;
}

我想让它更通用,接受任何实体并返回一个ID示例(粗略且可能不起作用,只是为了展示这个想法:

private int Add(T entity)
{
  Using (BarContect db = new BarContext())
  {
    // need to figure out how to find the appropriate set
    DbSet dbSet = //Set Appropriate dbset based on entity type
    dbSet.Add(entity);
    db.SaveChanges(); 
  }
  return entity.PrimaryKeyValue; // this would be the integer value of the id
}

现在,我可以使用反射来找到一个用[Key]标记的属性,以确定实体类中的哪个属性拥有ID,但我认为这不是最有效的。我也可以硬编码一些映射方法,使我能够确定要添加到什么DBSet……但我无法想象,没有什么东西已经以更有效的方式完成了这两项操作。

所以。。。我如何确定Key及其值,以及我如何确定在这种通用的东西中使用什么DbSet?

更新

根据下面的答案和其他类似的帖子,这就是我最终所做的。。。(几乎只是结合了两个答案)

private static int GetEntityKeyValue<T>(this T entity) where T : class
{
  int ret = 0;
  PropertyInfo key = typeof(T).GetProperties().FirstOrDefault(p => p.GetCustomAttributes(typeof(KeyAttribute), true).Length != 0);
  if (key != null)
  {
    ret = (int)key.GetValue(entity, null);
  }
  return ret;
}
private static int Add<T>(T entity) where T : class
{
  using (Foo db = new FooContext())
  {
    DbSet dbset = db.Set<T>();
    dbset.Add(entity);
    db.SaveChanges();
  }
  return entity.GetEntityKeyValue();
}

我想远离反思。。。但是,我。看起来就是这样。

谢谢大家。

C#实体框架,获取插入ID(泛型)

您的上下文有一个Set<T>()方法,该方法将返回正确类型的DBSet。

这将使你的方法

private int Add(T entity)
{
  Using (BarContect db = new BarContext())
  {
    DbSet dbSet = db.Set<T>();
    dbSet.Add(entity);
    db.SaveChanges(); 
  }
  return entity.PrimaryKeyValue; // this would be the integer value of the id
}

这是Set 的MSDN

http://msdn.microsoft.com/en-us/library/gg696521(v=vs.103).aspx