zoukankan      html  css  js  c++  java
  • DBHelper

    using System;
    using System.Data;
    using System.Data.Common;
    using Project.BaseFramework;
    using System.Collections.Generic;
    using System.Configuration;
    
    namespace Project.BaseFramework.DataProvider
    {
        public class DBHelper
        {
    
            #region Constuctor
            public DBHelper() { }
    
            private static string ConnectionString = ConfigurationManager.AppSettings["DBConnectionString"];
    
            private static IDBClient DBClient = DBClientFactory.GetDBClient(ConfigurationManager.AppSettings["DBClient"]);
    
            [ThreadStatic]
            private static TransConnection TransConnectionObj = null;
    
            #endregion
    
            #region ExecuteNonQuery
            public static int ExecuteNonQuery(CommandType cmdType, string cmdText, params DbParameter[] parameterValues)
            {
                int result = 0;
                bool mustCloseConn = true;
    
                DbCommand cmd = PrepareCmd(cmdType, cmdText, parameterValues, out mustCloseConn);
                OpenConn(cmd.Connection);
                result = cmd.ExecuteNonQuery();
    
                if (mustCloseConn) CloseConn(cmd.Connection);
                ClearCmdParameters(cmd);
                cmd.Dispose();
    
                return result;
            }
    
            #endregion ExecuteNonQuery
    
            #region ExecuteScalar
            public static object ExecuteScalar(CommandType cmdType, string cmdText, params DbParameter[] parameterValues)
            {
                object result = 0;
                bool mustCloseConn = true;
    
                DbCommand cmd = PrepareCmd(cmdType, cmdText, parameterValues, out mustCloseConn);
                OpenConn(cmd.Connection);
                result = cmd.ExecuteScalar();
    
                if (mustCloseConn) CloseConn(cmd.Connection);
                ClearCmdParameters(cmd);
                cmd.Dispose();
    
                return result;
            }
            #endregion ExecuteScalar
    
            #region ExecuteReader
            public static DbDataReader ExecuteReader(CommandType cmdType, string cmdText, params DbParameter[] parameterValues)
            {
                DbDataReader result = null;
                bool mustCloseConn = true;
                DbCommand cmd = PrepareCmd(cmdType, cmdText, parameterValues, out mustCloseConn);
                try
                {
                    OpenConn(cmd.Connection);
                    if (mustCloseConn)
                    {
                        result = cmd.ExecuteReader(CommandBehavior.CloseConnection);
                    }
                    else
                    {
                        result = cmd.ExecuteReader();
                    }
                    ClearCmdParameters(cmd);
                    return result;
                }
                catch (Exception ex)
                {
                    if (mustCloseConn) CloseConn(cmd.Connection);
                    ClearCmdParameters(cmd);
                    cmd.Dispose();
                    throw ;
                }
            }
            #endregion ExecuteReader
    
            #region ExecuteDataset
            public static DataSet ExecuteDataSet(CommandType cmdType, string cmdText, params DbParameter[] parameterValues)
            {
                DataSet result = null;
                bool mustCloseConn = true;
    
                DbCommand cmd = PrepareCmd(cmdType, cmdText, parameterValues, out mustCloseConn);
                using (DbDataAdapter da = DBClient.GetDbDataAdappter())
                {
                    da.SelectCommand = cmd;
                    result = new DataSet();
    
                    da.Fill(result);
                }
    
                if (mustCloseConn) CloseConn(cmd.Connection);
                ClearCmdParameters(cmd);
                cmd.Dispose();
    
                return result;
            }
            #endregion ExecuteDataset
    
            #region ExecuteDataTable
            public static DataTable ExecuteDataTable(CommandType cmdType, string cmdText, params DbParameter[] parameterValues) 
            {
                DataSet ds = ExecuteDataSet(cmdType,cmdText, parameterValues);
                if (ds != null && ds.Tables.Count > 0)
                    return ds.Tables[0];
                else
                    return null;
            }
            #endregion
    
            #region Transaction
            public static void BeginTransaction()
            {
                if (TransConnectionObj == null) 
                {
                    DbConnection conn = DBClient.GetDbConnection(ConnectionString);
                    OpenConn(conn);
                    DbTransaction trans = conn.BeginTransaction();
                    TransConnectionObj = new TransConnection();
                    TransConnectionObj.DBTransaction = trans;
                }
                else 
                {
                    TransConnectionObj.Deeps += 1;
                }
            }
    
            public static void CommitTransaction()
            {
                if (TransConnectionObj == null) return;
                if (TransConnectionObj.Deeps > 0)
                {
                    TransConnectionObj.Deeps -= 1;
                }
                else
                {
                    TransConnectionObj.DBTransaction.Commit();
                    ReleaseTransaction();
                }
            }
    
            public static void RollbackTransaction()
            {
                if (TransConnectionObj == null) return;
                if (TransConnectionObj.Deeps > 0)
                {
                    TransConnectionObj.Deeps -= 1;
                }
                else
                {
                    TransConnectionObj.DBTransaction.Rollback();
                    ReleaseTransaction();
                }
            }
    
            private static void ReleaseTransaction()
            {
                if (TransConnectionObj == null) return;
                DbConnection conn = TransConnectionObj.DBTransaction.Connection;
                TransConnectionObj.DBTransaction.Dispose();
                TransConnectionObj = null;
                CloseConn(conn);
            }
    
            #endregion
    
            #region Connection
            private static void OpenConn(DbConnection conn)
            {
                if (conn == null) conn = DBClient.GetDbConnection(ConnectionString);
                if (conn.State == ConnectionState.Closed) conn.Open();
            }
    
            private static void CloseConn(DbConnection conn)
            {
                if (conn == null) return;
                if (conn.State == ConnectionState.Open) conn.Close();
                conn.Dispose();
                conn = null;
            }
            #endregion
    
            #region Create DbParameter
    
            public static DbParameter CreateInDbParameter(string paraName, DbType type, int size, object value)
            {
                return CreateDbParameter(paraName, type, size, value, ParameterDirection.Input);
            }
    
            public static DbParameter CreateInDbParameter(string paraName, DbType type, object value)
            {
                return CreateDbParameter(paraName, type, 0, value, ParameterDirection.Input);
            }
    
            public static DbParameter CreateOutDbParameter(string paraName, DbType type, int size)
            {
                return CreateDbParameter(paraName, type, size, null, ParameterDirection.Output);
            }
    
            public static DbParameter CreateOutDbParameter(string paraName, DbType type)
            {
                return CreateDbParameter(paraName, type, 0, null, ParameterDirection.Output);
            }
    
            public static DbParameter CreateReturnDbParameter(string paraName, DbType type, int size)
            {
                return CreateDbParameter(paraName, type, size, null, ParameterDirection.ReturnValue);
            }
    
            public static DbParameter CreateReturnDbParameter(string paraName, DbType type)
            {
                return CreateDbParameter(paraName, type, 0, null, ParameterDirection.ReturnValue);
            }
    
            public static DbParameter CreateDbParameter(string paraName, DbType type, int size, object value, ParameterDirection direction)
            {
                DbParameter para = DBClient.GetDbParameter();
    
                para.ParameterName = paraName;
    
                if (size != 0)
                {
                    para.Size = size;
                }
    
                para.DbType = type;
    
                if (value != null)
                {
                    para.Value = value;
                }
                else
                {
                    para.Value = DBNull.Value;
                }
    
                para.Direction = direction;
    
                return para;
            }
    
            #endregion
    
            #region Command and Parameter
            /// <summary>
            /// 预处理用户提供的命令,数据库连接/事务/命令类型/参数
            /// </summary>
            /// <param>要处理的DbCommand</param>
            /// <param>数据库连接</param>
            /// <param>一个有效的事务或者是null值</param>
            /// <param>命令类型 (存储过程,命令文本, 其它.)</param>
            /// <param>存储过程名或都T-SQL命令文本</param>
            /// <param>和命令相关联的DbParameter参数数组,如果没有参数为'null'</param>
            /// <param><c>true</c> 如果连接是打开的,则为true,其它情况下为false.</param>
            private static DbCommand PrepareCmd(CommandType cmdType,string cmdText, DbParameter[] cmdParams, out bool mustCloseConn)
            {
                DbCommand cmd = DBClient.GetDbCommand(cmdText);
    
                DbConnection conn = null;
                if (TransConnectionObj != null)
                {
                    conn = TransConnectionObj.DBTransaction.Connection;
                    cmd.Transaction = TransConnectionObj.DBTransaction;
                    mustCloseConn = false;
                }
                else
                {
                    conn = DBClient.GetDbConnection(ConnectionString);
                    mustCloseConn = true;
                }
                cmd.Connection = conn;
    
                cmd.CommandType = cmdType;
    
                AttachParameters(cmd, cmdParams);
    
                return cmd;
            }
    
            /// <summary>
            /// 将DbParameter参数数组(参数值)分配给DbCommand命令.
            /// 这个方法将给任何一个参数分配DBNull.Value;
            /// 该操作将阻止默认值的使用.
            /// </summary>
            /// <param>命令名</param>
            /// <param>SqlParameters数组</param>
            private static void AttachParameters(DbCommand command, DbParameter[] commandParameters)
            {
                if (command == null) throw new ArgumentNullException("command");
                if (commandParameters != null)
                {
                    foreach (DbParameter p in commandParameters)
                    {
                        if (p != null)
                        {
                            // 检查未分配值的输出参数,将其分配以DBNull.Value.
                            if ((p.Direction == ParameterDirection.InputOutput || p.Direction == ParameterDirection.Input) &&
                            (p.Value == null))
                            {
                                p.Value = DBNull.Value;
                            }
                            command.Parameters.Add(p);
                        }
                    }
                }
            }
    
            private static void ClearCmdParameters(DbCommand cmd)
            {
                bool canClear = true;
                if (cmd.Connection != null && cmd.Connection.State != ConnectionState.Open)
                {
                    foreach (DbParameter commandParameter in cmd.Parameters)
                    {
                        if (commandParameter.Direction != ParameterDirection.Input)
                        {
                            canClear = false;
                            break;
                        }
                    }
                }
                if (canClear)
                {
                    cmd.Parameters.Clear();
                }
            }
            #endregion
        }
    }
    
    
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Text;
    using System.Data.Common;
    
    namespace Project.BaseFramework.DataProvider
    {
        internal class TransConnection
        {
            public TransConnection()
            {
                this.Deeps = 0;
            }
    
            public DbTransaction DBTransaction { get; set; }
    
            public int Deeps { get; set; }
        }
    }
    
    
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Text;
    using System.Data.Common;
    using System.Data.SqlClient;
    
    namespace Project.BaseFramework.DataProvider
    {
        public interface IDBClient
        {
            DbConnection GetDbConnection(string connectionString);
    
            DbCommand GetDbCommand(string cmdText);
    
            DbDataAdapter GetDbDataAdappter();
    
            DbParameter GetDbParameter();
        }
    }
    
    
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Text;
    using System.Data.Common;
    using System.Data.SqlClient;
    
    namespace Project.BaseFramework.DataProvider
    {
        public class SqlServerClient:IDBClient
        {
            public DbConnection GetDbConnection(string connectionString)
            {
                return new SqlConnection(connectionString);
            }
    
            public DbCommand GetDbCommand(string cmdText)
            {
                return new SqlCommand(cmdText);
            }
    
            public DbDataAdapter GetDbDataAdappter()
            {
                return new SqlDataAdapter();
            }
    
            public DbParameter GetDbParameter()
            {
                return new SqlParameter();
            }
        }
    }
    
    
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Text;
    using System.Data.Common;
    using MySql.Data.MySqlClient;
    
    namespace Project.BaseFramework.DataProvider
    {
        public class MySqlSqlClient:IDBClient
        {
            public DbConnection GetDbConnection(string connectionString)
            {
                return new MySqlConnection(connectionString);
            }
    
            public DbCommand GetDbCommand(string cmdText)
            {
                return new MySqlCommand(cmdText);
            }
    
            public DbDataAdapter GetDbDataAdappter()
            {
                return new MySqlDataAdapter();
            }
    
            public DbParameter GetDbParameter()
            {
                return new MySqlParameter();
            }
        }
    }
    
    
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Text;
    using System.Configuration;
    using System.Reflection;
    
    namespace Project.BaseFramework.DataProvider
    {
        public class DBClientFactory
        {
            private static readonly string path = "Project.BaseFramework";
    
            public static IDBClient GetDBClient(string dbClientClassName) 
            {
                if(string.IsNullOrEmpty(dbClientClassName))
                    dbClientClassName="SqlServerClient";
                string className = string.Format("{0}.DataProvider.{1}", path, dbClientClassName);
                return (IDBClient)Assembly.Load(path).CreateInstance(className);
            }
        }
    }
    <appSettings>
        <add key="DBConnectionString" value="Data Source=.;Initial Catalog=ProjectData;Persist Security Info=True;User ID=sa;Password=kjkj,911;"/>
        <add key="DBClient" value="SqlServerClient"/>
    </appSettings>

    转自:http://www.cnblogs.com/dataadapter/p/3866668.html收藏备用


    作者:GitLoft
    出处:http://www.cnblogs.com/joysky/
    本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。

  • 相关阅读:
    Python代码项目目录规范v1.0
    博客自定义
    Linux之查看CPU信息
    Python字符界面函数库
    数组的遍历你都会用了,那Promise版本的呢
    NPM实用指北
    如何从0开发一个Atom组件
    使用JavaScript实现一个俄罗斯方块
    使用PostMan进行API自动化测试
    一个有味道的函数
  • 原文地址:https://www.cnblogs.com/joysky/p/3867328.html
Copyright © 2011-2022 走看看