一、有时我们需要将常用的功能添加到PetaPoco中的Database类中
实现方式有2种,以下以查询字段为例
1.通过扩展方式实现,此方式不改变被调用(Database)类名(只能增加方法)
public static class DatabaseExtension
{
public static T SingleOrDefaultWithWhere<T>(this Database db, string name, object value)
{
string sql = "WHERE " + name + " = @0";
return db.SingleOrDefault<T>(sql, value);
}
}
2.通过继承Database类,以后所有调用都直接调用Db类(不但可以增加方法、还可以对虚方法进行重写[如执行的SQL语句日志等],推荐)
public partial class Db : Database
{
public T SingleOrDefaultWithWhere<T>(this Database db, string name, object value)
{
string sql = "WHERE " + name + " = @0";
return db.SingleOrDefault<T>(sql, value);
}
}
二、有时我们不需要返回实体对象,需要返回一个DataTable格式的数据,这时就需要对Database进行扩展,
以下是返回一个DataTable格式的数据:
public DataTable Table(string sql, params object[] args)
{
OpenSharedConnection();
try
{
using (var cmd = CreateCommand(Connection, sql, args))
{
using (DbDataAdapter dbDataAdapter = Provider.GetFactory().CreateDataAdapter())
{
DataTable dt = new DataTable();
dbDataAdapter.SelectCommand = (DbCommand)cmd;
dbDataAdapter.Fill(dt);
OnExecutedCommand(cmd);
return dt;
}
}
}
catch (Exception x)
{
if (OnException(x))
throw;
return null;
}
finally
{
CloseSharedConnection();
}
}