掀起你的盖头来：Unit Of Work-工作单元

 1 namespace ByteartRetail.Domain
 2 {
 3     /// <summary>
 4     /// 表示所有集成于该接口的类型都是Unit Of Work的一种实现。
 5     /// </summary>
 6     /// <remarks>有关Unit Of Work的详细信息，请参见UnitOfWork模式：http://martinfowler.com/eaaCatalog/unitOfWork.html。
 7     /// </remarks>
 8     public interface IUnitOfWork
 9     {
10         /// <summary>
11         /// 获得一个<see cref="System.Boolean"/>值，该值表述了当前的Unit Of Work事务是否已被提交。
12         /// </summary>
13         bool Committed { get; }
14         /// <summary>
15         /// 提交当前的Unit Of Work事务。
16         /// </summary>
17         void Commit();
18         /// <summary>
19         /// 回滚当前的Unit Of Work事务。
20         /// </summary>
21         void Rollback();
22     }
23 }

 1     /// <summary>
 2     /// Represents that the implemented classes are repository contexts.
 3     /// </summary>
 4     public interface IRepositoryContext : IUnitOfWork, IDisposable
 5     {
 6         /// <summary>
 7         /// Gets the unique-identifier of the repository context.
 8         /// </summary>
 9         Guid ID { get; }
10         /// <summary>
11         /// Registers a new object to the repository context.
12         /// </summary>
13         /// <typeparam name="TAggregateRoot">The type of the aggregate root.</typeparam>
14         /// <param name="obj">The object to be registered.</param>
15         void RegisterNew<TAggregateRoot>(TAggregateRoot obj)
16             where TAggregateRoot : class, IAggregateRoot;
17         /// <summary>
18         /// Registers a modified object to the repository context.
19         /// </summary>
20         /// <typeparam name="TAggregateRoot">The type of the aggregate root.</typeparam>
21         /// <param name="obj">The object to be registered.</param>
22         void RegisterModified<TAggregateRoot>(TAggregateRoot obj)
23             where TAggregateRoot : class, IAggregateRoot;
24         /// <summary>
25         /// Registers a deleted object to the repository context.
26         /// </summary>
27         /// <typeparam name="TAggregateRoot">The type of the aggregate root.</typeparam>
28         /// <param name="obj">The object to be registered.</param>
29         void RegisterDeleted<TAggregateRoot>(TAggregateRoot obj)
30             where TAggregateRoot : class, IAggregateRoot;
31     }

 1 using System.Data.Entity;
 2 using System.Threading;
 3 
 4 namespace ByteartRetail.Domain.Repositories.EntityFramework
 5 {
 6     public class EntityFrameworkRepositoryContext : RepositoryContext, IEntityFrameworkRepositoryContext
 7     {
 8         private readonly ThreadLocal<ByteartRetailDbContext> localCtx = new ThreadLocal<ByteartRetailDbContext>(() => new ByteartRetailDbContext());
 9 
10         public override void RegisterDeleted<TAggregateRoot>(TAggregateRoot obj)
11         {
12             localCtx.Value.Entry<TAggregateRoot>(obj).State = System.Data.EntityState.Deleted;
13             Committed = false;
14         }
15 
16         public override void RegisterModified<TAggregateRoot>(TAggregateRoot obj)
17         {
18             localCtx.Value.Entry<TAggregateRoot>(obj).State = System.Data.EntityState.Modified;
19             Committed = false;
20         }
21 
22         public override void RegisterNew<TAggregateRoot>(TAggregateRoot obj)
23         {
24             localCtx.Value.Entry<TAggregateRoot>(obj).State = System.Data.EntityState.Added;
25             Committed = false;
26         }
27 
28         public override void Commit()
29         {
30             if (!Committed)
31             {
32                 localCtx.Value.SaveChanges();
33                 Committed = true;
34             }
35         }
36 
37         public override void Rollback()
38         {
39             Committed = false;
40         }
41 
42         protected override void Dispose(bool disposing)
43         {
44             if (disposing)
45             {
46                 if (!Committed)
47                     Commit();
48                 localCtx.Value.Dispose();
49                 localCtx.Dispose();
50                 base.Dispose(disposing);
51             }
52         }
53 
54         #region IEntityFrameworkRepositoryContext Members
55 
56         public DbContext Context
57         {
58             get { return localCtx.Value; }
59         }
60 
61         #endregion
62     }
63 }