src/ShenNius.Share.Domain/Repository/TransactionAttribute.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

using AspectCore.DynamicProxy;
using Microsoft.Extensions.DependencyInjection;
using System;
using System.Threading.Tasks;
/*************************************
* 类名：TransactionInterceptorAttribute
* 作者：realyrare
* 邮箱：mhg215@yeah.net
* 时间：2021/3/30 16:27:18
*┌───────────────────────────────────┐　    
*│　   版权所有：神牛软件　　　　	     │
*└───────────────────────────────────┘
**************************************/

namespace ShenNius.Share.Domain.Repository
{
    /// <summary>
    /// AOP 事务
    /// </summary>
    public class TransactionAttribute : AbstractInterceptorAttribute
    {
        public async override Task Invoke(AspectContext context, AspectDelegate next)
        {
            // DbContext dbContext = new DbContext();
            var dbContext = context.ServiceProvider.GetService<DbContext>();
            try
            {
                dbContext.Db.BeginTran();

                var result = await RunAndGetReturn(context, next);
                if (result is Task)
                {
                    Task.WaitAll(result as Task);
                }
                dbContext.Db.CommitTran();
            }
            catch (AggregateException ex)
            {
                dbContext.Db.RollbackTran();
                throw ex;
            }
            catch (Exception)
            {
                dbContext.Db.RollbackTran();
                throw;
            }
        }
        /// <summary>
        /// 执行被拦截方法
        /// </summary>
        /// <param name="context"></param>
        /// <param name="next"></param>
        /// <returns></returns>
        private async Task<object> RunAndGetReturn(AspectContext context, AspectDelegate next)
        {
            await context.Invoke(next);
            return context.IsAsync()
            ? await context.UnwrapAsyncReturnValue()
            : context.ReturnValue;
        }