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); } }
