我也来写:数据库访问类DBHelper

一、前言

  相信许多人都百度过:“.net 数据库访问类”。然后就出来一大堆SqlHelper。我也用过这些SqlHelper,也自己写过,一堆静态方法,开始使用起来感觉很不错,它们也确实在很多时候可以很好的工作。ADO.NET已经封装很好了,我们很容易就可以实现自己的数据库访问类。

  很久前,忘记在哪里看到过了,有一个朋友写了一篇【如何做一个好用的数据库访问类】(有兴趣的朋友仍然可以搜索到),这篇文章确实写得很好,作者很详细的讲解了如何设计一个好的数据库访问类;所谓“好“是指:轻量、易用、通用、高效。

  其实代码是很久前就实现了,只是现在才总结记录,希望可以分享一下学习的过程。ok,在开始前先来看几个ADO.NET常见的面试题:

1. ADO.NET 5个核心对象是哪5个?

2. 与ADO.NET 相关对象中,哪些可以用于数据绑定?

3. DataSet与DataReader有什么区别?分别适用在什么情况?

二、需求

  这是一个简单的、基于ADO.NET的数据库访问类,它最起码要具备以下特点:  

1. 支持多种数据库

  搞.net的视乎有一个固定的思维:数据库就是用sql server。额,只是很多用sql server,但不是全部,也有很多用 my sql 等的。我们并不能限制一定用什么数据库。

2. 支持多个数据库

  有时候我们的应用程序会用到多个数据库,并且这些数据库还不是部署在同一台服务器上的。

3. 简单

  满足常见的操作。

4. 可扩展

  可以随时增加新的方法;对于具体的数据源,也可以有特有的操作。

三、主要说明

3.1 使用DbProviderFactory

  既然要支持多种数据库,那么我们之前常写的SqlConnection、SqlCommand 就都不能用了,因为它们是针对sql server 数据源的。如果换成 my sql 就是 MySqlConnection, Oracle 就是 OracleConnection 了。

  既然有那么多种Connection,很多朋友可能会想到通过设计模式来处理,例如定义一个父类(或接口),然后各种类型的数据库继承它,然后再通过一个工厂,来创建所需要的对象;以后要增加哪种类型的数据库就再增加一个对应的类就可以了。大概是像下面这样:

    public abstract class DBHelper
    {
        public abstract void Open(string key){}
        public abstract int ExecuteNonQuery() { }
        public abstract object ExecuteScalar() { }
        public abstract DataSet GetDataSet() { }
    }

    public class SqlHelper : DBHelper
    {
        const char _prefix = '@';
        //实现抽象方法...
    }

    public class MySqlHelper : DBHelper
    {
        const char _prefix = "@";
        //实现抽象方法...
    }

    public class OracleHelper : DBHelper
    {
        const char _prefix = ":";
        //实现抽象方法...
    }

    public class DBFactory
    {
        public static DBHelper GetDBHelper()
        {
            //根据条件返回DBHelper
        }
    }

  这样实现已经比用SqlXXX好很多了,这也是我之前写过的一种方式。但它仍然不够灵活,并且实现起来就会发现很多代码都是类似的,这就与我们上面的简单的需求相违背了。

  通过上面的分析,我们知道用工厂模式可以解决我们的问题,但这不用我们自己实现,.net 早就提供这样的工厂:DbProviderFactory。由名称可以指定DbProviderFactory就是数据源提供程序工厂,负责创建具体的数据源提供程序。它根据 ProviderName就可以创建对应数据源的访问对象了。这样我们的实现也由具体变成抽象了,具体的SqlConection变成了抽象的DbConnection。

  什么是 ProviderName? 在配置 web.config 的connectionStrings 时,就会有一个 providerNmae 属性,例如sql server就是 ”System.Data.SqlClient“,这个名称空间就是对应的数据源提供程序。

