Autofac与AOP功能例子-白红宇

Autofac与AOP功能例子

阅读量：6848 次

发布时间：2019-06-26

本文共 4656 字，大约阅读时间需要 15 分钟。

using Autofac.Extras.DynamicProxy;using System;using System.Collections.Generic;using System.Linq;using System.Text;using System.Threading.Tasks;namespace aopTest{    [Intercept(typeof(CallLogger))]    public interface IProduct    {        string Title { get; set; }        decimal Price { get; set; }        string Show();    }    [Intercept(typeof(CallLogger))]    public class Apple : IProduct    {        public Apple()        {            Title = "苹果";            Price = 5.0m;        }        public string Title { get ; set ; }        public decimal Price { get ; set; }        public virtual string Show()        {            return $"{Title} - {Price}";        }    }    [Intercept(typeof(CallLogger))]    public class Book : IProduct    {        public Book()        {            Title = "Asp.net开发";            Price = 35.0m;        }        public string Title { get; set; }        public decimal Price { get; set; }                [Auth("product.show")]        public virtual string Show()        {            Test();            return $"{Title} - {Price}";        }        protected virtual void Test()        {            Console.WriteLine("is a test code....");        }    }}

using System;using System.Collections.Generic;using System.Linq;using System.Text;using System.Threading.Tasks;namespace aopTest{    [AttributeUsage(AttributeTargets.Method)]    public class AuthAttribute : Attribute    {        public AuthAttribute(string code)        {            Code = code;        }        public string Code { get; set; }    }}using System;using System.Collections.Generic;using System.Linq;using System.Text;using System.Threading;using System.Threading.Tasks;namespace aopTest{    public class UserInfo    {        static LocalDataStoreSlot storeSlot = Thread.AllocateNamedDataSlot("user-info");        public static void SetUser(string name)        {            Thread.SetData(storeSlot, name);        }        public static string GetUser()        {            return Thread.GetData(storeSlot).ToString();        }    }}

using System;using System.Collections.Generic;using System.Diagnostics;using System.IO;using System.Linq;using System.Text;using System.Threading.Tasks;using Castle.DynamicProxy;namespace aopTest{    public class CallLogger : IInterceptor    {        TextWriter _output;        public CallLogger(TextWriter output)        {            _output = output;        }        public void Intercept(IInvocation invocation)        {            Stopwatch stopWatch = new Stopwatch();            stopWatch.Start();            string m = invocation.TargetType.ToString() + "." + invocation.Method.Name;            _output.WriteLine($"方法:{m}调用开始");            _output.Write("Calling method {0} with parameters {1}... ",            invocation.Method.Name,            string.Join(", ", invocation.Arguments.Select(a => (a ?? "").ToString()).ToArray()));            var authAttribute = invocation.Method.GetCustomAttributes(typeof(AuthAttribute), false).FirstOrDefault();            if (authAttribute != null)            {                var authInfo = (AuthAttribute)authAttribute;                if (UserInfo.GetUser() != "zhangsan" && authInfo.Code == "product.show")                {                    _output.WriteLine($"当前用户{UserInfo.GetUser()}没有方法{m}的访问权限，需要权限{authInfo.Code}");                }            }            invocation.Proceed();            _output.WriteLine("Done: result was {0}.", invocation.ReturnValue);            stopWatch.Stop();            _output.WriteLine($"方法:{m}调用结束，总耗时{stopWatch.ElapsedMilliseconds}毫秒");        }    }}

using Autofac;using Autofac.Extras.DynamicProxy;using System;using System.Collections.Generic;using System.Linq;using System.Text;using System.Threading.Tasks;namespace aopTest{    class Program    {        static void Main(string[] args)        {            var builder = new ContainerBuilder();            /*builder.RegisterType
     
      ()                   .As
      
       ()                   .EnableInterfaceInterceptors();            builder.RegisterType
       
        ()                   .As
        
         ()                   .EnableInterfaceInterceptors();*/            builder.RegisterType
         
          ()                  .As
          
           () .EnableClassInterceptors(); builder.RegisterType
           
            () .As
            
             () .EnableClassInterceptors(); builder.Register(c => new CallLogger(Console.Out)); var container = builder.Build(); UserInfo.SetUser("zhaoliu"); var products = container.Resolve
             
              
               >(); products.ToList().ForEach(product => { product.Show(); }); Console.ReadKey(); } }}