Repository.cs

public interface IDataRepository<TEntity> : IDisposable where TEntity : class
{
    ICollection<TEntity> Find(
        Expression<Func<TEntity, bool>> criteria,
        Func<IQueryable<TEntity>, IOrderedQueryable<TEntity>> orderBy = null,
        string includeProperties = "");
    Task<ICollection<TEntity>> FindAsync(
        Expression<Func<TEntity, bool>> criteria,
        Func<IQueryable<TEntity>, IOrderedQueryable<TEntity>> orderBy = null,
        string includeProperties = "");
    Task<ICollection<TEntity>> FindAsync(
        CancellationToken cancellationToken,
        Expression<Func<TEntity, bool>> criteria,
        Func<IQueryable<TEntity>, IOrderedQueryable<TEntity>> orderBy = null,
        string includeProperties = "");

    TEntity FindById(params object[] keyValues);
    Task<TEntity> FindByIdAsync(params object[] keyValues);
    Task<TEntity> FindByIdAsync(CancellationToken cancellationToken, params object[] keyValues);

    void Delete(TEntity entity);
    bool Delete(params object[] keyValues);
    Task<bool> DeleteAsync(params object[] keyValues);
    Task<bool> DeleteAsync(CancellationToken cancellationToken, params object[] keyValues);
    void Insert(TEntity entity);
    void Update(TEntity entity);
}

UnitOfWork.cs

public interface IUnitOfWork
{
    IDataRepository<TEntity> Repository<TEntity>() where TEntity : class;

    int Save();
    Task<int> SaveAsync();
    Task<int> SaveAsync(CancellationToken cancellationToken);

    void BeginTransaction(IsolationLevel isolationLevel = IsolationLevel.Unspecified);
    bool Commit();
    void Rollback();
}