Asp.netCore 3.1控制器屬性注入and非同步事務Aop by AutoFac

Aspect Oriented Programming（AOP）是較為熱門的一個話題。AOP，國內我們都習慣稱之為：面向切面程式設計

下面直接code 乾貨展示：

1:匯入相關的包：AutoFac 預設最新穩定版即可

Autofac.Extensions.DependencyInjection Version="7.1.0" 
Autofac.Extras.DynamicProxy Version="6.0.0"

 

2：AutoFac注入的擴充套件方法

using System;
using System.Linq;
namespace ZRF.CRM.Commoms
{
    using Autofac;
    using Autofac.Extras.DynamicProxy;
    using Microsoft.AspNetCore.Mvc;
    using System.Reflection;
    using ZRF.CRM.Commoms.Interceptors;
    using ZRF.CRM.MyAttrobutrs;

    public static class ContainerBuilderExtenTion
    {
        public static void DependInjecterIoc(this ContainerBuilder builder)
        {
            builder.RegisterType<TransactionScopeAsyncInterCeptor>();
            builder.RegisterAssemblyTypes(Assembly.Load("ZRF.CRM.Service"))
                .AsImplementedInterfaces().PropertiesAutowired().InterceptedBy(typeof(TransactionScopeAsyncInterCeptor)).EnableInterfaceInterceptors().OnRegistered(t =>
                {
                    Console.WriteLine("ZRF.CRM.Service===Register_ok");
                });
        }
    }
}

3：新增ConfigureContainer(ContainerBuilder builder)方法，並在Startup方法中註冊

using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Threading.Tasks;
using Autofac;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.Controllers;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.DependencyInjection.Extensions;
using Microsoft.Extensions.Hosting;
using WebApplication1.IOCS;
using ZRF.CRM.Commoms;
namespace WebApplication1
{
    public class Startup
    {
        public Startup(IConfiguration configuration)
        {
            Configuration = configuration;
        }
        public void ConfigureContainer(ContainerBuilder builder)
        {
            builder.DependInjecterIoc();

            //控制器屬性注入
            builder.RegisterModule<ControllerModule>();
        }
        public IConfiguration Configuration { get; }

        // This method gets called by the runtime. Use this method to add services to the container.
        public void ConfigureServices(IServiceCollection services)
        {
            services.AddControllersWithViews();

            //core 3.0級以上
            services.Replace(ServiceDescriptor.Transient<IControllerActivator, ServiceBasedControllerActivator>());
        }

        // This method gets called by the runtime. Use this method to configure the HTTP request pipeline.
        public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
        {
            if (env.IsDevelopment())
            {
                app.UseDeveloperExceptionPage();
            }
            else
            {
                app.UseExceptionHandler("/Home/Error");
            }
            app.UseStaticFiles();

            app.UseRouting();

            app.UseAuthorization();

            app.UseEndpoints(endpoints =>
            {
                endpoints.MapControllerRoute(
                    name: "default",
                    pattern: "{controller=Home}/{action=Index}/{id?}");
            });
        }
    }
}

4：Controller

using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Logging;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Threading.Tasks;
using WebApplication1.Models;
using ZRF.CRM.MyAttrobutrs;
using ZRF.CRM.InterFaces;
namespace WebApplication1.Controllers
{
    [ControllerCanInjectPropertity]
    public class HomeController : Controller
    {
        private readonly ILogger<HomeController> _logger;

        IDoLog _dolog;
        ITest002Service _service;
        public IDoLog _dolog2 { get; set; }

        public HomeController(ILogger<HomeController> logger, IDoLog dolog, ITest002Service service)
        {
            _logger = logger;
            _dolog = dolog;
            this._service = service;
        }

        public IActionResult Index()
        {
            var boolflag = _dolog2.LogWork("aaaa");
            bool flag = _dolog.LogWork("qqai");
            var getMsg = _service.Dowork("有aop的方法");
            Console.WriteLine("Aop獲取到方法：" + getMsg);
            //_service.DoWork02("沒有aop的普通方法");
            return View();
        }

        public IActionResult Privacy()
        {
            return View();
        }

        [ResponseCache(Duration = 0, Location = ResponseCacheLocation.None, NoStore = true)]
        public IActionResult Error()
        {
            return View(new ErrorViewModel { RequestId = Activity.Current?.Id ?? HttpContext.TraceIdentifier });
        }
    }
}

5:InterFace 

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

namespace ZRF.CRM.InterFaces
{
    
   public interface ITest002Service
    {
        string Dowork(string msg);
        string DoWork02(string msg);
    }
}

6：Service

using System;
using System.Collections.Generic;
using System.Text;
namespace ZRF.CRM.Service
{
    using ZRF.CRM.InterFaces;
    using ZRF.CRM.MyAttrobutrs;

    public class Test002Service : ITest002Service
    {
        [TransactionScopeAsync]
        public string Dowork(string msg)
        {

            return $"Test002Service:Dowork {DateTime.Now}-{msg}";
        }

        public string DoWork02(string msg)
        {
            return $"Test002Service:DoWork02:{DateTime.Now}-{msg}";
        }
    }
}

7：自定義特性，將來判斷哪些方法可以aop

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

namespace ZRF.CRM.MyAttrobutrs
{
    [AttributeUsage(AttributeTargets.Method)]
    public class TransactionScopeAsyncAttribute : Attribute
    {
    }

}

8:自定義類來實現Aop的IInterceptor方法

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

namespace ZRF.CRM.Commoms.Interceptors
{
    using Autofac;
    using Castle.DynamicProxy;
    using System.Reflection;
    using System.Transactions;
    using ZRF.CRM.MyAttrobutrs;

    public class TransactionScopeAsyncInterCeptor : IInterceptor
    {
        public void Intercept(IInvocation invocation)
        {

            if (HaveAsyncTrascAttribute(invocation))
            {
                TransactionScope scope = null;
                try
                {
                    scope = new TransactionScope(TransactionScopeAsyncFlowOption.Enabled);
                    Console.WriteLine("TransactionScopeAsyncFlowOption之前=======");
                    invocation.Proceed();
                    scope.Complete();
                    Console.WriteLine("invocation.Proceed()使用者自己的邏輯處理ok=======");
                    Console.WriteLine("TransactionScopeAsyncFlowOption之後=======");
                }
                catch (Exception)
                {
                    Console.WriteLine("TransactionScopeAsyncInterCeptor：發生異常");
                    if (scope != null)
                    {
                        scope.Dispose();
                    }
                }
            }
            else {
                Console.WriteLine("沒有非同步事務發生！");
                invocation.Proceed();
                return;
            }
        }

        private bool HaveAsyncTrascAttribute(IInvocation invocation)
        {
            var methodInfo = invocation.MethodInvocationTarget ?? invocation.Method;
            if (methodInfo.GetCustomAttribute<TransactionScopeAsyncAttribute>() != null)
            {
                return true;
            }
            else { return false; }
        }
    }
}

9：來一張測試的截圖

 

10：最後歡迎留言指教