通用工作单元

本文关键字：单元 工作 | 更新日期: 2023-09-27 18:05:44

我已经实现了EntityFramework模式以及Repository和Unit Of Work。实现类似于代码项目存储库示例，但是我需要对工作单元进行增强。

工作单位

public class GenericUnitOfWork : IDisposable
{
    // Initialization code
    public Dictionary<Type, object> repositories = new Dictionary<Type, object>();
    public IRepository<T> Repository<T>() where T : class
    {
        if (repositories.Keys.Contains(typeof(T)) == true)
        {
            return repositories[typeof(T)] as IRepository<T>
        }
        IRepository<T> repo = new Repository<T>(entities);
        repositories.Add(typeof(T), repo);
        return repo;
    }
    // other methods
}

上面的UoW是非常一般化的，它总是针对父Repository类。我有另一个实体，例如student，它有自己的存储库扩展repository类。特定于学生的存储库有一个方法"GetStudentMarks()"。现在我不能使用一般的Unit Of Work类，因为它总是指向父资源库。

如何实现一个通用的工作单元来处理这种情况?

通用工作单元

Generic UnitOfWork !!您实现了错误的工作单元

参见此代码:

using System.Data.Entity;
using System;

namespace EF_Sample07.DataLayer.Context
{
 public interface IUnitOfWork
{
    IDbSet<TEntity> Set<TEntity>() where TEntity : class;
    int SaveChanges();
   }
 }

为什么使用UnitOfWork?

因为:

    <
  • 更好的性能/gh>
  • 正确使用事务

参见示例:

   public class Category
{
    public int Id { get; set; }
    public virtual string Name { get; set; }
    public virtual string Title { get; set; }

    public virtual ICollection<Product> Products { get; set; }
}
产品

    public class Product
{
    public int Id { get; set; }
    public string Name { get; set; }
    public decimal Price { get; set; }

    [ForeignKey("CategoryId")]
    public virtual Category Category { get; set; }
    public int CategoryId { get; set; }
}

UnitOfWork 

  public interface IUnitOfWork
{
    IDbSet<TEntity> Set<TEntity>() where TEntity : class;
    int SaveChanges();
}

DbContext 

    public class Sample07Context : DbContext, IUnitOfWork
{
    public DbSet<Category> Categories { set; get; }
    public DbSet<Product> Products { set; get; }

    #region IUnitOfWork Members
    public new IDbSet<TEntity> Set<TEntity>() where TEntity : class
    {
        return base.Set<TEntity>();
    }
    public int SaveAllChanges()
    {
        return base.SaveChanges();
    }
    #endregion
}

您可以创建类GenericUnitOfWork泛型，指定实体和存储库类型:

public class GenericUnitOfWork<TRepo, TEntity> : IDisposable
    where TRepo : Repository<TEntity>
{
    // Initialization code
    public Dictionary<Type, TRepo> repositories = new Dictionary<Type, TRepo>();
    public TRepo Repository()
    {
        if (repositories.Keys.Contains(typeof(TEntity)) == true)
        {
            return repositories[typeof(TEntity)];
        }
        TRepo repo = (TRepo)Activator.CreateInstance(
            typeof(TRepo),
            new object[] { /*put there parameters to pass*/ });
        repositories.Add(typeof(TEntity), repo);
        return repo;
    }
    // other methods
}

我想，你想在GetStudentMarks方法中对StudentMarks执行一些操作。否则，如果您的模型被正确映射，您可以使用关系数据加载方法之一来加载它们。否则，如果通用存储库适合您的大多数实体，但您需要为少数实体提供一些额外的方法，那么我建议为这些存储库创建扩展方法:

public static class StudentReposityExtensions
{
    public List<Mark> GetStudentMarks(
        this IRepository<Student> studentRepostitory)
    {
        .....
    }
}
相关文章：