学习之路之三十：利用消息机制自定义数据库回滚特性

前段时间看到了几篇关于Attribute相关的文章，里面讲的很详细，参考：Attribute在.NET编程中的应用（五）

不过让我收获最多的关于“.NET Framework拦截机制的内容”，而且这也让我解决了一年前提出的问题！

地址：急丶急丶急 → 关于特性和IOC或者AOP的结合，其实跟IOC和AOP一点关于都没有，只是当时知识积累还不够，对解决问题的能力还不足！

下面我就结合一些理论知识以及对上面提出的案例进行改装，使它能够运用一般的方法之中(自定义一个数据库回滚特性)！

1.定义一个RollBack的特性

    [AttributeUsage(AttributeTargets.Method)]
    public class RollBackAttribute : Attribute
    {
        private bool _isCommit = true;

        public RollBackAttribute()
        { }

        public bool IsCommit //定义一个属性提示是否要提交数据
        {
            get { return _isCommit; }
            set { _isCommit = value; }
        }
    }

2.定义一个消息接收器

    public class RollBackMessageSink : IMessageSink
    {
        //定义下一个消息接收器实例
        private IMessageSink _nextSink;

        public RollBackMessageSink(IMessageSink messageSink) //通过构造函数来传递下一个消息接收器
        {
            _nextSink = messageSink;
        }

        //异步执行消息，一般情况下不需要写任何逻辑
        public IMessageCtrl AsyncProcessMessage(IMessage msg, IMessageSink replySink)
        {
            throw new NotImplementedException();
        }

        public IMessageSink NextSink
        {
            get
            {
                return _nextSink;
            }
        }

        //同步执行消息
        public IMessage SyncProcessMessage(IMessage msg)
        {
            //这边就是最重要的代码了
                //有一点要说明的：如果调用[_nextSink.SyncProcessMessage(msg)]就说明方法已经执行好了
                //这边有一个前处理和后处理的概念
                //前处理：也就是调用方法前的处理事件
                //后处理：也就是方法调用结束后的处理事件
                //前处理和后处理的标记：就是是否调用了[_nextSink.SyncProcessMessage(msg)],调用之前就是前处理，反之是后处理
            RollBackAttribute rollBack = GetRollBackAttribute(msg);
            if (rollBack == null) return _nextSink.SyncProcessMessage(msg);
            if (rollBack.IsCommit)
            {
                using (TransactionScope scope = new TransactionScope())
                {
                    return _nextSink.SyncProcessMessage(msg);
                }
            }
            return _nextSink.SyncProcessMessage(msg);
        }

        private RollBackAttribute GetRollBackAttribute(IMessage msg)
        {
            //特别要注意["__TypeName"]，其实这边是两个下划线，这个一定要记，可以通过调试来查看
 81             string typeName = (string)msg.Properties["__TypeName"];
            string methodName = (string)msg.Properties["__MethodName"];

            Console.WriteLine(typeName);
            Console.WriteLine(methodName);

            //根据类型和方法名获取自定义特性
            object[] customAttributes = Type.GetType(typeName).GetMethod(methodName).GetCustomAttributes(true);
            if (customAttributes != null && customAttributes.Length > 0)
            {
                RollBackAttribute rollback = customAttributes[0] as RollBackAttribute;
                if (rollback != null)
                {
                    //Console.WriteLine("是个对象啦");
                    return rollback;
                }
            }
            return null;
        }
    }

 3.下面来定义上下文属性，只要把消息接收器放到上下文中才行

    //必须要实现下面两个接口的
    //接下来我们定义上下文环境的属性，上下文环境属性必须根据你要创建的接收器类型来实现相应的接口，
    //比如：如果创建的是服务器上下文环境接收器，那么必须实现IContributeServerContextSink接口。 
    public class RollBackProperty : IContributeObjectSink, IContextProperty
    {
        //从接口名可以很清楚的看出来，通过这个上下文把消息接收器加到消息链中去
        //添加对象链
        #region IContributeObjectSink Members

        //这个方法就是最重要的了
        public IMessageSink GetObjectSink(MarshalByRefObject obj, IMessageSink nextSink)
        {
            //获取当前的消息接收器的实例
            return new RollBackMessageSink(nextSink);
        }

        #endregion

        #region IContextProperty Members

        //一般情况下不要实现，高级代码，o(∩_∩)o 哈哈
        //终结对象
        public void Freeze(Context newContext)
        {

        }

        //这是是问你是否想要创建新的上下文实例
        public bool IsNewContextOK(Context newCtx)
        {
            //默认情况下都是返回true
            return true;
        }

        public string Name
        {
            get
            {
                //可以任意指定字符串
                return "XXXXXXXXXXXXX";
            }
        }

        #endregion
    }

 4.倒数第二步了，定义特性，把第三个定义的属性加到运行时的上下文中去

    [AttributeUsage(AttributeTargets.Class)]
    public class RollBackContextAttribute: ContextAttribute
    {
        public RollBackContextAttribute()
            : base("VVVVVVVV")
        { }

        public override void GetPropertiesForNewContext(IConstructionCallMessage ctorMsg)
        {
            //把自定义好的上下文实例添加到上下文集合中去
            ctorMsg.ContextProperties.Add(new RollBackProperty());
        }
    }

5.最后一步，定义一个回滚帮助类

    [RollBackContext]
    //必须继承这个类，只有继承这个类才能使对象在多个上下文边界中进行传递
    public class RollBackHandler : ContextBoundObject
    {
    }

6.测试

    public class TestRollBack : RollBackHandler //必须继承这个回滚助手类
    {
        [通过消息机制实现回滚特性.RollBack] //为方法添加RollBack特性
        public void MyTest()
        {
            SqlConnectionStringBuilder connectionString = new SqlConnectionStringBuilder();
            connectionString.DataSource = @"LBDZ-20120514VC\SQLEXPRESS";
            connectionString.InitialCatalog = "My";
            connectionString.IntegratedSecurity = true;

            using (SqlConnection conn = new SqlConnection(connectionString.ToString()))
            {
                conn.Open();
                string strSQL = "INSERT dbo.MyNames (ListName) VALUES ( 'XXXXX')";
                SqlCommand cmd = conn.CreateCommand();
                cmd.CommandText = strSQL;
                cmd.ExecuteNonQuery();
            }
        }
    }

 运行结果可想而知。

好了，其实还没有对这些知识点内部机制有个很熟的理解，不过通过自己的动手实验，也让自己明白了这些东西是怎么个回事！

通过这件事也让俺明白了很多时候并不是你不笨，只是你的经验还不够，你的知识积累还很少，其实很多知识要领到了一定时候自然会明白的！

以同步至：个人文章目录索引