NWeiWMS/ICSSoft.FromERP/ICSSoft.FromERP/SqlHelper.cs

using System;
using System.Collections.Generic;
using System.Configuration;
using System.Data;
using System.Data.SqlClient;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace NFine.Data.Extensions
{
    /// <summary>
    /// 数据库访问辅助类，add by ngye, on 2013-08-14.
    /// </summary>
    public static class SqlHelper
    {
        #region [ 连接串相关 ]
        /// <summary>
        /// 数据中心DB的连接字符串
        /// </summary>
        public static string DataCenterConnString = ConfigurationManager.ConnectionStrings["connstr"].ConnectionString;
        /// <summary>
        /// 获取同步服务器的连接
        /// </summary>
        /// <returns></returns>
        public static SqlConnection GetDataCenterConn()
        {
            return new SqlConnection(DataCenterConnString);
        }

        /// <summary>
        /// 根据连接串获取连接
        /// </summary>
        /// <returns></returns>
        public static SqlConnection GetConnByString(string conn)
        {
            return new SqlConnection(conn);
        }

        /// <summary>
        /// 测试连接串是否能正确
        /// </summary>
        /// <param name="connectionString">连接串</param>
        /// <returns></returns>
        public static bool TestConnectionString(string connectionString)
        {
            bool result = true;
            try
            {
                using (SqlConnection conn = GetConnByString(connectionString))
                {
                    try
                    {
                        conn.Open();
                    }
                    catch (Exception ex)
                    {
                        result = false;
                    }
                }
            }
            catch (Exception)
            {
                result = false;
            }
            return result;
        }

        /// <summary>
        /// 测试连接串是否正确
        /// </summary>
        /// <param name="connectionString">连接串</param>
        /// <param name="timeOut">测试的超时秒数</param>
        /// <param name="errInfo">错误时输出的错误信息</param>
        /// <returns>是否能正常连接</returns>
        public static bool TestConnectionString(string connectionString, int timeOut, ref string errInfo)
        {
            bool result = true;
            string[] arr = connectionString.Split(new char[] { ';' });
            List<string> list = new List<string>();

            foreach (string s in arr)
            {
                if (s.ToLower().IndexOf("timeout") == -1)
                {
                    list.Add(s);
                }
                else
                {
                    list.Add(String.Format("Connection Timeout={0}", timeOut));
                }
            }

            SqlConnection con = null;

            try
            {
                con = new SqlConnection(string.Join(";", list.ToArray()));
                con.Open();
            }
            catch (Exception ex)
            {
                result = false;
                errInfo = ex.Message;
            }
            finally
            {
                if (con != null)
                {
                    con.Close();
                    con.Dispose();
                    con = null;
                }
            }
            return result;
        }
        #endregion

        #region [ 超时设置 ]
        public static int CommandTimeout
        {
            get
            {
                return 7200;
            }
        }
        #endregion


        #region [ ExecuteNonQuery ]
        /// <summary>
        /// 根据sql语句和参数，返回受影响行数
        /// </summary>
        /// <param name="sql">sql语句</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>受影响行数</returns>
        public static int ExecuteNonQuery(string sql, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetDataCenterConn())
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(sql, conn).AddTimeout();
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                return cmd.ExecuteNonQuery();
            }
        }

        public static int ExecuteNonQuery(string connStr, string sql, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = new SqlConnection(connStr))
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(sql, conn).AddTimeout();
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                return cmd.ExecuteNonQuery();
            }
        }
        #endregion

        #region [ ExecuteScalar_ForProc ]
        /// <summary>
        /// 根据存储过程和参数，返回Scalar结果
        /// </summary>
        /// <param name="proc">存储过程</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>object</returns>
        public static object ExecuteScalar_ForProc(string proc, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetDataCenterConn())
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(proc, conn).AddTimeout();
                cmd.CommandType = CommandType.StoredProcedure;
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                object obj = cmd.ExecuteScalar();
                return obj;
            }
        }

        /// <summary>
        /// 根据存储过程和参数，返回Scalar结果，但不加上超时设置 ( 避免循环做 AddTimeout )
        /// </summary>
        /// <param name="proc">存储过程</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>object</returns>
        public static object ExecuteScalar_ForProc_WithoutAddTimeout(string proc, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetDataCenterConn())
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(proc, conn);
                cmd.CommandType = CommandType.StoredProcedure;
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                object obj = cmd.ExecuteScalar();
                return obj;
            }
        }
        #endregion

        #region [ ExecuteScalar ]
        /// <summary>
        /// 根据sql语句和参数，返回Scalar结果
        /// </summary>
        /// <param name="sql">sql语句</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>object</returns>
        public static object ExecuteScalar(string sql, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetDataCenterConn())
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(sql, conn).AddTimeout();
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                object obj = cmd.ExecuteScalar();
                return obj;
            }
        }
        #endregion

        #region [ QueryByProc ]
        /// <summary>
        /// 根据存储过程及参数，返回DataSet
        /// </summary>
        /// <returns>DataSet</returns>
        public static DataSet GetDataSetByProc(string proc, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetDataCenterConn())
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(proc, conn).AddTimeout();
                cmd.CommandType = CommandType.StoredProcedure;
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                SqlDataAdapter adapter = new SqlDataAdapter(cmd);
                DataSet ds = new DataSet();
                adapter.Fill(ds);
                return ds;
            }
        }

        /// <summary>
        /// 根据存储过程及参数，返回DataTable
        /// </summary>
        /// <returns>DataTable</returns>
        public static DataTable GetDataTableByProc(string proc, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetDataCenterConn())
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(proc, conn).AddTimeout();
                cmd.CommandType = CommandType.StoredProcedure;
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                DataTable dt = cmd.ExecuteDataTable();
                return dt;
            }
        }

        /// <summary>
        /// 根据sql语句和参数，返回DataRow
        /// </summary>
        /// <param name="sql">sql语句</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>DataRow</returns>
        public static DataRow GetDataRowByProc(string proc, params SqlParameter[] spArr)
        {
            DataTable dt = GetDataTableByProc(proc, spArr);
            if (dt == null || dt.Rows.Count == 0)
                return null;

            return dt.Rows[0];
        }
        #endregion

        #region [ Query ]
        /// <summary>
        /// 根据sql语句和参数，返回DataSet
        /// </summary>
        /// <param name="sql">sql语句</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>DataSet</returns>
        public static DataSet GetDataSetBySql(string sql, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetDataCenterConn())
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(sql, conn).AddTimeout();
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                SqlDataAdapter adapter = new SqlDataAdapter(cmd);
                DataSet ds = new DataSet();
                adapter.Fill(ds);
                return ds;
            }
        }

        /// <summary>
        /// 根据sql语句和参数，返回DataTable
        /// </summary>
        /// <param name="sql">sql语句</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>DataTable</returns>
        public static DataTable GetDataTableBySql(string sql, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetDataCenterConn())
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(sql, conn).AddTimeout();
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                DataTable dt = cmd.ExecuteDataTable();
                return dt;
            }
        }

        /// <summary>
        /// 根据sql语句和参数，返回DataRow
        /// </summary>
        /// <param name="sql">sql语句</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>DataRow</returns>
        public static DataRow GetDataRowBySql(string sql, params SqlParameter[] spArr)
        {
            DataTable dt = GetDataTableBySql(sql, spArr);
            if (dt == null || dt.Rows.Count == 0)
                return null;

            return dt.Rows[0];
        }
        #endregion

        #region [ GetValue ]
        public static object GetValue(DataRow dr, string fieldName, object replaceValue)
        {
            if (dr.IsNull(fieldName))
                return replaceValue;
            return dr[fieldName];
        }

        public static object GetValue(DataRow dr, int idx, object replaceValue)
        {
            if (dr.IsNull(idx))
                return replaceValue;
            return dr[idx];
        }
        #endregion

        #region [ GetString ]
        public static string GetString(DataRow dr, string fieldName, string replaceValue)
        {
            if (dr.IsNull(fieldName))
                return replaceValue;
            return Convert.ToString(dr[fieldName]);
        }

        public static string GetString(DataRow dr, int idx, string replaceValue)
        {
            if (dr.IsNull(idx))
                return replaceValue;
            return Convert.ToString(dr[idx]);
        }
        #endregion

        #region [ GetDateTime ]
        public static DateTime GetDateTime(DataRow dr, string fieldName, DateTime replaceValue)
        {
            if (dr.IsNull(fieldName))
                return replaceValue;
            return Convert.ToDateTime(dr[fieldName]);
        }
        public static DateTime GetDateTime(DataRow dr, int idx, DateTime replaceValue)
        {
            if (dr.IsNull(idx))
                return replaceValue;
            return Convert.ToDateTime(dr[idx]);
        }
        #endregion

        #region [ 非数据中心库操作 ]

        #region [ ExecuteScalar ]
        /// <summary>
        /// 根据sql语句和参数，返回Scalar结果
        /// </summary>
        /// <param name="connString">连接串</param>
        /// <param name="sql">sql语句</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>object</returns>
        public static object ExecuteScalarWithConn(string connString, string sql, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetConnByString(connString))
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(sql, conn).AddTimeout();
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                object obj = cmd.ExecuteScalar();
                return obj;
            }
        }
        #endregion

        #region [ ExecuteNonQuery ]
        /// <summary>
        /// 根据sql语句和参数，返回受影响行数
        /// </summary>
        /// <param name="connString">连接串</param>
        /// <param name="sql">sql语句</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>受影响行数</returns>
        public static int ExecuteNonQueryWithConn(string connString, string sql, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetConnByString(connString))
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(sql, conn).AddTimeout();
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                return cmd.ExecuteNonQuery();
            }
        }
        #endregion

        #region [ Query ]
        /// <summary>
        /// 根据sql语句和参数，返回DataTable
        /// </summary>
        /// <param name="connString">连接串</param>
        /// <param name="sql">sql语句</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>DataTable</returns>
        public static DataTable GetDataTableBySqlWithConn(string connString, string sql, params SqlParameter[] spArr)
        {
            using (SqlConnection conn = SqlHelper.GetConnByString(connString))
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand(sql, conn).AddTimeout();
                if (spArr.Length > 0)
                    cmd.Parameters.AddRange(spArr.SetDBNull());
                DataTable dt = cmd.ExecuteDataTable();
                return dt;
            }
        }

        /// <summary>
        /// 根据sql语句和参数，返回DataRow
        /// </summary>
        /// <param name="connString">连接串</param>
        /// <param name="sql">sql语句</param>
        /// <param name="spArr">可变参数</param>
        /// <returns>DataRow</returns>
        public static DataRow GetDataRowBySqlWithConn(string connString, string sql, params SqlParameter[] spArr)
        {
            DataTable dt = GetDataTableBySqlWithConn(connString, sql, spArr);
            if (dt == null || dt.Rows.Count == 0)
                return null;

            return dt.Rows[0];
        }
        #endregion
        #endregion

        #region [ 根据SQL文件路径执行 ]
        /// <summary>
        /// 执行SQL文件
        /// </summary>
        /// <param name="connString"></param>
        /// <param name="filePath"></param>
        /// <returns></returns>
        public static bool ExecuteNonQueryWithConnAndSqlFilePath(string connString, string filePath)
        {
            string sql = System.IO.File.ReadAllText(filePath);
            return ExecuteNonQueryWithConnAndGO(connString, sql);
        }
        #endregion

        #region [ 执行带Go语句 ]
        /// <summary>  
        /// 执行带"GO"的SQL，返回最后一条SQL的受影响行数  
        /// </summary>  
        /// <param name="connString">连接串</param>  
        /// <param name="sql">sql语句</param>  
        /// <returns>返回最后一条SQL的受影响行数</returns>
        public static bool ExecuteNonQueryWithConnAndGO(string connString, string sql)
        {
            bool result = true;
            string[] arr = System.Text.RegularExpressions.Regex.Split(sql, @"\bGO\b", System.Text.RegularExpressions.RegexOptions.IgnoreCase);
            using (SqlConnection conn = GetConnByString(connString))
            {
                conn.Open();
                SqlCommand cmd = new SqlCommand().AddTimeout();
                cmd.Connection = conn;
                SqlTransaction tx = conn.BeginTransaction();
                cmd.Transaction = tx;
                try
                {
                    for (int n = 0; n < arr.Length; n++)
                    {
                        string strsql = arr[n];
                        if (strsql.Trim().Length > 1 && strsql.Trim().Replace(";", "") != "")
                        {
                            cmd.CommandText = strsql;
                            cmd.ExecuteNonQuery();
                        }
                    }
                    tx.Commit();
                }
                catch (System.Data.SqlClient.SqlException E)
                {
                    tx.Rollback();
                    result = false;
                    //return -1;  
                    throw new Exception(E.Message);
                }
                finally
                {
                    if (conn.State != ConnectionState.Closed)
                    {
                        conn.Close();
                        conn.Dispose();
                    }
                }
            }
            return result;
        }
        #endregion


        /// <summary>
        /// 设置Sql参数
        /// </summary>
        /// <param name="parameterName"></param>
        /// <param name="value"></param>
        /// <param name="dbType"></param>
        /// <returns></returns>
        public static SqlParameter SetSqlParameter(string parameterName, object value, SqlDbType dbType)
        {
            if (value == null || string.IsNullOrEmpty(value.ToString()))
            {
                return new SqlParameter(parameterName, DBNull.Value);
            }
            SqlParameter sp = new SqlParameter(parameterName, value);
            sp.SqlDbType = dbType;
            return sp;
        }

        /// <summary>
        /// 根据sql及带 in 的id参数列表, 得到DataTable
        /// </summary>
        /// <param name="sql">带in ( {0} )的sql</param>  
        /// <param name="ids">以逗号分隔的id字符串</param>
        /// <param name="idDbType">id的Sql数据类型</param>
        /// <returns>DataTable</returns>
        public static DataTable GetDataTableWithIn(string sql, string ids, SqlDbType idDbType)
        {
            SqlParameter[] spArr = GetWithInSqlParameters(ref sql, ids, idDbType);
            return GetDataTableBySql(sql, spArr);
        }

        /// <summary>
        /// 根据sql及带 in 的id参数列表,　得到受影响行数
        /// </summary>
        /// <param name="sql">带in ( {0} )的sql</param>  
        /// <param name="ids">以逗号分隔的id字符串</param>
        /// <param name="idDbType">id的Sql数据类型</param>
        /// <returns>DataTable</returns>
        public static int ExecuteNonQueryWithIn(string sql, string ids, SqlDbType idDbType)
        {
            SqlParameter[] spArr = GetWithInSqlParameters(ref sql, ids, idDbType);
            return ExecuteNonQuery(sql, spArr);
        }

        #region [ 带 in 不确定参数的执行方法 ]
        /// <summary>  
        /// 获取带 in 的sql参数列表  
        /// </summary>  
        /// <param name="sql">带in ( {0} )的sql</param>  
        /// <param name="ids">以逗号分隔的id字符串</param>
        /// <param name="idDbType">id的Sql数据类型</param>
        /// <returns>sql参数列表</returns>  
        public static SqlParameter[] GetWithInSqlParameters(ref string sql, string ids, SqlDbType idDbType)
        {
            if (string.IsNullOrEmpty(ids))
            {
                return null;
            }

            string[] idArr = ids.Split(',');
            //组建sql在in中的字符串  
            StringBuilder sbCondition = new StringBuilder();
            List<SqlParameter> spList = new List<SqlParameter>();
            for (int i = 0; i < idArr.Length; i++)
            {
                string id = idArr[i];
                string spName = string.Format("@id{0}", i);
                sbCondition.AppendFormat("{0},", spName);
                spList.Add(SetSqlParameter(spName, id, idDbType));
            }
            //重新构建sql  
            sql = string.Format(sql, sbCondition.ToString().TrimEnd(','));
            return spList.ToArray();
        }
        #endregion  
    }//end of class

    #region [ SqlHelper 的扩展方法类 ]
    /// <summary>
    /// 扩展方法类
    /// </summary>
    public static class SqlHelperExtensionMethods
    {
        /// <summary>
        /// 新建SqlCommand对象时，　自动加上指定的 CommandTimeout. by ngye, on 2013-07-11.
        /// </summary>
        /// <param name="cmd">SqlCommand对象</param>
        /// <returns>SqlCommand对象</returns>
        public static SqlCommand AddTimeout(this SqlCommand cmd)
        {
            cmd.CommandTimeout = SqlHelper.CommandTimeout;
            return cmd;
        }

        /// <summary>
        /// 新建SqlBulkCopy对象时，　自动加上指定的 BulkCopyTimeout. by ngye, on 2013-08-30.
        /// </summary>
        /// <param name="cmd">SqlBulkCopy对象</param>
        /// <returns>SqlBulkCopy对象</returns>
        public static SqlBulkCopy AddTimeout(this SqlBulkCopy bulkCopy)
        {
            bulkCopy.BulkCopyTimeout = SqlHelper.CommandTimeout;
            return bulkCopy;
        }

        /// <summary>
        /// 执行cmd得到 DataTable. by ngye, on 2013-08-01
        /// </summary>
        /// <param name="cmd"></param>
        /// <returns></returns>
        public static DataTable ExecuteDataTable(this SqlCommand cmd)
        {
            DataTable dt = new DataTable();
            using (SqlDataAdapter adapter = new SqlDataAdapter(cmd))
            {
                adapter.Fill(dt);
            }
            return dt;
        }

        /// <summary>
        /// 为SqlParameter设置参数. by ngye, on 2013-08-15.
        /// </summary>
        /// <param name="sp"></param>
        /// <returns></returns>
        public static SqlParameter SetValue(this SqlParameter sp, object value)
        {
            sp.Value = value;
            return sp;
        }

        /// <summary>
        /// 为SqlParameter设置SqlDbType. by ngye, on 2013-09-03.
        /// </summary>
        /// <param name="sp"></param>
        /// <returns></returns>
        public static SqlParameter SetSqlDbType(this SqlParameter sp, SqlDbType dbType)
        {
            sp.SqlDbType = dbType;
            return sp;
        }

        /// <summary>
        /// 对可以为空的值作这样的处理，一旦其为空，就设置为DBNull.value.
        /// </summary>
        /// <param name="sp"></param>
        /// <returns></returns>
        public static SqlParameter[] SetDBNull(this SqlParameter[] spArr)
        {
            if (spArr == null || spArr.Length == 0)
                return spArr;
            for (int i = 0; i < spArr.Length; i++)
            {
                SqlParameter sp = spArr[i];
                if (sp.Value == null)
                    sp.Value = DBNull.Value;
            }
            return spArr;
        }
    }
    #endregion
}