codesmith自定义模板之实体层

一说到codesmith相信大家有喜有恨的，喜之功能很牛叉，自定义代码模板，老外的技术就是牛叉，比其他的代码工具功能更强大，但是英文版就是我的语言障碍，相信多数codeMan不喜欢英文版得软件，在官方下载正式版，有使用期限，不想的怎么破解它，郁闷，不管了，能用几天是几天吧

先我们看看截图打开netiers类这是代码工厂必备的类，只有了解它才能生产自己的代码

接下里我要生产实体层代码。思路是：先找到数据源，找到表结构，然后查找列，以及相关属性，然后根据模板循环生产，输入到相关目录下

先定义一个主模板，功能是选择数据源和，选择保存路径，并且在里面注册生产实体层的子模板

先贴主模板代码：

<%@ CodeTemplate  Inherits="CodeTemplate" Language="C#" TargetLanguage="Text"
Description="NetTiers main template." Debug="True" ResponseEncoding="UTF-8"%>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Assembly Name="System.Design" %>
<%@ Assembly Name="System.DirectoryServices" %>
<%@ Assembly Name="System.Web" %>
<%@ Assembly Name="System.Xml" %>

<%@ Import Namespace="SchemaExplorer" %>
<%@ Import NameSpace="System.IO" %>
<%@ Import NameSpace="System.Text" %>
<%@ Import NameSpace="System.Text.RegularExpressions" %>
<%@ Import NameSpace="System.Diagnostics" %>
<%@ Import NameSpace="System.Xml" %>
<%@ Import NameSpace="System.Xml.Xsl" %>
<%@ Import NameSpace="System.Xml.XPath" %>
<%-- 注册实体层模板 --%>
<%@ Register Name="EntityClassTemplate" Template="BeingNet.Entity.cst" MergeProperties="False" ExcludeProperties="" %>

<%-- 1. Datasource --%>
<%@ Property Name="ChooseSourceDatabase" Type="SchemaExplorer.DatabaseSchema" DeepLoad="True" Optional="False" Category="01. Getting Started - Required" Description="Database that the tables views, and stored procedures should be based on. IMPORTANT!!! If SourceTables and SourceViews are left blank, the Entire Database will then be generated." %>

<% this.SaveEntityClasses();
%>

<script runat="template">

    private string templateOutputDirectory=@"E:\";

    [Editor(typeof(System.Windows.Forms.Design.FolderNameEditor), typeof(System.Drawing.Design.UITypeEditor))]
    [Optional, NotChecked]
    [Category("01. Getting Started - Required")]
    [Description("The directory to output the results to.")]
    [DefaultValue("")]
    public string OutputDirectory
    {
        get
        {
             return templateOutputDirectory;
        }
        set
        {
            if (value.EndsWith("\\")) value = value.Substring(0, value.Length - 1);
            templateOutputDirectory = value;
        }
    }

    private void SaveEntityClasses()
    {
          CodeTemplate entitytemplate = new EntityClassTemplate();
            foreach(TableSchema table in this.ChooseSourceDatabase.Tables)
            {
                  entitytemplate.SetProperty("CurrentTable",table);
                entitytemplate.RenderToFile(this.templateOutputDirectory + "\\"  + table.Name+".cs",true);

                Debug.WriteLine(table.Name);
            }

    }

</script>

下面是实体层模版代码：