ShowEasy/EuroTran/SqlSugar/Base/SugarBase.cs

using System;
using System.Data;

namespace SqlSugar.Base
{
    using log4net;
    using System.Collections.Generic;
    using System.Linq;
    using System.Text;

    /// <summary>
    /// Sugar ORM父類, 封裝一些基本的操作
    /// </summary>
    public static class SugarBase
    {
        public static readonly ILog mo_Log = LogManager.GetLogger("SQLSugarLog");//記錄異常

        /// <summary>
        /// 資料庫連結字串
        /// </summary>
        public static string DB_ConnectionString = GetConnectionString("ConnectionString");

        /// <summary>
        /// 獲取ORM資料庫連線物件(只操作資料庫一次的使用, 否則會進行多次資料庫連接和關閉) 默認超時時間為30秒 預設為SQL Server資料庫 預設自動關閉資料庫連結,
        /// 多次操作資料庫請勿使用該屬性, 可能會造成性能問題 要自訂請使用GetIntance()方法或者直接使用Exec方法, 傳委託
        /// </summary>
        public static SqlSugarClient DB
        {
            get
            {
                return InitDB(null, 30, DbType.SqlServer, true);
            }
        }

        /// <summary>
        /// 獲得SqlSugarClient(使用該方法, 默認請手動釋放資源, 如using(var db = SugarBase.GetIntance()){你的代碼},
        /// 如果把isAutoCloseConnection參數設置為true, 則無需手動釋放, 會每次操作資料庫釋放一次, 可能會影響性能, 請自行判斷使用)
        /// </summary>
        /// <param name="commandTimeOut">等待超時時間, 默認為30秒 (單位: 秒)</param>
        /// <param name="dbType">資料庫類型, 預設為SQL Server</param>
        /// <param name="isAutoCloseConnection">
        /// 是否自動關閉資料庫連接, 預設不是, 如果設置為true, 則會在每次操作完資料庫後, 即時關閉, 如果一個方法裡面多次操作了資料庫, 建議保持為false, 否則可能會引發性能問題
        /// </param>
        /// <returns></returns>
        public static SqlSugarClient GetIntance(List<SqlFuncExternal> expfn = null, int commandTimeOut = 30, DbType dbType = DbType.SqlServer, bool isAutoCloseConnection = false)
        {
            var db = SugarBase.InitDB(expfn, commandTimeOut, dbType, isAutoCloseConnection);

            db.Aop.OnLogExecuting = (sql, pars) =>
            {

                var temsql = sql;
                
                var dic = pars.ToDictionary(it => it.ParameterName, it => it.Value);
                foreach (var param in dic)
                {
                    if (param.Value != null)
                    {
                        temsql = temsql.Replace(param.Key, "'" + param.Value.ToString() + "'");
                    }
                }

                System.Diagnostics.Debug.WriteLine(temsql + "\r\n");
                //mo_Log.Info(temsql);  //紀錄LOG打開此行
            };

            return db;
        }

        public static SqlSugarClient InstanceByAttribute
        {
            get
            {
                var db = new SqlSugarClient(new ConnectionConfig { InitKeyType = InitKeyType.Attribute, ConnectionString = Config.ConnectionString, DbType = DbType.SqlServer, IsAutoCloseConnection = true });
                return db;
            }
        }

        /// <summary>
        /// 初始化ORM連線物件
        /// </summary>
        /// <param name="commandTimeOut">等待超時時間, 默認為30秒 (單位: 秒)</param>
        /// <param name="dbType">資料庫類型, 預設為SQL Server</param>
        /// <param name="isAutoCloseConnection">
        /// 是否自動關閉資料庫連接, 預設不是, 如果設置為true, 則會在每次操作完資料庫後, 即時關閉, 如果一個方法裡面多次操作了資料庫, 建議保持為false, 否則可能會引發性能問題
        /// </param>
        private static SqlSugarClient InitDB(List<SqlFuncExternal> expfn, int commandTimeOut = 30, DbType dbType = DbType.SqlServer, bool isAutoCloseConnection = false)
        {
            var expMethods = GetExpMethods(expfn);
            var db = new SqlSugarClient(new ConnectionConfig
            {
                ConnectionString = SugarBase.DB_ConnectionString,
                DbType = dbType,
                InitKeyType = InitKeyType.Attribute,
                IsAutoCloseConnection = isAutoCloseConnection,
                ConfigureExternalServices = new ConfigureExternalServices
                {
                    SqlFuncServices = expMethods//set ext method
                }
            });
            db.Ado.CommandTimeOut = commandTimeOut;
            return db;
        }

