安装codesmith以及创建模板
安装参考：https://www.cnblogs.com/xcsn/p/4529569.html
代码步骤：
1.批量生成模板BatchTest.cst
<%@ CodeTemplate Inherits="CodeTemplate" Language="C#" TargetLanguage="Text" Description="NetTiers main template." Debug="True" ResponseEncoding="UTF-8"%> <%-- 注册实体层Entity模板 --%> <%@ Register Name="EntityTemplate" Template="Model.cst" MergeProperties="Flase" ExcludeProperties=""%> <%-- 添加引用 --%> <%@ Import Namespace="System.Collections.Generic" %> <%@ Import Namespace="System.Linq" %>
<%-- 添加引用dll --%> <%@ Assembly Name="SchemaExplorer.SqlSchemaProvider" %> <script runat="template"> public string OutputDirectory=""; </script> <script runat="template"> //生成实体Entity类 private void GenerateEntityClasses() { try { //设置访问数据库地址 var SourceDatabase=new DatabaseSchema(); SourceDatabase.Provider=new SchemaExplorer.SqlSchemaProvider(); SourceDatabase.ConnectionString="Server=.;Initial Catalog=AAA;MultipleActiveResultSets=True;Max Pool Size=512;User Id=sa;Password=123456;"; OutputDirectory =Environment.CurrentDirectory+"\\"+"models\\";//输出文件夹 var mynamespace="WQB.Infrastructure.Data.Entity.DTO";//命名空间 //表名 var tables="TianBaoCallBack,TianbaoSynOrder"; var dictables=tables.ToLower().Split(new string[]{","},StringSplitOptions.RemoveEmptyEntries).ToList(); CodeTemplate Template =new EntityTemplate(); foreach(TableSchema table in SourceDatabase.Tables) { if (dictables.Contains( table.Name.ToLower())==false){ continue; } Debug.WriteLine(table.Name); string FileDirectory = OutputDirectory +"\\"+ table.Name +".cs"; //生成模板 Template.SetProperty("SourceTable",table); Template.SetProperty("Namespace",mynamespace); //文件输出 Template.RenderToFile(FileDirectory,true); Debug.WriteLine(FileDirectory +" 创建成功."); } // System.Diagnostics.Process.Start("explorer.exe", OutputDirectory); } catch(Exception e) {} } </script> <% //创建实体层Entity类 this.GenerateEntityClasses(); Debug.WriteLine("OK"); %> <%--输出提示--%> 生成成功！输出路径：<% Response.Write(OutputDirectory); %>
2.生成模板样例Model.cst
<%-- Name: Database Table Properties Author: lwx Description: Create a list of properties from a database table --%> <%@ CodeTemplate Language="C#" TargetLanguage="C#" Debug="False" Description="Create a list of properties from database table." %> <%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Category="Context" Description="Table that the object is based on." %> <%@ Map Name="CSharpAlias" Src="System-CSharpAlias" Description="System to C# Type Map" %> <%@ Property Name="Namespace" Type="String" Category="Context" Description="Table that the object is based on." %> <%@ Assembly Name="SchemaExplorer" %> <%@ Assembly Name="System.Data" %> <%@ Import Namespace="SchemaExplorer" %> <%@ Import Namespace="System.Data" %> using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading.Tasks; using WQB.Infrastructure.Core; using Chloe.Entity; namespace <%=Namespace %> { /// <summary> /// 实体类：<%= GetClassName(SourceTable) %>/// </summary> [Table("<%= GetClassName(SourceTable) %>")] [Serializable] public class <%= GetClassName(SourceTable) %>_M { <% foreach (ColumnSchema column in SourceTable.Columns) { %> <% string dataType = GetCSharpVariableType(column);string lowerDataType=dataType.ToLower();%> <% string defaultVal="=\"\";"; %> <% if (column.IsPrimaryKeyMember) {%> <% if (column.ExtendedProperties["CS_Default"].Value==""){ %> protected <%= GetPrimaryKeyType(SourceTable) %> <%=GetMemberVariableName(column) %>; <%} else {%> protected <%= GetPrimaryKeyType(SourceTable) %> <%=GetMemberVariableName(column) %>=<%=column.ExtendedProperties["CS_Default"].Value%>; <% } %> <%} else {%> <% if (column.ExtendedProperties["CS_Default"].Value.ToString().TrimStart(new char[1]{'('}).TrimEnd(new char[1]{')'})=="''"){ %> <%if(lowerDataType=="string") defaultVal = "=\"\";"; %> <%else if(lowerDataType.Contains("guid")) defaultVal = @"=Guid.NewGuid();"; %> <%else if(lowerDataType.Contains("datetime")) defaultVal = @";"; %> <%else if(lowerDataType.Contains("datetime")) defaultVal = @";"; %> <%else defaultVal = @"="""";"; %> protected <%= GetCSharpVariableType(column)%> <%=GetMemberVariableName(column) %> <%=defaultVal%> <%} else if (column.ExtendedProperties["CS_Default"].Value.ToString().TrimStart(new char[1]{'('}).TrimStart(new char[1]{'('}).TrimEnd(new char[1]{')'}).TrimEnd(new char[1]{')'})=="0"){%> <%if(lowerDataType=="string") defaultVal = @"=""0"";"; %> <%else if(lowerDataType.Contains("bool")) defaultVal = @"=false;"; %> <%else defaultVal=@"=0;"; %> protected <%= GetCSharpVariableType(column) %> <%=GetMemberVariableName(column) %> <%=defaultVal%> <%} else if (column.ExtendedProperties["CS_Default"].Value.ToString().TrimStart(new char[1]{'('}).TrimStart(new char[1]{'('}).TrimEnd(new char[1]{')'}).TrimEnd(new char[1]{')'})=="1"){%> <%if(lowerDataType=="string") defaultVal = @"=""1"";"; %> <%else if(lowerDataType.Contains("bool")) defaultVal = @"=true;"; %> <%else defaultVal = @"=1;"; %> protected <%= GetCSharpVariableType(column) %> <%=GetMemberVariableName(column) %> <%=defaultVal%> <%} else if (column.ExtendedProperties["CS_Default"].Value.ToString().TrimStart(new char[1]{'('}).TrimStart(new char[1]{'('}).TrimEnd(new char[1]{')'}).TrimEnd(new char[1]{')'})=="2"){%> <%if(lowerDataType=="string") defaultVal = @"=""2"";"; %> <%else defaultVal = @"=2;"; %> protected <%= GetCSharpVariableType(column) %> <%=GetMemberVariableName(column) %> <%=defaultVal%> <%} else if (column.ExtendedProperties["CS_Default"].Value.ToString().TrimStart(new char[1]{'('}).TrimEnd(new char[1]{')'}).TrimEnd(new char[1]{')'}).TrimEnd(new char[1]{'('})=="getdate"){%> <%if(lowerDataType=="string") defaultVal = @"=DateTime.Now.ToString(""yyyy-MM-dd HH:mm:ss"");"; %> <%else defaultVal = @"=DateTime.Now;"; %> protected <%= GetCSharpVariableType(column) %><%if (lowerDataType=="datetime"&&column.Name.ToLower()!="addtime"&&column.Name.ToLower()!="modifytime"&&column.Name.ToLower()!="updatetime"){%>?<%}%> <%=GetMemberVariableName(column) %><%if (defaultVal=="=\"\";")%>;<%else%><%=defaultVal%> <%} else if (column.ExtendedProperties["CS_Default"].Value.ToString().TrimStart(new char[1]{'('}).TrimEnd(new char[1]{')'}).TrimEnd(new char[1]{')'}).TrimEnd(new char[1]{'('})=="newid"){%> <%if(lowerDataType=="string") defaultVal = @"=Guid.NewGuid().ToString();"; %> <%else defaultVal = @"=Guid.NewGuid();"; %> protected <%= GetCSharpVariableType(column) %> <%=GetMemberVariableName(column) %><%if (defaultVal=="=\"\";")%>;<%else%><%=defaultVal%> <%} else {%> <% if (column.ExtendedProperties["CS_Default"].Value==""){ %> <%switch(lowerDataType){ %> <% case "string":%> <% defaultVal = @"="""";"; %> <% break;%> <% case "datetime":%> <% if (column.Name.ToLower()=="addtime"||column.Name.ToLower()=="modifytime"||column.Name.ToLower()=="updatetime") {%> <% defaultVal = "=DateTime.Now;"; %> <%}%> <%else{%> <%}%> <% break;%> <% case "short":%> <% case "short?":%> <% defaultVal = "=0;"; %> <% break;%> <% case "int":%> <% case "int?":%> <% defaultVal = "=0;"; %> <% break;%> <% case "long":%> <% case "long?":%> <% defaultVal = "=0;"; %> <% break;%> <% case "float":%> <% case "float?":%> <% defaultVal = "=0f;"; %> <% break;%> <% case "decimal":%> <% case "decimal?":%> <% defaultVal = "=0M;"; %> <% break;%> <% case "bool":%> <% case "bool?":%> <% defaultVal = "=false;"; %> <% break;%> <% case "byte":%> <% case "byte?":%> <% defaultVal = "=0;"; %> <% break;%> <% case "single":%> <% case "single?":%> <% defaultVal = "=0;"; %> <% break;%> <% default:%> <%break;%> <%}%> protected <%= dataType %><%if (lowerDataType=="datetime"&&column.Name.ToLower()!="addtime"&&column.Name.ToLower()!="modifytime"&&column.Name.ToLower()!="updatetime"){%>?<%}%> <%=GetMemberVariableName(column) %>=null; <% } %> <% else {%> protected <%= dataType %> <%=GetMemberVariableName(column) %><%if (defaultVal=="=\"\";")%>;<%else%><%=column.ExtendedProperties["CS_Default"].Value%> <% }} %> <% }} %> public <%= GetClassName(SourceTable) %>_M() { } public <%= GetClassName(SourceTable) %>_M(<%= GetPrimaryKeyType(SourceTable) %> id) { <%=GetMemberVariableName(SourceTable.PrimaryKey.MemberColumns[0]) %>=id; } <% for (int i = 0; i < SourceTable.Columns.Count; i++) { %> /// <summary> /// <%= SourceTable.Columns[i].Description %> /// </summary> <% if (SourceTable.Columns[i].IsPrimaryKeyMember) {%> [Column(IsPrimaryKey = true)] <%} %> <% if ((bool)SourceTable.Columns[i].ExtendedProperties["CS_IsIdentity"].Value) {%> [AutoIncrement] <%} %> public <%= GetCSharpVariableType(SourceTable.Columns[i]) %><%if (GetCSharpVariableType(SourceTable.Columns[i]).ToLower()=="datetime"&&SourceTable.Columns[i].Name.ToLower()!="addtime"&&SourceTable.Columns[i].Name.ToLower()!="modifytime"&&SourceTable.Columns[i].Name.ToLower()!="updatetime"){%>?<%}%> <%= GetPropertyName(SourceTable.Columns[i]) %> { get {return <%= GetMemberVariableName(SourceTable.Columns[i]) %>;} set {<%= GetMemberVariableName(SourceTable.Columns[i]) %> = value;} } <% if (i < SourceTable.NonPrimaryKeyColumns.Count - 1) Response.Write("\r\n"); %> <% } %> public static <%= GetClassName(SourceTable) %>_M Get<%= GetClassName(SourceTable) %>(<%= GetPrimaryKeyType(SourceTable) %> id) { return new <%= GetClassName(SourceTable) %>_M(id); } } } <script runat="template"> public string GetMemberVariableDeclarationStatement(ColumnSchema column) { return GetMemberVariableDeclarationStatement("protected", column); } public string GetMemberVariableDeclarationStatement(string protectionLevel, ColumnSchema column) { string statement ="protected "; if(column.IsForeignKeyMember) { statement+=column.Name.Substring(0,column.Name.Length-2)+" "+column.Name.Substring(0,column.Name.Length-2)+"Id"; } else { statement += GetCSharpVariableType(column) + " " + GetMemberVariableName(column); } string defaultValue = GetMemberVariableDefaultValue(column); if (defaultValue != "") { statement += " = " + defaultValue; } statement += ";"; return statement; } public string GetReaderAssignmentStatement(ColumnSchema column, int index) { string statement = "if (!reader.IsDBNull(" + index.ToString() + ")) "; statement += GetMemberVariableName(column) + " = "; if (column.Name.EndsWith("TypeCode")) statement += "(" + column.Name + ")"; statement += "reader." + GetReaderMethod(column) + "(" + index.ToString() + ");"; return statement; } public string GetCamelCaseName(string value) { return value.Substring(0, 1).ToLower() + value.Substring(1); } public string GetMemberVariableName(ColumnSchema column) { string propertyName = column.Name; //string memberVariableName = GetCamelCaseName(propertyName); //return memberVariableName; return "_"+propertyName.Substring(0,1).ToLower()+propertyName.Substring(1,propertyName.Length-1); } public string GetPropertyName(ColumnSchema column) { string propertyName = column.Name; //if (propertyName == column.Table.Name + "Name") return "Name"; //if (propertyName == column.Table.Name + "Description") return "Description"; //if (propertyName.EndsWith("TypeCode")) //propertyName =propertyName.Substring(0, propertyName.Length - 4).ToUpper();//propertyName.Substring(0,1).ToUpper()+propertyName.Substring(1,propertyName.Length); // //return propertyName.Substring(0,1).ToUpper()+propertyName.Substring(1,propertyName.Length-1).ToLower(); return propertyName; } public string GetMemberVariableDefaultValue(ColumnSchema column) { switch (column.DataType) { case DbType.Guid: { return "Guid.Empty"; } case DbType.AnsiString: case DbType.AnsiStringFixedLength: case DbType.String: case DbType.StringFixedLength: { return "String.Empty"; } default: { return ""; } } } public string GetCSharpVariableType(ColumnSchema column) { if (column.Name.EndsWith("TypeCode")) return column.Name; 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 string GetReaderMethod(ColumnSchema column) { switch (column.DataType) { case DbType.Byte: { return "GetByte"; } case DbType.Int16: { return "GetInt16"; } case DbType.Int32: { return "GetInt32"; } case DbType.Int64: { return "GetInt64"; } case DbType.AnsiStringFixedLength: case DbType.AnsiString: case DbType.String: case DbType.StringFixedLength: { return "GetString"; } case DbType.Boolean: { return "GetBoolean"; } case DbType.Guid: { return "GetGuid"; } case DbType.Currency: case DbType.Decimal: { return "GetDecimal"; } case DbType.DateTime: case DbType.Date: { return "GetDateTime"; } case DbType.Binary: { return "GetBytes"; } default: { return "__SQL__" + column.DataType; } } } public string GetClassName(TableSchema table) { return table.Name; } public string GetSqlDbType(ColumnSchema column) { switch (column.NativeType) { case "bigint": return "BigInt"; case "binary": return "Binary"; case "bit": return "Bit"; case "char": return "Char"; case "datetime": return "DateTime"; case "decimal": return "Decimal"; case "float": return "Float"; case "image": return "Image"; case "int": return "Int"; case "money": return "Money"; case "nchar": return "NChar"; case "ntext": return "NText"; case "numeric": return "Decimal"; case "nvarchar": return "NVarChar"; case "real": return "Real"; case "smalldatetime": return "SmallDateTime"; case "smallint": return "SmallInt"; case "smallmoney": return "SmallMoney"; case "sql_variant": return "Variant"; case "sysname": return "NChar"; case "text": return "Text"; case "timestamp": return "Timestamp"; case "tinyint": return "TinyInt"; case "uniqueidentifier": return "UniqueIdentifier"; case "varbinary": return "VarBinary"; case "varchar": return "VarChar"; default: return "__UNKNOWN__" + column.NativeType; } } public string GetPrimaryKeyType(TableSchema table) { if (table.PrimaryKey != null) { return GetCSharpVariableType(table.PrimaryKey.MemberColumns[0]); } else { throw new ApplicationException("This template will only work on tables with a primary key."); } } public override string GetFileName() { return this.GetClassName(this.SourceTable) + ".cs"; } public string GetMemberVariableDefaultValue2(ColumnSchema column) { switch (column.DataType) { case DbType.Guid: { return "Guid.Empty"; } case DbType.AnsiString: case DbType.AnsiStringFixedLength: case DbType.String: case DbType.StringFixedLength: { return "String.Empty"; } default: { return ""; } } } </script>
