.NET Core: attributes that execute before and after method

前端 未结 4 1967
余生分开走
余生分开走 2021-01-12 17:10

In Java, it is possible to use AspectJ for adding behavior before and after executing a method, using method annotations. Since C# Attributes seem to be very similar, I was

相关标签:
4条回答
  • 2021-01-12 17:27

    This can be accomplished using DynamicProxy.

    There is an implementation of a memory caching technique with logic that executes before the method being called. That can be extended to check for the existence of an attribute like this

    var attribute = Attribute.GetCustomAttribute(invocation.MethodInvocationTarget, typeof(CachedAttribute)) as CachedAttribute;
    if (attribute != null)
    {
      ...
    }
    

    The code above can be inside the Intercept method in the Interceptor implementation. CachedAttribute would be your attribute.

    0 讨论(0)
  • 2021-01-12 17:32

    You need some framework that is able to handle your attribute appropriately. Only because the attribute exists doesn´t mean it will have any affect.

    I wrote some easy engine that does that. It will determine if the attribute is present on the passed action and if so get the reflected methods in order to execute them.

    class Engine
    {
        public void Execute(Action action)
        {
            var attr = action.Method.GetCustomAttributes(typeof(MyAttribute), true).First() as MyAttribute;
            var method1 = action.Target.GetType().GetMethod(attr.PreAction);
            var method2 = action.Target.GetType().GetMethod(attr.PostAction);
    
            // now first invoke the pre-action method
            method1.Invoke(null, null);
            // the actual action
            action();
            // the post-action
            method2.Invoke(null, null);
        }
    }
    public class MyAttribute : Attribute
    {
        public string PreAction;
        public string PostAction;
    }
    

    Of course you need some null-ckecks, e.g. in the case the methods don´t exist or aren´t static.

    Now you have to decorate your action with the attribute:

    class MyClass
    {
        [MyAttribute(PreAction = "Handler1", PostAction = "Handler2")]
        public void DoSomething()
        {
            
        }
    
        public static void Handler1()
        {
            Console.WriteLine("Pre");
        }
        public static void Handler2()
        {
            Console.WriteLine("Post");
        }
    }
    

    Finally you can execute that method within our engine:

    var engine = new Engine();
    var m = new MyClass();
    engine.Execute(m.DoSomething);
    
    0 讨论(0)
  • 2021-01-12 17:34

    The question is similar to Run a method before all methods of a class, hence the same answer applies to both. Use https://github.com/Fody/Fody . The licencing model is based on voluntary contributions making it the better option to PostSharp which is a bit expensive for my taste.

    [module: Interceptor]
    namespace GenericLogging
    {
    
    [AttributeUsage(AttributeTargets.Method | AttributeTargets.Constructor | AttributeTargets.Assembly | AttributeTargets.Module)]
        public class InterceptorAttribute : Attribute, IMethodDecorator
        {
            // instance, method and args can be captured here and stored in attribute instance fields
            // for future usage in OnEntry/OnExit/OnException
            public void Init(object instance, MethodBase method, object[] args)
            {
                Console.WriteLine(string.Format("Init: {0} [{1}]", method.DeclaringType.FullName + "." + method.Name, args.Length));
            }
    
            public void OnEntry()
            {
                Console.WriteLine("OnEntry");
            }
    
            public void OnExit()
            {
                Console.WriteLine("OnExit");
            }
    
            public void OnException(Exception exception)
            {
                Console.WriteLine(string.Format("OnException: {0}: {1}", exception.GetType(), exception.Message));
            }
        }
    
        public class Sample
        {
            [Interceptor]
            public void Method(int test)
            {
                Console.WriteLine("Your Code");
            }
        }
    }
    
    [TestMethod]
    public void TestMethod2()
    {
        Sample t = new Sample();
        t.Method(1);
    }
    
    0 讨论(0)
  • 2021-01-12 17:39

    Just like with Java and AspectJ, you need separate AoP tooling to inject code like this in .NET.

    PostSharp is one such tool, probably the best known. I belive they have support for .NET core since version 5.

    0 讨论(0)
提交回复
热议问题