3.2 参数问题

  不同数据库参数查询的格式可能不一样,例如 sql server/my sql 支持“@变量” 形式,而 oracle 支持“:变量”的形式。像上面的父类的写法,子类就必须定义自己的参数前缀。但这些用了DbProviderFactory后也不是问题了。

3.3 using 问题

  我们都知道using是c#的语法糖,其实编译后就是 try-finaly;uisng写起来比较优雅,而且在有异常的时候会自动调用对象的Disponse方法,避免有些人忘记调用。所以嵌套的 using,编译后就是嵌套的try-finaly,但其实只要我们注意在抛异常的时候释放资源,一个try-finaly即可。

3.4 DbDataReader 问题

  实际项目中,我们更多的是使用DbDataReader而非DataSet/DataTable,而 DbDataReader需要自己逐行读取,这在每个调用的地方都这样写是很麻烦的,怎么解决?委托,又是它!

  说到委托还有一个小小的建议,有些人喜欢自己去定义委托,但其实.net已经内置了3种委托:Func、Action、Predicate,并且提供了多个重载版本,应该优先考虑使用这些委托,在不满足的情况下,再去自定义。

3.5 在分层架构里的角色

  为 DAL 层提供数据访问服务,由 DAL 直接调用;不涉及sql语句拼接、日志记录等。

四、例子

  假设要调用一个 P_GetFriends存储过程,接收一个id参数,返回一个好友列表。如下:

        public List<Friend> GetFriends(int id)
        {
            try
            {                
                DBHelper helper = new DBHelper("dbConnectionKey");
                DbParameter[] parameters = new DbParameter[]
                {
                    helper.CreateDbParameter("id",id)
                };
                return helper.ExecuteReader(CommandType.StoredProcedure, "P_GetFriends", parameters,
                reader =>
                {
                    return new Friend()
                    {
                        ID = reader.GetInt32(reader.GetOrdinal("ID")),
                        Name = reader.GetString(reader.GetOrdinal("Name"))
                    };
                });
            }
            catch
            {
                throw;
            }
        }  

