三、创建模板
首先创建一个最简单的模板，用于生成实体类，在右上角的Template Explorer中右键新建一个Csharp的模板Model.cst：
<%@ CodeTemplate Language="C#" TargetLanguage="C#" Debug="False" %> <%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Category="Context" Description="tableName" %> <%@ Property Name="FileNameFormat" Type="String" Category="Context" Default="{0}Model.cs" Description="FileName FormatStr" Optional="True" %> <%@ Property Name="NameSpace" Type="System.String" Default="" Category="NameSpace" Description="NameSpace" Optional="True" %> <%@ Map Name="CSharpAlias" Src="System-CSharpAlias" Description="System to C# Type Map" %> <%@ Assembly Name="SchemaExplorer" %> <%@ Import Namespace="SchemaExplorer" %> using System; namespace <%= string.IsNullOrEmpty(NameSpace)?"":NameSpace + "." %>Wx.Model { public class <%= StringUtil.ToPascalCase(SourceTable.Name) %> { <% foreach (ColumnSchema column in this.SourceTable.Columns) { %> public <%= CSharpAlias[column.SystemType.FullName] %> <%= StringUtil.ToPascalCase(column.Name) %> { get; set; } <% } %> } }
右键点击它选择execute，会弹出对话框让选择一个数据表，选择后点击生成按钮，会生成类似如下代码：
我要的肯定不是每次选择一个表，也不想一个个生成之后复制粘贴到项目中去。而且我也不是只需要这样的一个模板，我要自动生成的有好多。
四、制作用于批量调用的模板
首先仿照上例新建了几个模板，例如生成Controller的、ViewModel的，目录结构如下：
这些都是具体的“子模板”，然后按照上面的需求新建了一个名为Generate.cst的模板。
1 <%@ CodeTemplate Language="C#" TargetLanguage="C#" Debug="True" Description="模板输出" %> 2 3 <%@ Property Name="NameSpace" Type="System.String" Default="" Category="NameSpace" Optional="True" Description="项目的命名空间" %> 4 5 <%@ Property Name="SourceDatabase" Type="DatabaseSchema" Default="" Optional="True" Category="数据源" Description="选择数据库" %> 6 <%@ Property Name="SourceTables" Type="SchemaExplorer.TableSchemaCollection" Default="" Optional="True" Category="数据源" Description="选择数据表，默认空则为全部表" %> 7 8 9 <%@ Property Name="TemplateDirectory" Type="String" Editor="System.Windows.Forms.Design.FolderNameEditor" EditorBase="System.Drawing.Design.UITypeEditor" Optional="False" Category="模板" Description="--模板目录--"%> 10 <%@ Property Name="ExceptPrefix" Type="String" Default="" Optional="True" Category="模板" Description="排除前缀为__的" %> 11 12 <%@ Property Name="OutputDirectory" Type="String" Editor="System.Windows.Forms.Design.FolderNameEditor" EditorBase="System.Drawing.Design.UITypeEditor" Optional="False" Category="输出目录" Description="--输出目录--"%> 13 14 15 <%@ Assembly Name="System.Design" %> 16 <%@ Assembly Name="SchemaExplorer" %> 17 <%@ Import Namespace="SchemaExplorer" %> 18 19 <%@ Import Namespace="System.IO" %> 20 <%@ Import Namespace="System.Xml" %> 21 <%@ Import Namespace="System.Text" %> 22 <%@ Import Namespace="System.Windows.Forms.Design" %> 23 <%@ Import Namespace="System.Text.RegularExpressions" %> 24 <%@ Import Namespace="System.Collections.Specialized" %> 25 <%@ Import Namespace="System.Collections.Generic" %> 26 27 28 29 <script runat="template"> 30 //模板列表 31 Dictionary<CodeTemplate, string> templates = new Dictionary<CodeTemplate, string>(); 32 33 //入口 34 public void Generate() 35 { 36 GetTemplates(TemplateDirectory); //读取模板 37 38 int tableIndex = 0; 39 int templateIndex = 0; 40 TableSchemaCollection tables = SourceTables != null && SourceTables.Count > 0 ? SourceTables : SourceDatabase.Tables; //若未手动选择数据表，则默认为全数据库 41 foreach(TableSchema SourceTable in tables) 42 { 43 tableIndex++; 44 45 Response.Write(string.Format("{0}.Table {1}",tableIndex, SourceTable.Name)); 46 Response.WriteLine(); 47 templateIndex = 0; 48 foreach(var template in templates) 49 { 50 templateIndex++; 51 Response.Write(string.Format(" {0}.template {1}",templateIndex, template.Key.CodeTemplateInfo.FileName)); 52 Response.WriteLine(); 53 template.Key.SetProperty("NameSpace",NameSpace); //设置统一的命名空间 54 template.Key.SetProperty("SourceTable",SourceTable); //传入数据表的名称 55 56 //读取模板的文件命名格式，生成文件名 57 string FileName = template.Key.GetProperty("FileNameFormat") == null ? SourceTable.Name : string.Format(template.Key.GetProperty("FileNameFormat").ToString(),SourceTable.Name); 58 template.Key.RenderToFile(Path.Combine(template.Value, FileName), true); //按照模板的目录层级生成文件 59 } 60 Response.WriteLine(); 61 } 62 } 63 64 private void GetTemplates(string directoryStr) 65 { 66 DirectoryInfo directoryInfo = Directory.CreateDirectory(directoryStr); 67 GetTemplates(directoryInfo); 68 } 69 70 //递归方式读取所有满足要求的模板，记录模板的目录结构，生成代码文件时参照此时的目录结构 71 private void GetTemplates(DirectoryInfo directoryInfo) 72 { 73 foreach (var file in directoryInfo.GetFiles()) 74 { 75 if (!file.Extension.ToLower().Equals(".cst") || (!string.IsNullOrEmpty(ExceptPrefix) && file.Name.StartsWith(ExceptPrefix))) 76 { 77 continue; 78 } 79 80 CodeTemplateCompiler compiler = new CodeTemplateCompiler(file.FullName); 81 compiler.Compile(); //编译子模板 82 if (compiler.Errors.Count == 0) 83 { 84 templates.Add(compiler.CreateInstance(),directoryInfo.FullName.Replace(TemplateDirectory, OutputDirectory)); 85 } 86 else 87 { 88 Response.WriteLine("编译模板" + file.FullName + "错误！"); 89 } 90 } 91 92 foreach (var directory in directoryInfo.GetDirectories()) 93 { 94 GetTemplates(directory); 95 } 96 } 97 </script> 98 99 <% this.Generate(); %>
最上面是各种属性设置：
1. NameSpace: 用于设置项目统一的命名空间前缀。
2. SourceDatabase：可以弹出的对话框中选择已经配置好的数据库。
3. SourceTables：批量选择数据表，可以按住ctrl或shift多选。
4. TemplateDirectory：模板所在目录，支持树状目录结构。
5. ExceptPrefix：排除模板前缀，前缀为这样的模板不会被获取。
6. OutputDirectory：生成文件存储目录，其子目录结构会按照模板目录结构。
此模板文件的入口是Generate()，它首先会调用GetTemplates方法读取属性TemplateDirectory指定的目录下的所有模板，并记录各个模板所在的目录结构。然后判断属性SourceTables是否手动选择了数据表，如果没有则为所有表。
准备工作做好之后，就是遍历数据表、遍历模板，生成对应的代码文件了。具体情况已在代码中注释。
