CodeSmith模板程式碼生成實戰詳解

張龍豪發表於2015-12-03
MIT

前言

公司專案是基於soa面向服務的架構思想開發的，專案分解眾多子專案是必然的。然而子專案的架子結構種類也過多的話，就會對後期的開發維護產生一鍋粥的感覺。為了儘可能的在結構層避免出現這種混亂的現象，我們就做了一個決定，使用一個統一的架子結構，讓專案管理變的簡單起來。

這樣一來，結構中各層就會有很多重複的程式碼或者重複的邏輯出現，為啦提高開發效率，節約開發時間，我們採用了codesmith根據自定義模板，生成程式碼功能。讓單表的增刪改查功能從資料訪問層到ui展示層一鍵批量生成。下面就開始我的codeSmith模板編寫歷程回顧。

CodeSmith安裝下載

官網地址：http://www.codesmithtools.com

下載地址：http://www.codesmithtools.com/downloads

我使用的，帶破解註冊工具的codesmith連結：http://pan.baidu.com/s/1dDdndsd。

傻瓜式安裝，不做介紹。只不過你安裝完需要很多碼。那麼煩啦，就用我百度雲裡面的。帶註冊軟體，安裝完之後，不要急於開啟codesmith，先去用註冊軟體註冊下。

安裝完成，破解成功。

開啟codesmith主介面如下。

 

Note：開啟新建Csharp template，然後字尾名為cst的就是模板檔案，自己寫的模板程式碼，就在這種字尾格式的檔案中。然後游標放在模板檔案中，F5即可生成你要程式碼的檔案。

寫自己的codesmith模板程式碼。

1、自定義引數模板

Note:從這裡我們能看到引數的宣告，與基本語法的使用規則，需帶<%%>。熟悉之後，在右下角給引數賦值，然後游標放入模板中，點選f5生成程式碼，看下，推敲下。

2、遍歷資料庫中表的模板

 

Note：圖片展示的是怎麼設定資料庫配置

模板程式碼如下

<%--引入c#模板--%>
<%@ CodeTemplate Language="C#" TargetLanguage="C#" Debug="False" Description="Create an enum of tables." %>
<%--宣告資料庫的引數，在左下角的Database屬性中，選擇要操作的資料庫名稱--%>
<%@ Property Category="Database" Name="SourceDatabase" Type="SchemaExplorer.DatabaseSchema" Optional="False" Description="Database the table enums will come from." %>
<%--引入下面的類庫，運算元據庫必備的。不要糾結加入就行啦。--%>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Import Namespace="SchemaExplorer" %>
<%--SourceDatabase, 是你選擇資料庫的屬性類，涵蓋資料庫的名稱，建立時間，字串連結，描述等等，自己可以點點看 --%>
public enum <%=SourceDatabase.Name %>Tables
{
<%-- 遍歷資料庫中的表集合 --%>
<% for(int x = 0; x < SourceDatabase.Tables.Count; x++) 
{ 
    TableSchema table = SourceDatabase.Tables[x];
    if (x < SourceDatabase.Tables.Count -1)
        //輸出表名，這裡是c#的註釋，不會被寫進生成的程式碼中。\t為換行符。
        Response.WriteLine("\t{0},", table.Name);
    else
        Response.WriteLine("\t{0}", table.Name);
}
%>    
}

3、遍歷資料庫表中的欄位，宣告並使用自定義函式

