用StringTemplate重新制作代码生成工具
放假没事干,重做了一下以前做的代码生成工具(在《小小程序员的心得》中提到),正确来说,以前的是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.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$
}
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 分享 3 个 .NET 开源的文件压缩处理库,助力快速实现文件压缩解压功能!
· Ollama——大语言模型本地部署的极速利器
· [AI/GPT/综述] AI Agent的设计模式综述