        /// <summary>
        /// 執行資料庫操作
        /// </summary>
        /// <typeparam name="Result">返回數值型別 泛型</typeparam>
        /// <param name="func">方法委託</param>
        /// <param name="commandTimeOut">超時時間, 單位為秒, 預設為30秒</param>
        /// <param name="dbType">資料庫類型, 預設為SQL Server</param>
        /// <returns>泛型返回值</returns>
        public static Result Exec<Result>(Func<SqlSugarClient, Result> func, int commandTimeOut = 30, DbType dbType = DbType.SqlServer)
        {
            if (func == null) throw new Exception("委託為null, 交易處理無意義");
            using (var db = InitDB(null, commandTimeOut, dbType))
            {
                try
                {
                    return func(db);
                }
                catch (Exception ex)
                {
                    throw ex;
                }
                finally
                {
                    db.Dispose();
                }
            }
        }

        /// <summary>
        /// 帶交易處理的執行資料庫操作
        /// </summary>
        /// <typeparam name="Result">返回數值型別 泛型</typeparam>
        /// <param name="func">方法委託</param>
        /// <param name="commandTimeOut">超時時間, 單位為秒, 預設為30秒</param>
        /// <param name="dbType">資料庫類型, 預設為SQL Server</param>
        /// <returns>泛型返回值</returns>
        public static Result ExecTran<Result>(Func<SqlSugarClient, Result> func, int commandTimeOut = 30, DbType dbType = DbType.SqlServer)
        {
            if (func == null) throw new Exception("委託為null, 交易處理無意義");
            using (var db = InitDB(null, commandTimeOut, dbType))
            {
                try
                {
                    db.Ado.BeginTran(IsolationLevel.Unspecified);
                    var result = func(db);
                    db.Ado.CommitTran();
                    return result;
                }
                catch (Exception ex)
                {
                    db.Ado.RollbackTran();
                    throw ex;
                }
                finally
                {
                    db.Dispose();
                }
            }
        }

        /// <summary>
        /// 根據傳入的Key獲取設定檔中 相應Key的資料庫連接字串
        /// </summary>
        /// <param name="Key"></param>
        /// <returns></returns>
        public static string GetConnectionString(string Key)
        {
            try
            {
                var connectionString = DBUnit.GetAppSettings(Key);

                if (!String.IsNullOrEmpty(connectionString)) return connectionString;

                var sDbHost = DBUnit.GetAppSettings("DbHost");
                var sDbPort = DBUnit.GetAppSettings("DbPort");
                var sDbName = DBUnit.GetAppSettings("DbName");
                var sDbUser = DBUnit.GetAppSettings("DbUser");
                var sDbPassword = DBUnit.GetAppSettings("DbPassword");
                var sDbMinPoolSize = DBUnit.GetAppSettings("DbMinPoolSize");
                var sDbMaxPoolSize = DBUnit.GetAppSettings("DbMaxPoolSize");
                var sDbCharset = DBUnit.GetAppSettings("DbCharset");

                var sb = new StringBuilder();
                sb.Append("Data Source=").Append(sDbHost).Append(";");

                if (!String.IsNullOrEmpty(sDbPort))
                {
                    sb.Append("port=").Append(sDbPort).Append(";");
                }

                sb.Append("User ID=").Append(sDbUser).Append(";");
                sb.Append("Password=").Append(sDbPassword).Append(";");
                sb.Append("DataBase=").Append(sDbName).Append(";");

                if (!String.IsNullOrEmpty(sDbMinPoolSize))
                {
                    sb.Append("Min Pool Size=").Append(sDbMinPoolSize).Append(";");
                }

                if (!String.IsNullOrEmpty(sDbMinPoolSize))
                {
                    sb.Append("Max Pool Size=").Append(sDbMaxPoolSize).Append(";");
                }

                if (!String.IsNullOrEmpty(sDbCharset))
                {
                    sb.Append("charset=").Append(sDbCharset).Append(";");
                }

                return sb.ToString();
            }
            catch
            {
                throw new Exception("web.config檔appSettings中資料庫連接字串未配置或配置錯誤，必須為Key=\"connectionString\"");
            }
        }

        public static List<SqlFuncExternal> GetExpMethods(List<SqlFuncExternal> expfn)
        {
            var saMethods = new List<SqlFuncExternal>();
            var method = new SqlFuncExternal
            {
                UniqueMethodName = nameof(ISNULL),
                MethodValue = (expInfo, dbType, expContext) =>
                 {
                     return $"ISNULL({expInfo.Args[0].MemberName},'')";
                 }
            };
            if (expfn != null && expfn.Count > 0)
            {
                foreach (var _method in expfn)
                {
                    saMethods.Add(_method);
                }
            }

            return saMethods;
        }

        public static string ISNULL<T>(T str)
        {
            throw new NotSupportedException("Can only be used in expressions");
        }
    }
}