CodeSmith4-案例
执行子模板
Main.cst
<%@ Template Language="C#" TargetLanguage="Text" %>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Import Namespace="SchemaExplorer" %>
<%@ Property Name="SourceDB" Type="DatabaseSchema" Optional="False" %>
<script runat="template">
public void RenderChild1()
{
CodeTemplateCompiler compiler = new CodeTemplateCompiler(".\\Child.cst");
compiler.Compile();
if (compiler.Errors.Count == 0)
{
CodeTemplate template = compiler.CreateInstance();
//DatabaseSchema SourceDB = new DatabaseSchema(new MySQLSchemaProvider(), @"Data Source=127.0.0.1;User Id=root;Password=xYi7sR2Ivc;Database=meshop_www;Port=33306;");
TableSchema table = SourceDB.Tables["user_info"];
template.SetProperty("SourceTable", table);
//template.Render(Response);
template.RenderToFile(".\\"+table.Name+".cs",true);
}
else
{
for (int i = 0; i < compiler.Errors.Count; i++)
{
Console.Error.WriteLine(compiler.Errors[i].ToString());
}
}
}
</script>
<%RenderChild1(); %>
Child.cst
<%@ Template Language="C#" TargetLanguage="Text" %>
<%@ Property Name="SourceTable" Type="TableSchema" %>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Import Namespace="SchemaExplorer" %>
<script runat="template">
public void Output()
{
foreach(ColumnSchema column in SourceTable.Columns)
{
Response.WriteLine(column.Name);
}
}
</script>
<% Output(); %>
DB Model模板
<%@ Assembly Name="SchemaExplorer" %>
<%@ Import Namespace="SchemaExplorer" %>
<%@ CodeTemplate Language="C#" ResponseEncoding="UTF-8" Description="数据库实体模板"%>
<%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Category="属性" Description="要生成实体的数据表" %>
<%@ Property Name="Namespace" Type="String" Default="Model" Category="内容" Description="命名空间名称" %>
namespace <%=Namespace%>
{
/// <summary>
///<%=SourceTable.Description %>
/// </summary>
[Serializable]
public partial class <%=GetClassName(SourceTable) %>
{
#region 常量
///<summary>
///主键ID
///</summary>
public const string <%=GetConstName(GetPrimaryKeyPropertyName(SourceTable))%> = "<%=GetPrimaryKeyPropertyName(SourceTable)%>";
<%for(int i=0;i<SourceTable.NonPrimaryKeyColumns.Count;i++) {
var column = SourceTable.NonPrimaryKeyColumns[i];
%>
///<summary>
///<%=column.Description %>
///</summary>
public const string <%=GetConstName(GetPropertyName(column)) %> = "<%=GetPropertyName(column) %>";
<%} %>
#endregion
#region 私有字段
///<summary>
///主键ID
///</summary>
private <%=GetPrimaryKeyType(SourceTable)%> <%=GetPrimaryKeyFieldName(SourceTable) %>;
<%for(int i=0;i<SourceTable.NonPrimaryKeyColumns.Count;i++) {
var column = SourceTable.NonPrimaryKeyColumns[i];
%>
private <%=GetDataType(column) %> <%=GetFieldName(column) %><%=GetDefaultValueByDBColumn(column)%>;
<%} %>
#endregion
#region 公共属性
public <%=GetPrimaryKeyType(SourceTable) %> <%=GetPrimaryKeyPropertyName(SourceTable) %> {get; set;}
<%for(int i=0;i<SourceTable.NonPrimaryKeyColumns.Count;i++) {
var column = SourceTable.NonPrimaryKeyColumns[i];
%>
///<summary>
///<%=column.Description %>
///</summary>
public <%=GetDataType(column) %> <%=GetPropertyName(column) %> {get; set;}
<%} %>
#endregion
#region 重写的方法
public override bool Equals(object obj)
{
bool result = false;
if (obj is <%=GetClassName(SourceTable)%>)
{
result = (obj as <%=GetClassName(SourceTable)%>).ID == this.ID;
}
return result;
}
public override int GetHashCode()
{
return base.GetHashCode();
}
public override string ToString()
{
<%
string[] columns=new string[SourceTable.Columns.Count];
for(int i=0;i<SourceTable.Columns.Count;i++) {
var column = SourceTable.Columns[i];
columns[i] = "{"+GetPropertyName(column)+"}";
}
%>
return $"<%=string.Join(",",columns) %>";
}
#endregion
}
}
<script runat="template">
// 获取类名
public string GetClassName(TableSchema table)
{
string newTableName="";
if(table.Name.Contains("_"))
{
var words = table.Name.Split('_');
for(int i=0;i<words.Length;i++)
{
newTableName += ConvertToPascal(words[i]);
}
}
else
{
newTableName=table.Name;
}
return newTableName;
}
// 获取主键类型
public string GetPrimaryKeyType(TableSchema table)
{
return GetCSharpDataTypeByDBColumn(GetPrimaryKey(table));
}
// 获取主键属性名
public string GetPrimaryKeyPropertyName(TableSchema table)
{
return ConvertToPascal(GetPrimaryKey(table).Name);
}
// 获取主键字段名
public string GetPrimaryKeyFieldName(TableSchema table)
{
return "_"+ConvertToCamel(GetPrimaryKey(table).Name);
}
// 根据列对象获得列的类型
public string GetDataType(ColumnSchema column)
{
return GetCSharpDataTypeByDBColumn(column);
}
// 根据列对象获得列的字段名
public string GetFieldName(ColumnSchema column)
{
return "_"+ConvertToCamel(column.Name);
}
// 根据列对象获得列的属性名
public string GetPropertyName(ColumnSchema column)
{
return ConvertToPascal(column.Name);
}
// 获取常量名称
public string GetConstName(string name)
{
return "_"+name.ToUpper()+"_";
}
// 获取主键字段
public MemberColumnSchema GetPrimaryKey(TableSchema table)
{
if (table.PrimaryKey != null)
{
if (table.PrimaryKey.MemberColumns.Count == 1)
{
return table.PrimaryKey.MemberColumns[0];
}
else
{
throw new ApplicationException("此模板只支持单个列的主键");
}
}
else
{
throw new ApplicationException("此模板需要有主键的表");
}
}
//获取字段默认值
public string GetDefaultValueByDBColumn(ColumnSchema column)
{
switch (column.DataType)
{
case DbType.AnsiString:
case DbType.AnsiStringFixedLength:
case DbType.String:
case DbType.StringFixedLength: return "= \"\"";
case DbType.Date:
case DbType.DateTime: return " = new DateTime(1900,1,1)";
default:
return "";
}
}
//获取字段类型
public string GetCSharpDataTypeByDBColumn(ColumnSchema column)
{
switch (column.DataType)
{
case DbType.AnsiString: return "string";
case DbType.AnsiStringFixedLength: return "string";
case DbType.Binary: return "byte[]";
case DbType.Boolean: return "bool";
case DbType.Byte: return "byte";
case DbType.Currency: return "decimal";
case DbType.Date: return "DateTime";
case DbType.DateTime: return "DateTime";
case DbType.Decimal: return "decimal";
case DbType.Double: return "double";
case DbType.Guid: return "Guid";
case DbType.Int16: return "short";
case DbType.Int32: return "int";
case DbType.Int64: return "long";
case DbType.Object: return "object";
case DbType.SByte: return "sbyte";
case DbType.Single: return "float";
case DbType.String: return "string";
case DbType.StringFixedLength: return "string";
case DbType.Time: return "TimeSpan";
case DbType.UInt16: return "ushort";
case DbType.UInt32: return "uint";
case DbType.UInt64: return "ulong";
case DbType.VarNumeric: return "decimal";
default:
return "__UNKNOWN__" + column.NativeType;
}
}
//重写文件名
public override string GetFileName()
{
return GetClassName(SourceTable) + ".cs";
}
//Pascal命名转换
public string ConvertToPascal(string str)
{
return str.Substring(0,1).ToUpper() + str.Substring(1);
}
// 骆驼命名转换
public string ConvertToCamel(string str)
{
return str.Substring(0,1).ToLower() + str.Substring(1);
}
</script>
