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();
}
我想远离反思。。。但是,我。看起来就是这样。
谢谢大家。
您的上下文有一个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