使用Castle做类的增强

2021-09-15 02:23:09

我们已经实现了用户注册功能，现在想增加日志记录功能。具体来讲就是在用户注册前后，分别输出一条日志。我们当然可以修改原有的业务代码。

现在换个角度来问两个问题：
1. 团队开发中，我们很可能根本拿不到源代码，那又怎么去增加这个功能呢？
2. 这次需求是增加日志，以后再增加其他需求（比如异常处理），是不是仍然要改业务类呢？

总结一下：
我们要在不修改原有类业务代码的前提下，去做类的增强。我们的设计要符合面向对象的原则：对扩展开放，对修改封闭！

都有哪些办法呢？我们尝试以下几种方法：

原有业务类

业务模型

namespace testAopByDecorator
{
    public class User
    {
        public string Name { get; set; }
        public int Id { get; set; }
    }
}

接口设计

namespace testAopByDecorator
{
    public interface IUserProcessor
    {
        void RegisterUser(User user);
    }
}

业务实现

using System;

namespace testAopByDecorator
{
    public class UserProcessor : IUserProcessor
    {
        public void RegisterUser(User user)
        {
            if (user == null)
            {
                return;
            }

            Console.WriteLine(string.Format("注册了一个用户{0}：{1}", user.Id, user.Name));
        }
    }
}

上层调用

using System;

namespace testAopByDecorator
{
    class Program
    {
        private static User user = new User { Id = 1, Name = "滇红" };

        static void Main(string[] args)
        {
            Register();
            Console.ReadKey();
        }

        private static void Register()
        {
            IUserProcessor processor = new UserProcessor();
            processor.RegisterUser(user);
        }
    }
}

使用Castle做类的增强

我们将使用第三方的Castle.Core来对原有的类做业务增强，首先使用NuGet安装。
使用Castle做类的增强
业务增强实现

using System;
using Castle.DynamicProxy;

namespace testAopByCastle
{
    public class UserProcessorCastle : IInterceptor
    {
        public void Intercept(IInvocation invocation)
        {
            User user = invocation.Arguments[0] as User;

            before(user);
            invocation.Proceed();
            after(user);
        }

        private void after(User user)
        {
            Console.WriteLine("注册用户后：" + user.Name);
        }

        private void before(User user)
        {
            Console.WriteLine("注册用户前：" + user.Name);
        }
    }
}

需要原有业务类的方法声明为虚方法：virtual

using System;

namespace testAopByCastle
{
    public class UserProcessor : IUserProcessor
    {
        public virtual void RegisterUser(User user)
        {
            if (user == null)
            {
                return;
            }

            Console.WriteLine(string.Format("注册了一个用户{0}：{1}", user.Id, user.Name));
        }
    }
}

上层调用

using Castle.DynamicProxy;
using System;

namespace testAopByCastle
{
    class Program
    {
        private static User user = new User { Id = 1, Name = "滇红" };

        static void Main(string[] args)
        {
            RegisterAndLog();
            Console.ReadKey();
        }

        private static void RegisterAndLog()
        {
            ProxyGenerator generator = new ProxyGenerator();
            UserProcessorCastle castle = new UserProcessorCastle();
            IUserProcessor proxy = generator.CreateClassProxy<UserProcessor>(castle);
            proxy.RegisterUser(user);
        }
    }
}

对比一下扩展前后的业务展现
使用Castle做类的增强
这种方式修改了原有业务类的方法，需要显示定义为virtual，不是很符合对修改关闭原则。不过我们要根据实际项目情况灵活把握。

上一篇：使用Autofac做类的增强


下一篇：使用Unity做类的增强