将实体插入通用存储库后如何返回实体

Question

我正在使用实体框架并编写了一个通用存储库来实现 UnitOfWork 模式。

我需要返回我插入的实体，并带有分配的 ID。

public class GenericRepository<TEntity> where TEntity : class
{
    internal AppDb Context;
    internal DbSet<TEntity> DbSet;

    public GenericRepository(AppDb context)
    {
        this.Context = context;
        this.DbSet = context.Set<TEntity>();
    }

    // ...

    public virtual void Insert(TEntity entity)
    {
         DbSet.Add(entity);
    }

    // ...
}

Answer 1

你是说这个吗？

public virtual TEntity Insert(TEntity entity)
{
     return DbSet.Add(entity);
}

DbSet<T>.Add() returns T，所以只需更改返回类型并键入return。