James写的CodeSmith模板（一）

//说明：

1、 以下我写的CodeSmith 模板，可以生成属性类层， 数据访问层。

2、 我只打算把代码贴出来共享，如果想研究如何写，自己研究下咯

3、 本人email: yulin_deng@126.com ，欢迎讨论 James写的CodeSmith模板（一）

 

CodeSmithJames/Main.cst

 

 

 <%-- Name: CodeSmith Author: James Description: CodeSmith主模板，用来呼叫从模板，产生实体层、数据访问层 --%> <%-- 打开菜单Tools--〉Options 3、确认Enable unicode support钩上--%> <%-- 指明模板使用的语言和生成的目标语言. --%> <%@ CodeTemplate Language="C#" Src="CommonSqlCode.cs" Inherits="MoM.Templates.CommonSqlCode" ResponseEncoding="UTF-8" TargetLanguage="Text" Debug="False" Description="主模板，用来呼叫从模板.." %> <%-- 节点DataSource--%> <%@ Property Name="SourceDatabase" Type="SchemaExplorer.DatabaseSchema" Category="DataSource" Description="The Database, the Code will been generated depend on it ." %> <%@ Property Name="NameSpace" Default="FIH" Optional="True" Type="System.String" Category="NameSpaces" Description="NameSpaces." %> <%@ Property Name="YourName" Default="James Deng" Optional="True" Type="System.String" Category="UserInfo" Description="UserInfo." %> <%-- 加载使用访问数据库的组件SchemaExplorer，并声明其使用的命名空间--%> <%@ Assembly Name="SchemaExplorer" %> <%@ Import Namespace="SchemaExplorer" %> <%@ Assembly Name="System.Design" %> <%@ Assembly Name="System.Web" %> <%@ Assembly Name="System.Xml" %> <%@ Import NameSpace="System.IO" %> <%@ Import NameSpace="System.Text" %> <%@ Import NameSpace="System.Text.RegularExpressions" %> <%@ Import NameSpace="System.Diagnostics" %> <%@ Import NameSpace="System.Xml" %> <%@ Import NameSpace="System.Xml.Xsl" %> <%@ Import NameSpace="System.Xml.XPath" %> <%@ Import Namespace="System.Collections" %> <mce:script runat="template"><!-- #region "定义OutputDirectory属性、SourceTables属性、SourceViews属性" private TableSchemaCollection _sourceTables; private ViewSchemaCollection _sourceViews; private string _outputDirectory = String.Empty; //定义OutputDirectory属性 public string OutputDirectory { get { if (_outputDirectory.Length == 0) { return @"E:/CodeSmith" + System.DateTime.Now.Date.ToShortDateString(); } else { return _outputDirectory; } } set { _outputDirectory = value; } } //定义SourceTables属性 [Category("DataSource")] [Description("所依据的所有tables.")] //[CodeTemplateProperty(CodeTemplatePropertyOption.Optional)] public TableSchemaCollection SourceTables { get { if (this._sourceTables != null && this._sourceTables.Count > 0 ) return this._sourceTables; else return null; } set { this._sourceTables = value; } } //定义SourceViews属性 [Category("DataSource")] [Description("所依据的所有views.")] [CodeTemplateProperty(CodeTemplatePropertyOption.Optional)] public ViewSchemaCollection SourceViews { get { if (this._sourceViews != null && this._sourceViews.Count > 0 ) return this._sourceViews; else return null; } set { this._sourceViews = value; } } #endregion #region "定义NameSpace属性、YourName属性" private string _NameSpace; private string _YourName; #endregion // --></mce:script> <mce:script runat="template"><!-- #region 一些Go()函数中调用的方法 /// <summary> /// 创建将要输出的文件夹 /// </summary> public void SafeCreateDirectory(string path) { if (!Directory.Exists(path)) { Directory.CreateDirectory(path); } } /// <summary> /// 定义一个Hashtable的实例,用于装载所有的文件名称 /// </summary> private System.Collections.Hashtable _CodeTemplates = new System.Collections.Hashtable(); /// <summary> /// 所有子模板的字符串数组 /// </summary> private string[] _templatesFileNames = new string[] { "DALayer//base//DALayerBase.cst", "DALayer//DALayer.cst", "EntityLayer//EntityLayer.cst", "SQL//SQL.cst" }; /// <summary> /// 装载所有的子模板 /// </summary> public void LoadTemplates() { foreach(string _templatesFileName in this._templatesFileNames) { //得到本句子模板的模板文件名称 string FileName = System.IO.Path.GetFileName(_templatesFileName); //假如Hashtable的实例中已经包含此文件名称,则跳过 if (_CodeTemplates.Contains(FileName)) { continue; } //Hashtabl实例新增数据记录 _CodeTemplates.Add(FileName, this.CompileTemplate(this.CodeTemplateInfo.DirectoryName + _templatesFileName)); } } /// <summary> /// 得到CodeTemplate实例 /// </summary> public CodeTemplate GetTemplate(string templateType) { return (CodeSmith.Engine.CodeTemplate)_CodeTemplates[templateType]; } public CodeTemplate CompileTemplate(string templateName) { CodeTemplateCompiler compiler = new CodeTemplateCompiler(templateName); compiler.Compile(); if (compiler.Errors.Count == 0) { return compiler.CreateInstance(); } else return null; } #endregion // --></mce:script> <mce:script runat="template"><!-- /// <summary> /// 产生数据实体层、访问层的主函数 /// </summary> public void Go() { // 产生输出路径 string rootPath = OutputDirectory ; string rootPathEntityLayer = OutputDirectory + "//" + "EntityLayer"; string rootPathDALayerBase = OutputDirectory + "//" + "DALayer" + "//Base"; string rootPathDALayer = OutputDirectory + "//" + "DALayer" ; string rootPathSQL = OutputDirectory + "//" + "SQL"; // 装载所有的子模板 LoadTemplates(); // 创建将要输出的文件夹 SafeCreateDirectory(rootPath); SafeCreateDirectory(rootPathEntityLayer); SafeCreateDirectory(rootPathDALayerBase); SafeCreateDirectory(rootPathDALayer); SafeCreateDirectory(rootPathSQL); //创立所有SourceTables的输出文件 for (int i=0; i < this.SourceTables.Count; i++) { TableSchema SourceTable = SourceTables[i]; Response.WriteLine( SourceTable.Name + " List:"); //初始化DALayerBase.cst子模板的变量 this.GetTemplate("DALayerBase.cst").SetProperty("SourceTable", SourceTable); this.GetTemplate("DALayerBase.cst").SetProperty("NameSpace", NameSpace ); this.GetTemplate("DALayerBase.cst").SetProperty("YourName", YourName ); //初始化DALayer.cst子模板的输出路径与输出文件名称 this.GetTemplate("DALayerBase.cst").RenderToFile(rootPathDALayerBase + "//" + GetClassName(SourceTable.Name)+"DAL" + "Base.cs",true); Response.WriteLine( "/t" +SourceTable.Name + " DALayer base class generated ;"); //初始化DALayer.cst子模板的变量 this.GetTemplate("DALayer.cst").SetProperty("SourceTable", SourceTable); this.GetTemplate("DALayer.cst").SetProperty("NameSpace", NameSpace ); this.GetTemplate("DALayer.cst").SetProperty("YourName", YourName ); //初始化DALayer.cst子模板的输出路径与输出文件名称 this.GetTemplate("DALayer.cst").RenderToFile(rootPathDALayer + "//" + GetClassName(SourceTable.Name)+"DAL" + ".cs",true); Response.WriteLine( "/t" +SourceTable.Name + " DALayer class generated ;"); //初始化EntityLayer.cst子模板的变量 this.GetTemplate("EntityLayer.cst").SetProperty("SourceTable", SourceTable); this.GetTemplate("EntityLayer.cst").SetProperty("NameSpace", NameSpace ); this.GetTemplate("EntityLayer.cst").SetProperty("YourName", YourName ); //初始化DALayer.cst子模板的输出路径与输出文件名称 this.GetTemplate("EntityLayer.cst").RenderToFile(rootPathEntityLayer + "//" + GetClassName(SourceTable.Name)+"Attribute" + ".cs",true); Response.WriteLine( "/t" + SourceTable.Name + " EntityLayer class generated ;"); Response.WriteLine("~~~~~~~~~~~~~~~~~~~"); } //创立所有SourceViews的输出文件 //初始化DALayer.cst子模板的变量 //this.GetTemplate("SQL.cst").SetProperty("SourceDatabase", SourceDatabase); //this.GetTemplate("SQL.cst").SetProperty("SourceTables", SourceTables); //this.GetTemplate("SQL.cst").RenderToFile(rootPathSQL + "//" + SourceDatabase.Name + ".sql", true); Response.WriteLine("代码生成完毕. 执行完成时间: " + DateTime.Now.ToShortDateString()+","+DateTime.Now.ToLongTimeString()); } // --></mce:script> <% this.Go(); %>