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

Question

我有一个名为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();
}

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

谢谢大家。

Answer 1

您的上下文有一个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