CodeSmith批量代码生成并存放到指定目录

其实CodeSmith自带的例子是很好的学习材料，其中一个例子就有批量生成所有表(指定表)并存放到指定目录的，但例子还是归例子，要符合实际需要的使用，还是得改一下的。下面发布下本人修改后的批量生成多种模板多个表的代码文件，并存放到指定目录下：

<%@ CodeTemplate Language="C#" TargetLanguage="C#" Src="" Inherits="" Debug="True" CompilerVersion="v3.5" ResponseEncoding="UTF-8" Description="输出所有模块文件" %>
<%-- Context --%>
<%-- Object --%>
<%@ Property Name="NameSpace" Type="System.String" Default="NameSpace" Category="Object" Description="项目的命名空间" %>
<%@ Property Name="Assembly" Type="System.String" Default="Assembly" Category="Object" Description="项目的程序集" %>
<%@ Property Name="TableDivideMark" Type="System.String" Default="_" Category="Object" Description="表名分隔符" %>
<%@ Property Name="TablePrefix" Type="System.Boolean" Default="true" Category="Object" Description="是否有表前缀，如果存在表前缀的，会自动把第一个表名分隔符前的表前缀删除掉。" %>
<%@ Property Name="TablePrefixLength" Type="System.Int32" Default="4" Category="Object" Description="表前缀长度，如果表前缀大于这个值，则认为没有表前缀" %>
<%@ Property Name="ColumnDivideMark" Type="System.String" Default="_" Category="Object" Description="字段名分隔符" %>
<%@ Property Name="ColumnPrefix" Type="System.Boolean" Default="false" Category="Object" Description="是否有字段前缀，如果存在字段前缀的，会自动把第一个字段名分隔符前的字段前缀删除掉。" %>
<%@ Property Name="ColumnPrefixLength" Type="System.Int32" Default="0" Category="Object" Description="字段前缀长度，如果字段前缀大于这个值，则认为没有字段前缀" %>
<%@ Property Name="ForceId" Type="System.Boolean" Default="true" Category="Object" Description="强制性自增量标识主键" %>
<%@ Property Name="ForceIdProperty" Type="System.String" Default="ID" Category="Object" Description="强制性自增量标识主键的属性名称" %>
<%@ Property Name="SourceTableNames" Type="System.String" Default="" Category="Object" Description="当指定表名时，只执行字符串中表名列表中的表" %>
<%-- Context --%>
<%@ Assembly Name="System.Design" %>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Import Namespace="SchemaExplorer" %>
<%@ Import Namespace="System.IO" %>
<%@ Import Namespace="System.Xml" %>
<%@ Import Namespace="System.Text" %>
<%@ Import Namespace="System.Windows.Forms.Design" %>
<%@ Import Namespace="System.Text.RegularExpressions" %>
<%@ Import Namespace="System.Collections.Specialized" %>

