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

时间:2021-03-24 08:31:13

一、前言

  相信许多人都百度过:“.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
}