附源代码

    public class DBHelper
    {
        #region 属性

        /// <summary>
        /// 链接字符串
        /// </summary>
        private string conStr;

        /// <summary>
        /// DB工厂
        /// </summary>
        private DbProviderFactory provider;

        #endregion

        #region 构造函数

        /// <summary>
        /// 构造函数
        /// </summary>
        /// <param name="key">链接字符串键</param>
        public DBHelper(string key)
        {
            if (string.IsNullOrEmpty(key))
            {
                throw new ArgumentNullException("key");
            }
            ConnectionStringSettings css = WebConfigurationManager.ConnectionStrings[key];
            if (css == null)
            {
                throw new InvalidOperationException("未找到指定的链接字符串!");
            }
            this.conStr = css.ConnectionString;
            this.provider = DbProviderFactories.GetFactory(css.ProviderName);
        }

        /// <summary>
        /// 构造函数
        /// </summary>
        /// <param name="conStr">链接字符串</param>
        /// <param name="providerStr">数据源提供程序</param>
        public DBHelper(string conStr, string providerStr)
        {
            if (string.IsNullOrEmpty(conStr))
            {
                throw new ArgumentNullException("conStr");
            }
            if (string.IsNullOrEmpty(providerStr))
            {
                throw new ArgumentNullException("providerStr");
            }
            this.provider = DbProviderFactories.GetFactory(providerStr);
            this.conStr = conStr;
        }

        #endregion

        #region 外部方法

        /// <summary>
        /// 执行命令,返回受影响行数
        /// </summary>
        /// <param name="commandType">命令类型</param>
        /// <param name="sql">sql语句或存储过程名称</param>
        /// <param name="parameters">参数数组</param>
        /// <returns>受影响行数,失败返回-1</returns>
        public virtual int ExecuteNonQuery(CommandType commandType, string sqlOrProcName, IEnumerable<DbParameter> parameters)
        {
            DbConnection con = CreateConnection();
            DbCommand cmd = CreateCommand(con, commandType, sqlOrProcName, parameters);
            try
            {
                con.Open();
                return cmd.ExecuteNonQuery();
            }
            finally
            {
                cmd.Dispose();
                con.Dispose();
            }
        }

        /// <summary>
        /// 执行命令,返回第一行第一列对象
        /// </summary>
        /// <param name="commandType">命令类型</param>
        /// <param name="sql">sql语句或存储过程名称</param>
        /// <param name="parameters">参数数组</param>
        /// <returns>执行结果</returns>
        public virtual object ExecuteScalar(CommandType commandType, string sqlOrProcName, IEnumerable<DbParameter> parameters)
        {
            DbConnection con = CreateConnection();
            DbCommand cmd = CreateCommand(con, commandType, sqlOrProcName, parameters);
            try
            {
                con.Open();
                return cmd.ExecuteScalar();
            }
            finally
            {
                cmd.Dispose();
                con.Dispose();
            }
        }

        /// <summary>
        /// 执行命令返回DataSet
        /// </summary>
        /// <param name="commandType">命令类型</param>
        /// <param name="sql">sql语句或存储过程名称</param>
        /// <param name="parameters">参数数组</param>
        /// <returns>DataSet</returns>
        public virtual DataSet GetDataSet(CommandType commandType, string sqlOrProcName, IEnumerable<DbParameter> parameters)
        {
            DbConnection con = CreateConnection();
            DbCommand cmd = CreateCommand(con, commandType, sqlOrProcName, parameters);
            DataSet set = new DataSet();
            DbDataAdapter adapter = this.provider.CreateDataAdapter();
            try
            {
                con.Open();
                adapter.SelectCommand = cmd;
                adapter.Fill(set);
                return set;
            }
            finally
            {
                adapter.Dispose();
                cmd.Dispose();
                con.Dispose();
            }
        }

        /// <summary>
        /// 执行命令返回DbDataReader
        /// </summary>
        /// <param name="commandType">命令类型</param>
        /// <param name="sql">sql语句或存储过程名称</param>
        /// <param name="parameters">参数数组</param>
        /// <param name="action">委托</param>
        /// <returns>对象列表</returns>
        public virtual List<T> ExecuteReader<T>(CommandType commandType, string sqlOrProcName, IEnumerable<DbParameter> parameters,
            Func<DbDataReader, T> action)
        {
            DbConnection con = CreateConnection();
            DbCommand cmd = CreateCommand(con, commandType, sqlOrProcName, parameters);
            DbDataReader reader = null;
            List<T> result = new List<T>();
            try
            {
                con.Open();
                reader = cmd.ExecuteReader();                
                while (reader.Read())
                {
                    var item = action(reader);
                    result.Add(item);
                }
                return result;               
            }
            finally
            {
                if (reader != null)
                {
                    reader.Dispose();
                }
                cmd.Dispose();
                con.Dispose();
            }
        }

        /// <summary>
        /// 批量执行sql语句
        /// </summary>
        /// <param name="sqlList">sql语句集合</param>
        /// <param name="paramList">参数数组集合</param>
        /// <returns>执行成功或失败</returns>
        public virtual bool ExecuteSqlBatchByTrans(IEnumerable<string> sqlList, IEnumerable<List<DbParameter>> paramList)
        {
            DbConnection con = CreateConnection();
            DbCommand cmd = CreateCommand(con, CommandType.Text);
            DbTransaction trans = null;
            try
            {
                con.Open();
                trans = con.BeginTransaction();
                cmd.Transaction = trans;
                int length = sqlList.Count();
                IEnumerable<DbParameter> parameters = null;
                for (int i = 0; i < length; i++)
                {
                    cmd.CommandText = sqlList.ElementAt<string>(i);
                    cmd.Parameters.Clear();
                    parameters = paramList.ElementAt<List<DbParameter>>(i);
                    foreach (DbParameter pm in parameters)
                    {
                        cmd.Parameters.Add(pm);
                    }
                    cmd.ExecuteNonQuery();
                }
                trans.Commit();
                return true;
            }
            catch
            {
                if (trans != null)
                {
                    trans.Rollback();
                }
                throw;
            }
            finally
            {
                if (trans != null)
                {
                    trans.Dispose();
                }
                cmd.Dispose();
                con.Dispose();
            }
        }

        #endregion

        #region CreateDbParameter

        public DbParameter CreateDbParameter(string name, object value)
        {
            DbParameter parameter = this.provider.CreateParameter();
            parameter.ParameterName = name;
            parameter.Value = value;
            return parameter;
        }

        public DbParameter CreateDbParameter(string name, object value, ParameterDirection direction)
        {
            DbParameter parameter = this.provider.CreateParameter();
            parameter.ParameterName = name;
            parameter.Value = value;
            parameter.Direction = direction;
            return parameter;
        }

        public DbParameter CreateDbParameter(string name, object value, int size)
        {
            DbParameter parameter = this.provider.CreateParameter();
            parameter.ParameterName = name;
            parameter.Value = value;
            parameter.Size = size;
            return parameter;
        }

        public DbParameter CreateDbParameter(string name, object value, int size, DbType type)
        {
            DbParameter parameter = this.provider.CreateParameter();
            parameter.ParameterName = name;
            parameter.Value = value;
            parameter.Size = size;
            parameter.DbType = type;
            return parameter;
        }

        public DbParameter CreateDbParameter(string name, object value, int size, DbType type, ParameterDirection direction)
        {
            DbParameter parameter = this.provider.CreateParameter();
            parameter.ParameterName = name;
            parameter.Value = value;
            parameter.Size = size;
            parameter.DbType = type;
            parameter.Direction = direction;
            return parameter;
        }

        #endregion

        #region 私有方法

        /// <summary>
        /// 获取链接实例
        /// </summary>
        /// <returns>链接实例</returns>
        private DbConnection CreateConnection()
        {
            DbConnection con = this.provider.CreateConnection();
            con.ConnectionString = this.conStr;
            return con;
        }

        /// <summary>
        /// 获取命令实例
        /// </summary>
        /// <param name="con">链接实例</param>
        /// <param name="commandType">命令类型</param>
        /// <param name="sqlOrProcName">sql语句或存储过程名称</param>
        /// <returns>命令实例</returns>
        private DbCommand CreateCommand(DbConnection con, CommandType commandType, string sqlOrProcName, IEnumerable<DbParameter> parameters)
        {
            DbCommand cmd = InitCommand(con, commandType, parameters);
            cmd.CommandText = sqlOrProcName;
            return cmd;
        }

        /// <summary>
        /// 获取命令实例
        /// </summary>
        /// <param name="con">链接实例</param>
        /// <param name="commandType">命令类型</param>
        /// <returns>命令实例</returns>
        private DbCommand CreateCommand(DbConnection con, CommandType commandType)
        {
            return InitCommand(con, commandType, null);
        }

        /// <summary>
        /// 初始化命令
        /// </summary>
        /// <param name="commandType">命令类型</param>
        /// <param name="parameters">参数集合</param>
        /// <returns></returns>
        private DbCommand InitCommand(DbConnection con, CommandType commandType, IEnumerable<DbParameter> parameters)
        {
            DbCommand cmd = con.CreateCommand();
            cmd.CommandType = commandType;
            if (parameters != null)
            {
                foreach (DbParameter pm in parameters)
                {
                    cmd.Parameters.Add(pm);
                }
            }
            return cmd;
        }

        #endregion
    }

 

posted @ 2015-09-10 11:20  我是攻城狮  阅读(9659)  评论(43编辑  收藏  举报