Castle.DynamicProxy实现事务单元控制

using System;
using System.Collections.Generic;
using System.Text;

namespace Initial.Extensions
{
    /// <summary>
    /// 工作单元
    /// 仅用来做特性标记 
    /// </summary>
    public class UnitOfWorkAttribute : Attribute
    {

    }
}

using Castle.DynamicProxy;
using Microsoft.AspNetCore.Http;
using DataCommon.UnitOfWork;
using System;
using System.Diagnostics;
using System.Linq;
using System.Reflection;
using System.Threading;
using System.Threading.Tasks;

namespace Initial
{
    public class UnitOfWorkIInterceptor : IInterceptor
    {
        #region 必须的
        private readonly ILogger _logger;
        private readonly IUnitOfWork _unitOfWork;
        private readonly IHttpContextAccessor _httpContextAccessor;
        private readonly HttpContext _httpContext;
        #endregion

        public UnitOfWorkIInterceptor(
             ILogger logger,
             IUnitOfWork unitOfWork,
             IHttpContextAccessor httpContextAccessor
            )
        {
            _logger = logger;
            _unitOfWork = unitOfWork;
            _httpContextAccessor = httpContextAccessor;
            _httpContext = httpContextAccessor.HttpContext;
        }
        public void Intercept(IInvocation invocation)
        {
            MethodInfo methodInfo = invocation.MethodInvocationTarget;
            if (methodInfo == null)
                methodInfo = invocation.Method;

            UnitOfWorkAttribute transaction = methodInfo.GetCustomAttributes<UnitOfWorkAttribute>(true).FirstOrDefault();
            //如果标记了 [UnitOfWork]，并且不在事务嵌套中。
            if (transaction != null)
            {
                try
                {
                    //开启事务
                    _unitOfWork.BeginTransaction();
                    invocation.Proceed();
                    //提交事务
                    var result = ((Task<bool>)invocation.ReturnValue).Result;
                    _unitOfWork.Commit();
                }
                catch (Exception ex)
                {
                    //回滚
                    _unitOfWork.Rollback();
                    _logger.Info($"[{(_httpContextAccessor.HttpContext ?? _httpContext)?.TraceIdentifier}]" +
                        $"_{methodInfo.Name}执行异常{ex.ToString()}");
                    throw new BusinessException("提交异常");
                }
            }
            else
            {
                //如果没有标记[UnitOfWork]，直接执行方法
                invocation.Proceed();
            }
        }
    }
}

using Castle.Core;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace Repository
{
    [Interceptor(typeof(UnitOfWorkIInterceptor))]
    public class TemplateRepository : ITemplateRepository
    {
        #region 必须的
        private readonly ILogger _logger;
        private readonly IDBInstance _IDBInstance;
        #endregion

        public TemplateRepository(
           ILogger logger,
           IDBInstance IDBInstance
           )
        {
            _logger = logger;
            _IDBInstance = IDBInstance;

        }

        #region 单表增删改查规范
        #endregion
     
 
 
        #region 多表操作

        [UnitOfWork]
        public async Task<bool> AddTepWithLogAsync(MSE_RPT_TEMPLATE mSE_RPT_TEMPLATE,
            MSE_RPT_TEMPLATE_LOG mSE_RPT_TEMPLATE_LOG)
        {
            await _IDBInstance.DBOperation.InsertAsync(mSE_RPT_TEMPLATE);
            await _IDBInstance.DBOperation.InsertAsync(mSE_RPT_TEMPLATE_LOG);
            return true;
       }

        [UnitOfWork]
        public async Task<bool> UpdateTepWithLogAsync(MSE_RPT_TEMPLATE mSE_RPT_TEMPLATE,
            MSE_RPT_TEMPLATE_LOG mSE_RPT_TEMPLATE_LOG)
        {
            await _IDBInstance.DBOperation.UpdateAsync(mSE_RPT_TEMPLATE);
            await _IDBInstance.DBOperation.InsertAsync(mSE_RPT_TEMPLATE_LOG);
            return true;
        }

        #endregion

    }
}

1、第一个代码块主要是为对需要事务的方法进行标注

2、第二个代码块主要是对事务进行控制的；