调用CodeSmith类库实现代码生成(含源码)

      CodeSmith的作用是不言而喻的,用过的人都会觉得它非常强大.根据自定义模板,快速生成代码.只是我们使用的时候,要在它提供的CodeSmith Studio环境下使用模板,再传入相应参数,最终生成NET的类文件.

      如果我们可以通过编码方式,把CodeSmith功能融入NET代码中,这样就会比较灵活,方便我们控制了.根据下面的操作步骤我们就可以实现.

      先说下环境,我安装的CodeSmith版本是5.1,当然是破解版(你懂的).居说CodeSmith注册费用还是比较昂贵滴.我们需要如下几个DLL文件,这几个文件在安装路径下都可以找到.

CodeSmith.Engine.dll

CodeSmith.Feedback.Client

SchemaExploer.dll

SchemaExploer.SqlSchemaProvider.dll

      把这几个DLL文件引入项目文件之后,接下来就需要制作模板了,我们也可以直接使用CodeSmith自带的模板.打开CodeSmith Studio,右侧就是Template Exploer部分,选择一个模板,右键,选择Open Folder In Window Exploer菜单项,确定好模板路径后,就可以开始编码了.

      在写代码之前,我们还是先看下模板的组成,如图-1所示

图-1

      代码第2句,Property关键字,就表示这个模板的属性,也是需要我们传入的参数,看类型是SchemaExploer.DataBaseSchema,也就是说,我们只要传入这个类型的参数值即可

代码如图-2所示

图-2

      代码非常简单,先要加载模板再编译,然后传入属性值,再输出就OK了.模板编译部分是CompileTemplate方法,如图-3所示

图-3

传入属性值的代码是CodeTemplate.SetProperty;

输出类文件的代码是RenderToFile;

      OK,整个过程结束,看看我们的成果吧,如图-4所示

图-4

      这个模板的作用是生成Db数据库中所有Db表的枚举类.

      编码方式使用CodeSmith还是非常灵活的,比如我想生成数据库中Db表的CreateSQL语句,以前我们可能是自己拼写字符串实现,现在使用CodeSmith模板,就非常简单了.如图-5所示,大家参考这个代码自己试下,是不是非常简单呢?

 图-5

完整代码如下

//引入的命名空间
//using SchemaExplorer;
//using System.CodeDom;
//using System.CodeDom.Compiler;

        /// <summary>
        /// 编译CodeSmith模板
        /// </summary>
        /// <param name="templateName"></param>
        /// <returns></returns>
        public CodeSmith.Engine.CodeTemplate CompileTemplate(string templateName,ref string errors)
        {
            CodeSmith.Engine.CodeTemplate codeTemplate = null;

            //编译器
            CodeSmith.Engine.CodeTemplateCompiler compiler = new CodeSmith.Engine.CodeTemplateCompiler(templateName);
            compiler.Compile();

            //判断是否编译成功
            if (compiler.Errors.Count == 0)
            {
                codeTemplate = compiler.CreateInstance();
            }
            else
            {
                StringBuilder sbErrors = new StringBuilder();
                for (int i = 0; i < compiler.Errors.Count; i++)
                {
                    sbErrors.Append(compiler.Errors[i].ToString());
                }
                errors = sbErrors.ToString();
            }

            return codeTemplate;
        }

        private void btnCodeSmith_Click(object sender, EventArgs e)
        {            
            string errors = string.Empty;
            string templateName = string.Empty;
            CodeSmith.Engine.CodeTemplate codeTemplate = null;

            IDbSchemaProvider provider = new SqlSchemaProvider();
            DatabaseSchema dbSchema = new DatabaseSchema(provider, DevelopToolManager.DbConnectionString);

            //TableEnum模板
            templateName = @"C:\Program Files\Navi.Kernel\CodeSmith Template\TableEnum.cst";
            codeTemplate = this.CompileTemplate(templateName, ref errors);
            if (string.IsNullOrEmpty(errors))
            {
                codeTemplate.SetProperty("SourceDatabase", dbSchema);

                string csFileName = @"I:\待整理文件\tableEnum.cs";
                codeTemplate.RenderToFile(csFileName, true);
            }

            //TableSchema模板
            templateName = @"C:\Program Files\Navi.Kernel\CodeSmith Template\TableSchema.cst";
            codeTemplate = this.CompileTemplate(templateName, ref errors);
            if (string.IsNullOrEmpty(errors))
            {
                TableSchema tableSchema = dbSchema.Tables["AD_AUTHOR"];
                codeTemplate.SetProperty("SourceTable", tableSchema);
                codeTemplate.SetProperty("ScriptCreate", "true");
                //生成字符串
                string codeTotal = codeTemplate.RenderToString();
                MessageBox.Show(codeTotal);
                //生成文件
                string csFileName = @"I:\待整理文件\TableSchema_AD_AUTHOR.cs";
                codeTemplate.RenderToFile(csFileName, true);
            }

        }