<%@ 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." %>
<%--引入system型別轉為c#的資料型別的對映字典 --%>
<%@ Map Name="CSharpAlias" Src="System-CSharpAlias" Description="System to C# Type Map" %>
<%--引入下面的類庫，運算元據庫必備的。--%>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Import Namespace="SchemaExplorer" %>
<%--遍歷資料庫表的欄位屬性--%>
<% foreach (ColumnSchema column in this.SourceTable.Columns) {  %>
<%--拼接字串，輸出c#中實體的屬性--%>
public <%= ControlType(CSharpAlias[column.SystemType.FullName]) %> <%= StringUtil.ToPascalCase(column.Name) %>{ get; set; }

<% } %>
<script runat="template">
 //如果型別為int，或datetime型別輸出可空型別
 public string ControlType(object val)
 {
     var ty=val.ToString();
     if(ty=="int")
     {
         return "int?";
     }
     if(ty=="System.DateTime")
     {
         return "System.DateTime?";
     }
     return ty;
 }
</script>

4、批量生成檔案，並指定生成檔案位置

 

程式碼如下

<%@ Template Language="C#" TargetLanguage="Text" %>
<%-- 註冊要生成的模板 --%>
<%@ Register Name="TableEnumTemplate" Template="TableEnum.cst" MergeProperties="Flase" ExcludeProperties=""%>
<%@ Register Name="TableClumTemplate" Template="TableProperties.cst" MergeProperties="Flase" ExcludeProperties=""%>

<%--宣告資料庫的引數，在左下角的Database屬性中，選擇要操作的資料庫名稱--%>
<%@ Property Category="Database" Name="SourceDatabase" Type="SchemaExplorer.DatabaseSchema" Optional="False" Description="Database the table enums will come from." %>
<%--Type資料型別為TableSchema，表明引數Table是一個表物件。--%>
<%@ Property Name="SourceTable" Type="TableSchema" DeepLoad="True" Optional="False" Category="Table" Description="Table Name"%>
<%-- 執行輸出檔案的函式 --%>
<% this.OutPutFile(); %>
<script runat="template">
    //輸出檔案
    private void OutPutFile()
    {
        //生成列舉表名的模板
        CodeTemplate table =new TableEnumTemplate();
        //指定輸出路徑
        string tableFilePath = OutputDirectory +"\\"+ this.SourceDatabase.Name +".cs";
        //給子模板引數賦值
        table.SetProperty("SourceDatabase",this.SourceDatabase);
        table.RenderToFile(tableFilePath,true);
        
        //生成列表表欄位的模板
        CodeTemplate cloumn =new TableClumTemplate();
        //指定輸出路徑
        string cloumnFilePath = OutputDirectory +"\\"+ this.SourceTable.Name +".cs";
         //給子模板引數賦值
        cloumn.SetProperty("SourceTable",this.SourceTable);
        cloumn.RenderToFile(cloumnFilePath,true);
    }
    //解決方案輸出路徑
    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>

資料庫表生成md文件模板

 

<%--目標語言C#--%>
<%@ CodeTemplate Inherits="CodeTemplate" Language="C#" TargetLanguage="Text" Description="" Debug="True" ResponseEncoding="UTF-8"%>

<%--Type資料型別為TableSchema，表明引數Table是一個表物件。--%>
<%@ Property Name="Table" Type="TableSchema" DeepLoad="True" Optional="False" Category="Table" Description="Table Name"%>

<%--引入資料庫操作元件--%>
<%@ Assembly Name="SchemaExplorer"%>
<%@ Import Namespace="SchemaExplorer"%>
### <%=Table.FullName %>表描述

|欄位名|資料型別|是否可空|資料庫型別|長度|描述|
|--|--|--|--|--|--|
<%for(int i=0;i<Table.Columns.Count;i++){%>
|<%=Table.Columns[i].Name.ToString()%>|<%=Table.Columns[i].SystemType.ToString()%>|<%=Table.Columns[i].AllowDBNull?"Yes":"No" %> |<%=Table.Columns[i].NativeType.ToString() %>|<%=Table.Columns[i].Size %>|<%=Table.Columns[i].Description.ToString()%>|
<%}%>

 

　　

 

好啦，就這麼多啦，能滿足我的需求啦。

小結

如果你在看到本文後有什麼疑問，請加入部落格左上角群，一起交流學習。

相關文章