项目架构开发：数据访问层之UnitOfWork （补充）

应lisansi同学回复（项目架构开发：数据访问层之UnitOfWork）要求，补上Dapper的DbContext实现

using Dapper.Contrib.Extensions;
using HG.Infrastructure.Common;
using System;
using System.Collections.Generic;
using System.Data;
using System.Transactions;

namespace HG.Data.Dapper
{
    public class DbContext
    {
        public IDbConnection Conn { get; set; }

        private Dictionary<object, Action> addEntities;
        private Dictionary<object, Action> updateEntities;
        private Dictionary<object, Action> deleteEntities;

        public DbContext()
        {
            Conn = DbConnectionFactory.CreateConnection(Enums.Core);

            addEntities = new Dictionary<object, Action>();
            updateEntities = new Dictionary<object, Action>();
            deleteEntities = new Dictionary<object, Action>();
        }

        public void Add<TEntity>(TEntity entity) where TEntity : class
        {
            this.addEntities.Add(entity, () =>
            {
                Conn.Insert<TEntity>(entity);
            });
        }

        public void Update<TEntity>(TEntity entity) where TEntity : class
        {
            this.updateEntities.Add(entity, () =>
            {
                Conn.Update(entity);
            });
        }

        public void Delete<TEntity>(TEntity entity) where TEntity : class
        {
            this.deleteEntities.Add(entity, () =>
            {
                Conn.Delete(entity);
            });
        }

        public TEntity Get<TEntity>(int id) where TEntity : class
        {
            return Conn.Get<TEntity>(id);
        }

        public int SaveChanges()
        {
            int count = 0;
            using (TransactionScope scope = new TransactionScope())
            {
                foreach (var entity in deleteEntities.Keys)
                {
                    this.deleteEntities[entity]();
                }

                foreach (var entity in updateEntities.Keys)
                {
                    this.updateEntities[entity]();
                }

                foreach (var entity in addEntities.Keys)
                {
                    this.addEntities[entity]();
                }

                scope.Complete();

                count = deleteEntities.Count + updateEntities.Count + addEntities.Count;

                deleteEntities.Clear();
                updateEntities.Clear();
                addEntities.Clear();

                if (Conn != null)
                {
                    if (Conn.State == ConnectionState.Open)
                    {
                        Conn.Close();
                    }

                   // Conn.Dispose();
                }
            }

            return count;
        }
    }
}

 

UnitOfWork相应改成如下

using HG.Data.Interface;
using System;
using System.Data;

namespace HG.Data.Dapper
{
    public class UnitOfWork : DbContext, IUnitOfWork, ICRUDUnitOfWork
    {
        public void RegisterNew<TEntity>(TEntity entity)
            where TEntity : class
        {
            base.Add<TEntity>(entity);
        }

        public void RegisterDirty<TEntity>(TEntity entity)
            where TEntity : class
        {
            base.Update<TEntity>(entity);
        }

        public void RegisterDeleted<TEntity>(TEntity entity)
            where TEntity : class
        {
            base.Delete<TEntity>(entity);
        }

        public void RegisterClean<TEntity>(TEntity entity)
            where TEntity : class
        { }

        //public TEntity Get<TEntity>(int id) 
        //    where TEntity : class
        //{
        //    return base.Get<TEntity>(id);
        //}

        public bool Commit()
        {
            return base.SaveChanges() > 0;
        }

        public void Rollback()
        {
            throw new NotImplementedException();
        }
    }
}