.Net框架搭建之辅助模版代码生成工具

在.Net框架搭建讲解中，提到了代码自动生成来减少开发人员人工作量，比如数据库对应的Model、最基层的Repository类和接口等等，除了类名不一样，其他都一样，没必要再手写一遍。

我在这里，介绍使用CodeSmith模版生成Model类，其他的都一样，拿到表名和各字段名、字段类型等等，想怎么生成就能怎么生成。

首先，在硬盘中，建立一个文件夹，用来放模版文件，比如：

F:\pukuimin\InjectExample\ExampleCodeSmithTemplate

点击添加文件夹位置，把这个文件夹加到CodeSmith管理器



好了之后，就可以新建模版文件了，在刚加的文件夹上右键，新建CS类型模版文件。



重命名为 Model.cst

内容：

<%--
Name:
Author: pukuimin
Description:
--%>
<%@ Template Language="C#" TargetLanguage="C#" %>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Import Namespace="SchemaExplorer" %>
<%@ Import Namespace="System.Globalization" %>
<%@ Import Namespace="System.Collections.Generic" %>
<%@ Map Name="CSharpAlias" Src="System-CSharpAlias" Description="数据库字段类型与C#类型的映射表" %>

<%-- 要打印的表 --%>
<%@ Property Name="Table"  Type="SchemaExplorer.TableSchema" DeepLoad="True" Optional="False" Category="需要的数据表" Description="table ." %>
<%
string TableName=Table.Name;
/*
for(int i=TableName.Length-1;i>=0;i--)
{
if(TableName[i]>='A'&& TableName[i]<='Z')
{
TableName=TableName.Substring(0,i)+TableName.Substring(i,1).ToLower()+TableName.Substring(i+1);
break;
}
}
*/
string primaryname=Table.PrimaryKey.MemberColumns[0].Name;
string primarytype= CSharpAlias[Table.PrimaryKey.MemberColumns[0].SystemType.FullName];
string autoname="",autotype="";
foreach(ColumnSchema cs in Table.Columns)
{
if( ((bool)cs.ExtendedProperties["CS_IsIdentity"].Value) == true)
{
autoname=cs.Name;
autotype=CSharpAlias[cs.SystemType.FullName];
break;
}
}
List<string> fkeys=new List<string>();//存储外键列及对应的主键表名
foreach(var item in Table.ForeignKeys)
{
fkeys.Add(item.ForeignKeyMemberColumns[0].Name);
//Response.Write(item.ForeignKeyMemberColumns[0].Name+"--"+item.PrimaryKey.Table.Name+"\n");
}
%>
/* ==============================================================================
* 功能描述：SysUserInfo
* 创 建 者：蒲奎民
* 创建日期：2016-08-29 15:58:13
* CLR Version :4.0.30319.42000
* ==============================================================================*/
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace InjectExample.Model
{
public partial class <%=Table.Name%>
{
<% for(int i=0;i<Table.Columns.Count;i++) {
ColumnSchema col=Table.Columns[i];
string temptype=CSharpAlias[col.SystemType.FullName];
%>
/// <summary>
/// <%=col.Description==""?col.Name:col.Description %>
/// </summary>
public <%=temptype+(col.AllowDBNull&&temptype!="string"?"?":"")%> <%=col.Name%> { get; set; }

<%

}
%>
}
}

保存，然后，点击工具栏的Build Template,生成一下。

右下角会有选择数据库表，我们就需要添加数据库。



添加数据库，再选择库中的一张表。



选择好了之后，点击工具栏的 Generate 生成model内容了。



生成的内容：



这个代码就能直接使用了，如果不能用，要自己改改模版。

其他模版不一一介绍，大家可以参考此模版改造。

再加一个，把模版生成的代码直接保存到文件的CS模版。



模版内容：

<%--
Name:GenerateFiles.cst
Author: pukuimin
Description: 生成并输出cs文件的模版
--%>
<%@ Template Language="C#" TargetLanguage="Text" Inherits="CodeTemplate" Encoding="utf-8"%>
<%@ Assembly Name="SchemaExplorer"%>
<%@ Import Namespace="SchemaExplorer"%>

<%-- 数据库 --%>
<%@ Property Name="SourceDatabase"  Type="SchemaExplorer.DatabaseSchema" DeepLoad="True" Optional="False" Category="需要的数据库" Description="Database"%>
<%-- 注册实体层Model模板 --%>
<%@ Register Name="ModelTemplate" Template="Model.cst" MergeProperties="Flase" ExcludeProperties=""%>
<script runat="template">
//解决方案输出路径
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;
}
}
public string GetSubspace(string tableName)
{

for(int i=tableName.Length-1;i>=0;i--)
{
if(tableName[i]>='A'&& tableName[i]<='Z')
{
tableName=tableName.Substring(0,i)+tableName.Substring(i,1).ToLower()+tableName.Substring(i+1);
break;
}
}
return tableName;
}
//生成实体Entity类
private void GenerateEntityClasses()
{
CodeTemplate modelTemplate =new ModelTemplate();

foreach(TableSchema table in this.SourceDatabase.Tables)
{
string TableName=table.Name;
string subspace=GetSubspace(TableName);
if(TableName!="") continue;//可以指定只生成一个表，不等于这个表名就跳过生成，为空就生成所有表
//#error 上面行修改成要生成的表名（修改后可注释本行）
Response.WriteLine("……………………"+TableName +" 开始生成……………………");

string modelDirectory = OutputDirectory +"\\InjectExample.Model\\"+TableName+".cs";

//生成Model.cs模板文件
modelTemplate.SetProperty("Table",table);
modelTemplate.RenderToFile(modelDirectory,true);//文件输出
Response.WriteLine(modelDirectory +" 创建成功.");

Response.WriteLine("……………………"+TableName +" 完成生成……………………");
}
}
</script>
<%
//创建实体层Entity类
this.GenerateEntityClasses();
Debug.WriteLine("全部生成完成！");
Response.WriteLine("全部生成完成！");
%>

点击生成之后，在你选择的目录下，就直接生成了Model.cs文件，连创建新文件都省了。

版权声明：

作者：真爱无限

出处：http://blog.csdn.net/pukuimin1226/

本文为博主原创文章版权归作者所有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文链接.