• petapoco 对存储过程的扩展 干货


      好久没发表文章了。心血来潮,简单的介绍下这次工作中的问题。

    项目中运用了Petapoco,可是petapoco对存储过程的支持不够好。或者说对于某些特殊场景,petapoco的sql支持度有限。

    比如rowcount, 比如传入的字符创是{'1','2'}

    因为我的项目是老系统改造,老系统很多奇怪的sql都有而且比较复杂,为了保证系统稳定,复杂sql还是沿用下来了。但这些历史遗留问题,就带来了附加问题。

    原来打算用 petapoco+sqlhelper来处理,因为时间比较紧迫。不过2个connection下必须开启分布式事务才能解决一致性,觉得有点没有必要。

    so 决定改造petapoco ,其实内部代码我已经改过很多了。petapoco很好扩展。就是侵入性有点强,必须要改源代码。

    1.扩展存储过程

    首先源代码需要修改的地方

           string _connectionString;
            string _providerName;
            DbProviderFactory _factory;
            protected  IDbConnection _sharedConnection;
           // IDbConnection _sharedConnection;  私有方法,不好扩展啊,必须改作用域
            IDbTransaction _transaction;
            int _sharedConnectionDepth;
            int _transactionDepth;
            bool _transactionCancelled;
            string _lastSql;
            object[] _lastArgs;
            string _paramPrefix = "@";     

    然后就可以在扩展类中实现需要的功能了。

    对database进行扩展。

    目前实现了2个功能

    a.存储过程

    b.打印sql.网上搜索的sql都是带参数的,和没打印一样。我直接输出结果了。

    namespace Hg.Framework.Repository
    {
        public class DataBaseE : Database
        {
            public DataBaseE(IDbConnection connection)
                : base(connection)
            {
            }
    
            public DataBaseE(string connectionString, string providerName)
                : base(connectionString, providerName)
            {
            }
    
            public DataBaseE(string connectionString, DbProviderFactory provider)
                : base(connectionString, provider)
            {
            }
    
            public DataBaseE(string connectionStringName)
                : base(connectionStringName)
            {
            }
    
           public SqlDataReader RunProcedure(string storedProcName, IDataParameter[] parameters)
            {
                try
                {
                    OpenSharedConnection();
                    try
                    {
    
                        SqlDataReader returnReader;
    
                        SqlCommand command = BuildQueryCommand(_sharedConnection, storedProcName, parameters);
                        command.CommandType = CommandType.StoredProcedure;
                        returnReader = command.ExecuteReader(CommandBehavior.CloseConnection);
                        return returnReader;
    
                 
                    }
                    finally
                    {
                        CloseSharedConnection();
                    }
                }
                catch (Exception x)
                {
                    OnException(x);
                    throw;
                }
            }
    
    
           /// <summary>
           /// 执行存储过程,返回影响的行数        
           /// </summary>
           /// <param name="storedProcName">存储过程名</param>
           /// <param name="parameters">存储过程参数</param>
           /// <param name="rowsAffected">影响的行数</param>
           /// <returns></returns>
           public   int RunProcedure(string storedProcName, IDataParameter[] parameters, out int rowsAffected)
           {
               try
               {
                   OpenSharedConnection();
                   try
                   {
                       int result; 
                       SqlCommand command = BuildIntCommand(_sharedConnection, storedProcName, parameters);
                       rowsAffected = command.ExecuteNonQuery();
                       result = (int)command.Parameters["ReturnValue"].Value;
                       //Connection.Close();
                       return result;
                   }
                   finally
                   {
                       CloseSharedConnection();
                   }
               }
               catch (Exception x)
               {
                   OnException(x);
                   throw;
               }
           }
    
           /// <summary>
           /// 创建 SqlCommand 对象实例(用来返回一个整数值)    
           /// </summary>
           /// <param name="storedProcName">存储过程名</param>
           /// <param name="parameters">存储过程参数</param>
           /// <returns>SqlCommand 对象实例</returns>
           private static SqlCommand BuildIntCommand(IDbConnection _connection, string storedProcName, IDataParameter[] parameters)
           {
               SqlConnection connection = _connection as SqlConnection;
               SqlCommand command = BuildQueryCommand(connection, storedProcName, parameters);
               command.Parameters.Add(new SqlParameter("ReturnValue",
                   SqlDbType.Int, 4, ParameterDirection.ReturnValue,
                   false, 0, 0, string.Empty, DataRowVersion.Default, null));
               return command;
           }
                    /// <summary>
                    /// 构建 SqlCommand 对象(用来返回一个结果集,而不是一个整数值)
                    /// </summary>
                    /// <param name="connection">数据库连接</param>
                    /// <param name="storedProcName">存储过程名</param>
                    /// <param name="parameters">存储过程参数</param>
                    /// <returns>SqlCommand</returns>
                    private static SqlCommand BuildQueryCommand(IDbConnection _connection, string storedProcName, IDataParameter[] parameters)
                    {
                        SqlConnection connection = _connection as SqlConnection;
                        SqlCommand command = new SqlCommand(storedProcName, connection);
                        command.CommandType = CommandType.StoredProcedure;
                        foreach (SqlParameter parameter in parameters)
                        {
                            if (parameter != null)
                            {
                                // 检查未分配值的输出参数,将其分配以DBNull.Value.
                                if ((parameter.Direction == ParameterDirection.InputOutput || parameter.Direction == ParameterDirection.Input) &&
                                    (parameter.Value == null))
                                {
                                    parameter.Value = DBNull.Value;
                                }
                                command.Parameters.Add(parameter);
                            }
                        }
    
                        return command;
                    }
    
            public override void OnExecutedCommand(IDbCommand cmd)
            {
    
                Debug.Print(FormatCommand2());
    
                base.OnExecutedCommand(cmd);
            }
    
            private string FormatCommand2()
            {
                string sql = LastSQL;
                object[] args = LastArgs;
    
                if (sql == null)
                    return "";
                string arg = "";
                if (args != null && args.Length > 0)
                {
                    for (int i = 0; i < args.Length; i++)
                    {
    
                        switch (args[i].GetType().Name)
                        {
                            case "Int32":
                            case "Decimal":
                                arg = args[i].ToString();
                                break;
                            default:
                                arg = "'" + args[i].ToString() + "'";
                                break;
                        }
    
                        var fh = "@" + i;
                        int startIndex = sql.IndexOf(fh);
                        int charLength = startIndex + fh.Length;
    
                     sql=   sql.Substring(0, charLength).Replace(fh, arg) + sql.Substring(charLength);
    
                    Log.Debug(sql,null);
                        //sb.AppendFormat("	 -> {0}{1} [{2}] = "{3}"
    ", "@", i, args[i].GetType().Name, args[i]);
                    }
    
                }
                return sql;
            }
    
        }
    }

    好了。总体来说 petapoco还是很不错的。能满足大部分功能。如果能开放connection 和 transcation接口的话,会更灵活点。

    OK,就到这里了。

  • 相关阅读:
    【OBIEE】补全维度成员
    【OBIEE】超过了已配置的最大允许输入记录数
    【OBIEE】开启和关闭OBIEE的SSO登陆认证
    【OBIEE】OBIEE重装删除DEV前缀
    【OBIEE】OBIEE 12C迁移
    【OBIEE】OBIEE集成第三方js
    【Java】webservice调用
    【前端】汉字转拼音首字母
    【Oracle】Redhat安装Oracle 11g
    CAS单点登录:基础框架搭建(一)
  • 原文地址:https://www.cnblogs.com/yuanhuaming/p/4097751.html
Copyright © 2020-2023  润新知