首页 > 代码库 > CodeSmith 初尝试
CodeSmith 初尝试
看了黄聪写的 黄聪：如何使用CodeSmith批量生成代码(原创系列教程)，感觉可以马上自己练练手。
写了两个文件：Database.cst，Table.cst。Database引用table，然后生成数据库中所有表的Model类，代码如下：
Table.cst
//模板第一句?,必?的<%@ CodeTemplate Language="C#" TargetLanguage="Text" Src=http://www.mamicode.com/"" Inherits="" Debug="False" CompilerVersion="v3.5" Description="Template description here." %>//引用程序集<%@ Assembly Name="SchemaExplorer" %>//加载程序集中的命名空间<%@ Import Namespace="SchemaExplorer" %>//定义模板参数<%@ Property Name="Table" Type="SchemaExplorer.TableSchema" Default="" Optional="False" Description=""%>//模板内容using System;using System.Collections.Generic;using System.Linq;using System.Text;namespace Fw.Ts.Domain{ [ModelExtension] public partial class <%=Table.Name%> { <%foreach(ColumnSchema col in Table.Columns) {%> [ModelColumnExtension("<%=Table.Name%>", "<%=col.Name%>", "<%=Table.Name%>.Id")] public <%=this.DataType2CSharpType(col.DataType)%> <%=col.Name%> { get; set; } <%}%> }}//自定义函数，可直接用this引用<script runat="Template"> public string DataType2CSharpType(System.Data.DbType dbType) { switch (dbType) { 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.DateTime2: return "DateTime"; case DbType.DateTimeOffset: 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"; case DbType.Xml: return "string"; default: return "object"; } }</script>
Database.cst
//模板第一句话,必须的<%@ CodeTemplate Language="C#" TargetLanguage="Text" Src=http://www.mamicode.com/"" Inherits="" Debug="False" CompilerVersion="v3.5" Description="Template description here." %>//引用程序集<%@ Assembly Name="SchemaExplorer" %>//加载程序集中的命名空间<%@ Import Namespace="SchemaExplorer" %>//定义模板参数<%@ Property Name="DataSource" Type="SchemaExplorer.DatabaseSchema" Default="" Optional="False" Description=""%>//注册外部的模板，注册成变量，能在当前的模板中直接使用，如 CodeTemplate Template = new EntityTemplate();<%@ Register Name="EntityTemplate" Template="Table.cst" MergeProperties="False" ExcludeProperties="" %>//执行模板<%//运行this.GenerateEntityClasses();Console.WriteLine("OK");%>//自定义函数,可直接用this引用<script runat="template"> private void GenerateEntityClasses() { CodeTemplate Template =new EntityTemplate(); foreach(TableSchema table in this.DataSource.Tables) { string FileDirectory = OutputDirectory +"\\"+ table.Name +".cs"; Template.SetProperty("Table",table); Template.RenderToFile(FileDirectory,true); } } private string Directory = String.Empty; //设置输出路径，特性[Editor(typeof(System.Windows.Forms.Design.FolderNameEditor), typeof(System.Drawing.Design.UITypeEditor))] [Editor(typeof(System.Windows.Forms.Design.FolderNameEditor), typeof(System.Drawing.Design.UITypeEditor))] [Optional, NotChecked] [DefaultValue("")] public string OutputDirectory { get { return Directory; } set { if (value.EndsWith("\\")) value = http://www.mamicode.com/value.Substring(0, value.Length -1); Directory = value; } }</script>
生成结果：
CodeSmith 初尝试
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。