在.Net框架搭建讲解中,提到了代码自动生成来减少开发人员人工作量,比如数据库对应的Model、最基层的Repository类和接口等等,除了类名不一样,其他都一样,没必要再手写一遍。
我在这里,介绍使用CodeSmith模版生成Model类,其他的都一样,拿到表名和各字段名、字段类型等等,想怎么生成就能怎么生成。
首先,在硬盘中,建立一个文件夹,用来放模版文件,比如:
F:pukuiminInjectExampleExampleCodeSmithTemplate
点击添加文件夹位置,把这个文件夹加到CodeSmith管理器
好了之后,就可以新建模版文件了,在刚加的文件夹上右键,新建CS类型模版文件。
重命名为 Model.cst
内容:
<%-- Name: Author: pukuimin Description: --%> <%@ Template Language="C#" TargetLanguage="C#" %> <%@ Assembly Name="SchemaExplorer" %> <%@ Import Namespace="SchemaExplorer" %> <%@ Import Namespace="System.Globalization" %> <%@ Import Namespace="System.Collections.Generic" %> <%@ Map Name="CSharpAlias" Src="System-CSharpAlias" Description="数据库字段类型与C#类型的映射表" %> <%-- 要打印的表 --%> <%@ Property Name="Table" Type="SchemaExplorer.TableSchema" DeepLoad="True" Optional="False" Category="需要的数据表" Description="table ." %> <% string TableName=Table.Name; /* for(int i=TableName.Length-1;i>=0;i--) { if(TableName[i]>='A'&& TableName[i]<='Z') { TableName=TableName.Substring(0,i)+TableName.Substring(i,1).ToLower()+TableName.Substring(i+1); break; } } */ string primaryname=Table.PrimaryKey.MemberColumns[0].Name; string primarytype= CSharpAlias[Table.PrimaryKey.MemberColumns[0].SystemType.FullName]; string autoname="",autotype=""; foreach(ColumnSchema cs in Table.Columns) { if( ((bool)cs.ExtendedProperties["CS_IsIdentity"].Value) == true) { autoname=cs.Name; autotype=CSharpAlias[cs.SystemType.FullName]; break; } } List<string> fkeys=new List<string>();//存储外键列及对应的主键表名 foreach(var item in Table.ForeignKeys) { fkeys.Add(item.ForeignKeyMemberColumns[0].Name); //Response.Write(item.ForeignKeyMemberColumns[0].Name+"--"+item.PrimaryKey.Table.Name+" "); } %> /* ============================================================================== * 功能描述:SysUserInfo * 创 建 者:蒲奎民 * 创建日期:2016-08-29 15:58:13 * CLR Version :4.0.30319.42000 * ==============================================================================*/ using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading.Tasks; namespace InjectExample.Model { public partial class <%=Table.Name%> { <% for(int i=0;i<Table.Columns.Count;i++) { ColumnSchema col=Table.Columns[i]; string temptype=CSharpAlias[col.SystemType.FullName]; %> /// <summary> /// <%=col.Description==""?col.Name:col.Description %> /// </summary> public <%=temptype+(col.AllowDBNull&&temptype!="string"?"?":"")%> <%=col.Name%> { get; set; } <% } %> } }
保存,然后,点击工具栏的Build Template,生成一下。
右下角会有选择数据库表,我们就需要添加数据库。
添加数据库,再选择库中的一张表。
选择好了之后,点击工具栏的 Generate 生成model内容了。
生成的内容:
这个代码就能直接使用了,如果不能用,要自己改改模版。
其他模版不一一介绍,大家可以参考此模版改造。
再加一个,把模版生成的代码直接保存到文件的CS模版。
模版内容:
<%-- Name:GenerateFiles.cst Author: pukuimin Description: 生成并输出cs文件的模版 --%> <%@ Template Language="C#" TargetLanguage="Text" Inherits="CodeTemplate" Encoding="utf-8"%> <%@ Assembly Name="SchemaExplorer"%> <%@ Import Namespace="SchemaExplorer"%> <%-- 数据库 --%> <%@ Property Name="SourceDatabase" Type="SchemaExplorer.DatabaseSchema" DeepLoad="True" Optional="False" Category="需要的数据库" Description="Database"%> <%-- 注册实体层Model模板 --%> <%@ Register Name="ModelTemplate" Template="Model.cst" MergeProperties="Flase" ExcludeProperties=""%> <script runat="template"> //解决方案输出路径 private string Directory = String.Empty; [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 = value.Substring(0, value.Length -1); Directory = value; } } public string GetSubspace(string tableName) { for(int i=tableName.Length-1;i>=0;i--) { if(tableName[i]>='A'&& tableName[i]<='Z') { tableName=tableName.Substring(0,i)+tableName.Substring(i,1).ToLower()+tableName.Substring(i+1); break; } } return tableName; } //生成实体Entity类 private void GenerateEntityClasses() { CodeTemplate modelTemplate =new ModelTemplate(); foreach(TableSchema table in this.SourceDatabase.Tables) { string TableName=table.Name; string subspace=GetSubspace(TableName); if(TableName!="") continue;//可以指定只生成一个表,不等于这个表名就跳过生成,为空就生成所有表 //#error 上面行修改成要生成的表名(修改后可注释本行) Response.WriteLine("……………………"+TableName +" 开始生成……………………"); string modelDirectory = OutputDirectory +"\InjectExample.Model\"+TableName+".cs"; //生成Model.cs模板文件 modelTemplate.SetProperty("Table",table); modelTemplate.RenderToFile(modelDirectory,true);//文件输出 Response.WriteLine(modelDirectory +" 创建成功."); Response.WriteLine("……………………"+TableName +" 完成生成……………………"); } } </script> <% //创建实体层Entity类 this.GenerateEntityClasses(); Debug.WriteLine("全部生成完成!"); Response.WriteLine("全部生成完成!"); %>
点击生成之后,在你选择的目录下,就直接生成了Model.cs文件,连创建新文件都省了。