放假没事干,重做了一下以前做的代码生成工具(在《小小程序员的心得》中提到),正确来说,以前的是CodeSmith的模板,必须安装该软件才能使用,很不爽。
这次用StringTemplate来生成模板,而自己则负责写以前CodeSmith负责做的东西,如获取数据库,表,列的Schema,并封装成一个一个类。
制作这个代码生成工具时,考虑到的和遇到的一些问题总结一下:
1.必需适应各种不同的数据库
a)使用DbProviderFactory来创建各种对象,如DbParameter。想起以前一直都是hardcode SqlParameter。DbProviderFactory会根据数据库驱动的不同,创建不同的对象,如 providerName="System.Data.SqlClient"时,用DbProviderFactory.CreateParameter,会创建SqlParameter类型的对象。
b)把MS的SqlHelper改写(不知道是不是MS写的),全部改写成System.Data.Common下的数据库对象(DbProviderFactory Create出来的对象,用DbType而不用SqlDbType),并只保留一些常用的方法(事实上是全部改写很麻烦),如PrepareParameter,ExecuteReader,ExecuteNonQuery,ExecuteDataSet(ExecuteDataSet还是尽量避免使用的,留下来以防万一)。
c)在获取Schema时遇到的问题-命名规范:Select * from 表名,表名中出现保留字时,不同的数据库可能不一样,比较难解决,不同的数据库解决方法可能不同。SqlServer中需要为表名加上"[]"(如果表名中包含了"["或"]"中任意一个,还需要特殊处理,自己一般不会这样来给表命名,忽略- -)。
d)在获取Schema时遇到的问题2:当SQL2005里面遇到视图如[HumanResouce].[vEmployee],则select * from vEmployee无效,必须要select * from [HumanResouce].[vEmployee],不知道其他数据库是什么情况- -看来要做成通用的,真的很难,还是以后根据实际情况再改吧:)
e)使用DbDataReader.GetSchemaTable才会遇到上述c、d这两个需要写sql才能获取schema的问题。还有一种方法是使用DbConnection.GetSchema来获取字段的信息,但是此方法比较繁琐,本人尚未成功地获取Primary key和identity Column等的详细信息(T_T),有空再研究研究。
f)从数据库中取出的原始信息,为适应各种数据库建立不同TableWrapper和CoumnWrapper(即使用原始信息对不同数据库进行封装),对不同的程语编言建立各种别名(如用CSharpAlias类能够把System.Int32转变成int),对不同的数据库创建不同的sql模板(有待完成)。
2.使用何种模板引擎
这里选择了StringTemplate。为什么?因为学习简单,使用也简单,功能很强大。当使用stringTemplate.SetAttribute("pkColumns", tableWrapper.PkColumns); 后,可以在模板里访问PkColumns里面的public Field和Property。包含判断和循环的语法。支持嵌套模板。
还差一个可视化使用界面,懒,暂时先将就使用一下- -(PS:v1.0a版增加了表的选择界面,数据库连接仍然要手动配置- -)
附上模板的代码:(下载地址)
using
System;
using
System.Configuration;
using
System.Data;
using
System.Data.Common;
using
System.Collections.Generic;
namespace
{
public
class
$table.EntityName$Dao
{
private
static
ConnectionStringSettings dbSettings
=
DataConnection.;
private
static
string
connectionString
=
dbSettings.ConnectionString;
private
static
DbProviderFactory dbFactory
=
DbProviderFactories.GetFactory(dbSettings.ProviderName);
public
static
List
<
$table.EntityName$
>
GetAll()
{
var list
=
new
List
<
$table.EntityName$
>
();
DbDataReader dr
=
null
;
try
{
dr
=
SqlHelper.ExecuteReader(dbFactory, connectionString
, CommandType.Text
,
"
select $allColumns:{$it.QualifiedColumnName$};separator=
"
,
"
$ from $table.QualifiedTableName$
"
);
while
(dr
!=
null
&&
dr.Read())
{
var obj
=
new
$table.EntityName$();
$allColumns:{obj.$it.PropName$
=
Field.Get$it.SystemTypeShortName$(dr,
"
$it.Name$
"
);};separator
=
"
\r\n
"
$
list.Add(obj);
}
}
finally
{
if
(dr
!=
null
)
{
dr.Close();
dr.Dispose();
}
}
return
list;
}$
if
(table.IsTable)$
public
static
$table.EntityName$ Get($pkColumns:{$it.SystemType$ $it.VarName$};separator
=
"
,
"
$)
{
$table.EntityName$ obj
=
null
;
DbDataReader dr
=
null
;
try
{
var paras
=
new
DbParameter[]{
$pkColumns:{SqlHelper.PrepareParameter(dbFactory,
"
$it.Name$
"
, $it.DBType$, $it.VarName$)};separator
=
"
,\r\n
"
$
};
dr
=
SqlHelper.ExecuteReader(dbFactory, connectionString
, CommandType.Text
,
"
select $allColumns:{$it.QualifiedColumnName$};separator=
"
,
"
$ from $table.QualifiedTableName$ where $pkColumns:{$it.QualifiedColumnName$=@$it.PropName$};separator=
"
and
"
$
"
, paras);
if
(dr
!=
null
&&
dr.Read())
{
obj
=
new
$table.EntityName$();
$allColumns:{obj.$it.PropName$
=
Field.Get$it.SystemTypeShortName$(dr,
"
$it.Name$
"
);};separator
=
"
\r\n
"
$
}
}
finally
{
if
(dr
!=
null
)
{
dr.Close();
dr.Dispose();
}
}
return
obj;
}
public
static
bool
Insert($table.EntityName$ obj)
{
bool
flag
=
false
;
try
{
var paras
=
new
DbParameter[]{
$nonIdColumns:{SqlHelper.PrepareParameter(dbFactory,
"
$it.Name$
"
, $it.DBType$, obj.$it.PropName$)};separator
=
"
,\r\n
"
$
};
SqlHelper.ExecuteNonQuery(dbFactory, connectionString
, CommandType.Text
,
"
insert into $table.QualifiedTableName$($nonIdColumns:{$it.QualifiedColumnName$};separator=
"
,
"
$) values($nonIdColumns:{@$it.PropName$};separator=
"
,
"
$)
"
, paras);
flag
=
true
;
}
catch
(Exception)
{
flag
=
false
;
}
return
flag;
}
public
static
bool
Update($table.EntityName$ obj)
{
bool
flag
=
false
;
try
{
var paras
=
new
DbParameter[]{
$allColumns:{SqlHelper.PrepareParameter(dbFactory,
"
$it.Name$
"
, $it.DBType$, obj.$it.PropName$)};separator
=
"
,\r\n
"
$
};
SqlHelper.ExecuteNonQuery(dbFactory, connectionString
, CommandType.Text
,
"
update $table.QualifiedTableName$ set $updateColumns:{$it.QualifiedColumnName$=@$it.PropName$};separator=
"
,
"
$ where $pkColumns:{$it.QualifiedColumnName$=@$it.PropName$};separator=
"
and
"
$
"
, paras);
flag
=
true
;
}
catch
(Exception)
{
flag
=
false
;
}
return
flag;
}
public
static
bool
Delete($pkColumns:{$it.SystemType$ $it.VarName$};separator
=
"
,
"
$)
{
bool
flag
=
false
;
try
{
var paras
=
new
DbParameter[]{
$pkColumns:{SqlHelper.PrepareParameter(dbFactory,
"
$it.Name$
"
, $it.DBType$, $it.VarName$)};separator
=
"
,\r\n
"
$
};
SqlHelper.ExecuteNonQuery(dbFactory, connectionString
, CommandType.Text
,
"
delete from $table.QualifiedTableName$ where $pkColumns:{$it.QualifiedColumnName$=@$it.PropName$};separator=
"
and
"
$
"
, paras);
flag
=
true
;
}
catch
(Exception)
{
flag
=
false
;
}
return
flag;
}$endif$
}
}