<script runat="template">


    private DatabaseSchema _sourceDatabase;
    private string _outputDirectory;
    private bool _implementNotification = true;
    
    //模板对象列表（根据模板名称产生出来的模板对象）
    public CodeTemplate[] CurrentTemplates;
    //模板名称数组
    public string[] TemplateNames = {"piggy_projectFileInclue.cst","piggy_SqlMap.config.cst","piggy_dao.config.cst","piggy_service.config.cst","piggy_Model_Auto.cst","iBatis_sqlmap_Auto.cst","piggy_Service.cst","piggy_Model.cst","piggy_DAL.cst","piggy_BLL.cst","iBatis_sqlmap.cst"}; 
    //模板输出格式数组
    public string[] strFormats = {"projectFileInclue.txt","sqlmap.config","Dao_Auto.config","Service_Auto.config","{0}_Auto.cs","{0}_Auto.xml","I{0}BLL.cs","{0}.cs","{0}DAO.cs","{0}BLL.cs","{0}.xml"};
    //模板输出路径
    public string[] OutputPaths={"","\\Data","\\Data\\config","\\Data\\config","\\Data_Auto\\Model","\\Data_Auto\\Map","\\Data\\service","\\Data\\Model","\\Data\\DAL","\\Data\\BLL","\\Data\\Map"};
    [Category("Database")]
    [Description("Database that the mapping file should be based on.")]
    public DatabaseSchema SourceDatabase {
        get { return _sourceDatabase; }
        set { _sourceDatabase = value; }
    }

    [Editor(typeof(System.Windows.Forms.Design.FolderNameEditor), typeof(System.Drawing.Design.UITypeEditor))]
    [Category("Class")]
    [Description("The folder to save the generated class files.")]
    public string OutputDirectory
    {
        get {return _outputDirectory;}
        set {_outputDirectory= value;}
    }

    //根据模板名称产生出来的模板对象列表
    public void CreateTemplate()
    {
        if (CurrentTemplates==null)
        {
            CurrentTemplates=new CodeTemplate[TemplateNames.Length];
            for (int i = 0; i < TemplateNames.Length; i++)
            {
                CurrentTemplates[i]=CompileTemplate(CodeTemplateInfo.DirectoryName + TemplateNames[i]);
            }
        }
    }
    //单独生成一个模板对象
    public CodeTemplate CompileTemplate(string templateName)
    {
        CodeTemplateCompiler compiler = new CodeTemplateCompiler(templateName);
        compiler.Compile();

        if (compiler.Errors.Count == 0)
        {
            return compiler.CreateInstance();
        }
        else
        {
            for (int i = 0; i < compiler.Errors.Count; i++)
            {
                Response.WriteLine(compiler.Errors[i].ToString());
            }
            return null;
        }

    }

    //执行输出脚本
    public void Generate()
    {
        if (CurrentTemplates==null)
            CreateTemplate();
        StringCollection ExcludedTables=new StringCollection();
        if (SourceTableNames.Length>0)
        {
            int mIndex=-1;
            mIndex=SourceTableNames.IndexOf(',');
            while (mIndex>0) {
                ExcludedTables.Add(SourceTableNames.Substring(0,mIndex));
                SourceTableNames=SourceTableNames.Remove(0,mIndex+1);
                mIndex=SourceTableNames.IndexOf(',');
            }
            ExcludedTables.Add(SourceTableNames);
        }    
        foreach(TableSchema SourceTable in SourceDatabase.Tables)
        {
            if (SourceTableNames.Length>0)
            {
                if (!ExcludedTables.Contains(SourceTable.Name))
                    continue;
            }
            Response.Write(string.Format("Processing Table {0} ... ", SourceTable.Name));
            Response.WriteLine();
            try
            {
            string className = ClearDivideMarkAndPrefix(SourceTable.Name,TableDivideMark,TablePrefix,TablePrefixLength);
            for (int i = 0; i < TemplateNames.Length; i++)
                {
                string FileName;
                if (i<=3)
                {
                    CurrentTemplates[i].SetProperty("SourceDatabase", SourceDatabase);
                    FileName=strFormats[i];
                }
                else {
                    CurrentTemplates[i].SetProperty("SourceTable", SourceTable);
                    FileName=string.Format(strFormats[i],className);
                }
                FileName=Path.Combine(OutputDirectory+OutputPaths[i], FileName);
                //统一属性赋值
                CurrentTemplates[i].SetProperty("NameSpace",NameSpace);
                CurrentTemplates[i].SetProperty("Assembly",Assembly);
                CurrentTemplates[i].SetProperty("TableDivideMark",TableDivideMark);
                CurrentTemplates[i].SetProperty("TablePrefix",TablePrefix);
                CurrentTemplates[i].SetProperty("TablePrefixLength",TablePrefixLength);
                CurrentTemplates[i].SetProperty("ColumnDivideMark",ColumnDivideMark);
                CurrentTemplates[i].SetProperty("ColumnPrefix",ColumnPrefix);
                CurrentTemplates[i].SetProperty("ColumnPrefixLength",ColumnPrefixLength);
                CurrentTemplates[i].SetProperty("ForceId",ForceId);
                CurrentTemplates[i].SetProperty("ForceIdProperty",ForceIdProperty);
                
                Response.WriteLine(string.Format("{0} In {1} ", TemplateNames[i],FileName));
                CurrentTemplates[i].RenderToFile(FileName, true);
                }
            }
            catch (Exception ex)
            {
                Response.WriteLine("Error: " + ex);
            }
        }
    }
</script>

<% this.Generate();  %>
<!-- #include file="Function.inc" -->

下面是Function.inc文件的部分代码。截取出来的，如果有遗留了哪些代码，请指正：

<script runat="template">
private Regex cleanRegEx = new Regex(@"\s+|_|-|\.", RegexOptions.Compiled);
private Regex cleanID = new Regex(@"(_ID|_id|_Id|\.ID|\.id|\.Id|ID|Id)", RegexOptions.Compiled);

//清除字符串中的分隔符以及第一个分隔符前的字符串长度小于等于prefixLength的前缀
public string ClearDivideMarkAndPrefix(string name,string mark,bool prefix,int prefixLength)
{
    string strResult=name;
    //处理表前缀
    if (prefix)
    {
        int mIndex=name.IndexOf(mark);
        if (mIndex<=prefixLength)
            strResult=name.Remove(0,mIndex+mark.Length);
    }
    if (mark.Length>0)
        strResult=GetFirstUpStr(strResult,mark);
    return strResult;
}
//受字母大写且mark分隔符后第一个字母大写
    public string GetFirstUpStr(string strValue,string mark)
    {
        string m_Str=strValue.ToLower();
        System.Text.StringBuilder mResult=new System.Text.StringBuilder("");
        int mLen=m_Str.Length;
        int j=-1;
        j=m_Str.IndexOf(mark);
        while (j>0)
        {
            mResult.Append(m_Str.Substring(0,1).ToUpper());
            mResult.Append(m_Str.Substring(1,j-1));
            m_Str=m_Str.Remove(0,j+mark.Length);
            j=m_Str.IndexOf(mark);
        }
        if (m_Str.Length>0)
        {
            mResult.Append(m_Str.Substring(0,1).ToUpper());
            mResult.Append(m_Str.Substring(1,m_Str.Length-1));
            }
        return mResult.ToString();
    }
</script>

其实，这个批量输出的可以重复使用的，只需要把模板拷贝到对应的模板目录中后，修改变量TemplateNames、strFormats、OutputPaths为自己实际需要的参数就能应用到不同的模板库中了，十分方便，如果看官们的模板参数不一样，那就得修改“//统一属性赋值”下的SetProperty相关的属性就可以了。

原创作品出自努力偷懒，转载请说明文章出处：http://www.cnblogs.com/kfarvid/