CaiQie/DB/DbCommandInterceptor.cs

using System.Data;
using System.Diagnostics;
using System.Text;
using Chloe.Infrastructure.Interception;

namespace DB
{


    public class DbCommandInterceptor : IDbCommandInterceptor
    {

        public void ReaderExecuting(IDbCommand command, DbCommandInterceptionContext<IDataReader> interceptionContext)
        {
            ToDb(command, interceptionContext.Exception);
        }

        public void ReaderExecuted(IDbCommand command, DbCommandInterceptionContext<IDataReader> interceptionContext)
        {
            ToDb(command, interceptionContext.Exception);
        }

        public void NonQueryExecuting(IDbCommand command, DbCommandInterceptionContext<int> interceptionContext)
        {
            ToDb(command, interceptionContext.Exception);
        }

        public void NonQueryExecuted(IDbCommand command, DbCommandInterceptionContext<int> interceptionContext)
        {
            ToDb(command, interceptionContext.Exception);
        }

        public void ScalarExecuting(IDbCommand command, DbCommandInterceptionContext<object> interceptionContext)
        {
            ToDb(command, interceptionContext.Exception);
        }

        public void ScalarExecuted(IDbCommand command, DbCommandInterceptionContext<object> interceptionContext)
        {
            ToDb(command, interceptionContext.Exception);
        }

        private void ToDb(IDbCommand command, Exception exception)
        {

#if DEBUG
            Trace.WriteLine(command.CommandText);
            Trace.WriteLine(AppendDbCommandInfo(command));
#endif
        }

        private string AppendDbCommandInfo(IDbCommand command)
        {
            var sb = new StringBuilder();

            foreach (IDbDataParameter param in command.Parameters)
            {
                if (param == null)
                    continue;

                object value;
                if (param.Value == null || param.Value == DBNull.Value)
                {
                    value = "NULL";
                }
                else
                {
                    value = param.Value;

                    if (param.DbType == DbType.String || param.DbType == DbType.AnsiString ||
                        param.DbType == DbType.DateTime)
                        value = "'" + value + "'";
                }

                sb.AppendFormat("{3} {0} {1} = {2};", Enum.GetName(typeof(DbType), param.DbType), param.ParameterName,
                    value, Enum.GetName(typeof(ParameterDirection), param.Direction));
                sb.AppendLine();
            }

            return sb.ToString();
        }
    }
}