CodeSmith单表生成实体模板与生成多表实体模板

 生成单实体模板：

<%@ Template Language="C#" TargetLanguage="C#" %>
<%@ Assembly Name="SchemaExplorer"%>
<%@ Import Namespace="SchemaExplorer"%>
<%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema"%>

using System;
using System.Collections.Generic;
namespace Model
{
    public class <%=GetClassName()%>Entity
    {
    <%foreach(ColumnSchema column in this.SourceTable.Columns)%>
    <%{%>   
        /// <summary>        
        ///<%=column.Description%>
        /// </summary>
        public <%=GetCSDataType(column)%> <%=ToPascal(column.Name)%>{get;set;}
    <%}%>
    }
}

<script runat="template">
    //Pascal命名法（将首字母大写）
    public string ToPascal(string s)
    {
        return s.Substring(0,1).ToUpper()+s.Substring(1);
    }
    //骆驼命名法（将首字母小写）
    public string ToCamel(string s)
    {
        return s.Substring(0,1).ToLower()+s.Substring(1);
    }
    //得到类的名字（由表名而来）
    public string GetClassName()
    {
        string s=this.SourceTable.Name;//取到表名
        //判断表名是不是以S结尾，如果是去掉S
        if (s.EndsWith("s"))
        {
             return ToPascal(s.Substring(0,s.Length-1));
        }
        return ToPascal(s);
    }
    //得到C#的数据类型（将基本常用的数据类型意逐个转换）
    public static string GetCSDataType(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 override string GetFileName()
    {
        return this.GetClassName()+".cs";
    }
</script>

 生成多表实体模板：

MultipleTemplate.cst

<%@ CodeTemplate Language="C#" TargetLanguage="Text" Src="" Inherits="" Debug="False" Description="Template description here." %>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Import Namespace="SchemaExplorer"%>
<%-- 数据库 --%>
<%@ Property Name="SourceDatabase" type="SchemaExplorer.DatabaseSchema" DeepLoad="True" Optional="False" Category="01. GettingStarted - Required" Description="Database that the tables views, and storedprocedures should be based on. IMPORTANT!!! If SourceTables and SourceViews areleft blank, the Entire Database will then be generated." %>
<%-- 注册实体层Entity模板 --%>
<%@ Register Name="EntityTemplate" Template="Entity.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;
        }
    }
</script>

<script runat="template">
private void GenerateEntityClasses()
{
    Debug.WriteLine("----------------------实体Entity类 生成Start----------------------");
    CodeTemplate Template = new EntityTemplate();
    foreach(TableSchema table in this.SourceDatabase.Tables)
    {
        //string filedirectory = outputdirectory +@"" + table.name + ".cs";
        string FileDirectory = this.GetFileDirectory("Model",table.Name,"");  //创建Model子文件夹
        Template.SetProperty("Table",table);
        //文件输出
        Template.RenderToFile(FileDirectory,true);
        Debug.WriteLine(FileDirectory +" 创建成功.");
    }
    Debug.WriteLine("----------------------实体Entity类 生成End----------------------");
}
    //将字符串首字母转换为大写  
    private string MakeCamel(string value)
    {
        return value.Substring(0, 1).ToUpper() + value.Substring(1);
    }
    
    private string GetFileDirectory(string flolderName,string tabName,string surfix)
    {
        return string.Format("{0}\{1}\{2}{3}.cs",OutputDirectory,flolderName,MakeCamel(tabName),surfix);
    }
</script>
<%
this.GenerateEntityClasses();
Debug.WriteLine("OK");
%>

Entity.cst

<%@ CodeTemplate Language="C#" Inherits="CodeTemplate" TargetLanguage="Text" Description="NetTiers main template." Debug="True" ResponseEncoding="UTF-8"%>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Import Namespace="SchemaExplorer" %>

<%@ Property Name="Table" type="TableSchema" DeepLoad="True" Optional="False" Category="01. Getting Started - Required"Description="Database that the tables views, and stored procedures shouldbe based on. IMPORTANT!!! If SourceTables and SourceViews are left blank, theEntire Database will then be generated." %>
<script runat="template">
    //Pascal命名法（将首字母大写）
    public string ToPascal(string s)
    {
        return s.Substring(0,1).ToUpper()+s.Substring(1);
    }
    //骆驼命名法（将首字母小写）
    public string ToCamel(string s)
    {
        return s.Substring(0,1).ToLower()+s.Substring(1);
    }
    //得到类的名字（由表名而来）
    public string GetClassName()
    {
        string s=this.Table.Name;//取到表名
        //判断表名是不是以S结尾，如果是去掉S
        if (s.EndsWith("s"))
        {
             return ToPascal(s.Substring(0,s.Length-1));
        }
        return ToPascal(s);
    }
    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 "DateTime";
            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>
using System;
using System.Collections.Generic;
using System.Text;
namespace Entity
{
    public class <%=GetClassName()%>Entity
    {
    <% foreach(ColumnSchema col in Table.Columns)
        { %>
            /// <summary>        
            ///<%=col.Description%>
            /// </summary>
            public <%= DataType2CSharpType(col.DataType) %> <%=ToPascal(col.Name) %>{ get;set; }
     <% } %>
     }
 }

 很久不用，再用发现不会连接数据库了，又查了下：

编译过后会在属性窗口出现可以设置的属性

 

SourceTable属性默认为空， 点击右侧的那个三个点按钮，会弹出对话框

点击Data Source右侧的那个两个点按钮

 

点Add按钮新增一个数据库连接

输入Name，Provider Type选择SqlSchemaProvider,点击Connection String 右侧的三点按钮

 

 

 

输入Server Name,可以选择Windows身份认证或者SQL Server 密码认证，然后选择一个数据库，最后点击OK，接着再点OK，回到选择表界面，下拉框中选择刚加入的数据库，会自动列出该库中的所有表

2、设置SourceTable
先执行
Tools-Build
然后在属性杂项一栏中设置SourceTable
选择数据库，再选择表。

Tools  ->generate  生成表实体