上文我们最后虽然用模板创建了一个实体类,但是类的内容仍旧是静态的,这里我们需要用动态方式生成类的内容。因为需要查询数据库这里又免不了各种繁琐的连接数据库操作,为了使我们的编码更加直观,仍然采用C#编码习惯来书写T4代码。
在JSP中,我们可以使用include标签来包含另一个JSP文件。在T4模板中也可以在一个模板文件中包含另一个模板文件。所以我们尽可能把公共模块的代码提取到一个T4模板文件中,方便重复使用。包含指令如下:
<#@ include file="$(ProjectDir)IncludeFile.tt" #>
使用的@ include指令。其中$(ProjectDir)是宏变量,指代当前项目工程路径。
首先创建一个Utility.tt模板文件,该文件主要放一些常用的工具类如DBHelper等,因为工具类只在调用时才执行,而.tt文件在保存时就执行,所以每次保存该文件都会执行里面的代码很繁琐,可以右击该文件【属性】把【自定义工具】中:TextTemplatingFileGenerator属性去掉。这样该模板保存时就不会执行。(注:关于设计时模板和运行时模板的区别和用途这里就不做介绍,这样做主要把问题简化)完整代码如下:
<#+ //数据库操作 public static class DBHelper { private static SqlCommand PrepareCommand(SqlConnection conn, string cmdText, CommandType cmdType, SqlParameter[] parameters) { if (conn.State != ConnectionState.Open) { conn.Open(); } SqlCommand cmd = new SqlCommand(cmdText, conn); cmd.CommandType = cmdType; if (parameters != null) { cmd.Parameters.Clear(); cmd.Parameters.AddRange(parameters); } return cmd; } public static DataSet GetDataSet(string connString,string strSQL) { using (SqlConnection conn = new SqlConnection(connString)) { SqlDataAdapter da = new SqlDataAdapter(strSQL, conn); DataSet ds = new DataSet(); da.Fill(ds); return ds; } } public static DataSet GetSchemaInfo(string connString,string tableName) { StringBuilder sbSQL = new StringBuilder(); sbSQL.AppendLine("SELECT "); sbSQL.AppendLine(" [Id]=C.column_id,"); sbSQL.AppendLine(" [Name]=C.name,"); sbSQL.AppendLine(" [Type]=T.name,"); sbSQL.AppendLine(" [Length]=C.max_length,"); sbSQL.AppendLine(" [Identity]=CASE WHEN C.is_identity=1 THEN N‘T‘ELSE N‘‘ END,"); sbSQL.AppendLine(" [PrimaryKey]=ISNULL(PKInfo.PrimaryKey,N‘‘),"); sbSQL.AppendLine(" [ForeignKey]=CASE WHEN FKInfo.parent_column_id>0 THEN N‘T‘ELSE N‘‘ END,"); sbSQL.AppendLine(" [ForeignKeyTable]=ISNULL(FKInfo.name,N‘‘),"); sbSQL.AppendLine(" [IsNull]=CASE WHEN C.is_nullable=1 THEN N‘T‘ELSE N‘‘ END,"); sbSQL.AppendLine(" [Default]=ISNULL(DC.definition,N‘‘),"); sbSQL.AppendLine(" [ColumnDesc]=ISNULL(EP.value,N‘‘) "); sbSQL.AppendLine("FROM sys.columns C "); sbSQL.AppendLine("INNER JOIN sys.objects O ON C.object_id=o.object_id AND O.type=‘U‘ AND O.is_ms_shipped=0 "); sbSQL.AppendLine("INNER JOIN sys.types T ON C.user_type_id=T.user_type_id "); sbSQL.AppendLine("LEFT JOIN sys.default_constraints DC ON C.object_id=DC.parent_object_id AND C.column_id=DC.parent_column_id AND C.default_object_id=DC.object_id "); sbSQL.AppendLine("LEFT JOIN sys.extended_properties EP ON EP.class=1 AND C.object_id=EP.major_id AND C.column_id=EP.minor_id "); sbSQL.AppendLine("LEFT JOIN (SELECT IC.object_id,IC.column_id,PrimaryKey=CASE WHEN I.is_primary_key=1 THEN N‘T‘ELSE N‘‘ END FROM sys.indexes I INNER JOIN sys.index_columns IC ON I.[object_id]=IC.[object_id] AND I.index_id=IC.index_id)PKInfo ON PKInfo.object_id=C.object_id AND PKInfo.column_id=C.column_id "); sbSQL.AppendLine("LEFT JOIN (SELECT FKC.parent_object_id,FKC.parent_column_id,O.name FROM sys.foreign_key_columns FKC INNER JOIN sys.objects O ON FKC.referenced_object_id=O.object_id)FKInfo ON C.object_id=FKInfo.parent_object_id AND C.column_id=FKInfo.parent_column_id "); sbSQL.AppendFormat("WHERE O.name=‘{0}‘ ORDER BY Id ASC", tableName); return GetDataSet(connString,sbSQL.ToString()); } } //类型转换 public static class TypeConvertor { //将数据库类型映射成C#类型 public static string MapType(string dbType) { if (string.IsNullOrEmpty(dbType)) return dbType; dbType = dbType.ToLower(); string csType = "object"; switch (dbType) { case "char": csType = "string"; break; case "date": csType = "DateTime"; break; case "datetime": csType = "DateTime"; break; case "decimal": csType = "decimal"; break; case "float": csType = "double"; break; case "int": csType = "int"; break; case "money": csType = "decimal";break; case "nchar": csType = "string"; break; case "ntext": csType = "string"; break; case "nvarchar": csType = "string"; break; case "smalldatetime": csType = "DateTime"; break; case "smallint": csType = "short"; break; case "smallmoney": csType = "decimal"; break; case "text": csType = "string"; break; case "time": csType = "TimeSpan"; break; case "varchar": csType = "string"; break; default: csType = "object"; break; } return csType; } } #>
该文件把自动生成的指令标签全部去掉了,只用了<#+ #>标签来包含代码块。该文件中包含两个类:DBHelper和TypeConvertor,为了方便演示我简化了这些类中的代码,只留了需要用到的方法。接下来打开原来的EntityTemplate.tt文件修改代码如下:
<#@ template debug="false" hostspecific="false" language="C#" #> <#@ assembly name="System.Core" #> <#@ assembly name="System.Data" #> <#@ assembly name="System.Xml" #> <#@ import namespace="System.Linq" #> <#@ import namespace="System.Text" #> <#@ import namespace="System.Collections.Generic" #> <#@ import namespace="System.Data" #> <#@ import namespace="System.Data.SqlClient" #> <#@ import namespace="System.Xml" #> <#@ output extension=".cs" #> <#@ include file="$(ProjectDir)Utility.tt" #> <# string connString="Data Source=192.168.1.101;Database=DB_Test;uid=sa;pwd=123"; string tableName="Base_Person"; string nameSpace="EntityGenerator"; #> using System; using System.Collections.Generic; using System.Linq; using System.Text; namespace <#=nameSpace #> { public class <#= tableName #>Entity { <# DataTable dt = DBHelper.GetSchemaInfo(connString,tableName).Tables[0]; foreach(DataRow dr in dt.Rows) { #> /// <summary> /// <#= dr["ColumnDesc"].ToString() #> /// </summary> public <#= TypeConvertor.MapType(dr["Type"].ToString()) #> <#= dr["Name"].ToString() #> { get; set; } <# } #> } }
该文件中使用<#@ include file="$(ProjectDir)Utility.tt" #>来包含上述定义的Utility模板。
<#@ assembly name="System.Core" #> <#@ assembly name="System.Data" #> <#@ assembly name="System.Xml" #> <#@ import namespace="System.Linq" #> <#@ import namespace="System.Text" #> <#@ import namespace="System.Collections.Generic" #> <#@ import namespace="System.Data" #> <#@ import namespace="System.Data.SqlClient" #> <#@ import namespace="System.Xml" #>
这里添加数据库操作类需要使用的程序集和命名空间。
<# string connString="Data Source=192.168.1.101;Database=MicroERP;uid=sa;[email protected]#"; string tableName="Base_Person"; string nameSpace="EntityGenerator"; #>
这里我们把需要用的变量提成全局,以方便后面变更数据库、项目时修改。
<#= #>这和ASP.NET一样为输出表达式。
<# DataTable dt = DBHelper.GetSchemaInfo(connString,tableName).Tables[0]; foreach(DataRow dr in dt.Rows) { #> /// <summary> /// <#= dr["ColumnDesc"].ToString() #> /// </summary> public <#= TypeConvertor.MapType(dr["Type"].ToString()) #> <#= dr["Name"].ToString() #> { get; set; } <# } #>
这里就是连接数据库获取到表结构,再根据表结构,动态生成类的属性,注释,以及数据库类型转化为C#类型。保存该模板效果如下:
这样一个完整的实体类就动态生成了,关于获取数据库表结构,后面将花一整个篇幅来讲解不同数据库怎么提取表结构。估计也有可能放在ORM框架中讲解,这里仅仅先演示其作用。