星空资讯: FreeSql.Generator命令行代码生成器是如何实现的

当我们使用DB First时，设计好的数据库，我们怎么生成一些实体类、通用的代码、控制器、服务层、Dto呢。今天我来给大家介绍一下FreeSql项目中的一些工具。当然，不使用此ORM的小伙伴也能使用此工具，因为他是通用。

FreeSql

FreeSql 是功能强大的对象关系映射技术(O/RM)，支持 .NETCore 2.1+ 或 .NETFramework 4.0+ 或 Xamarin。

有一个强大的ORM，也方便我们开发一个代码生成器。

一般情况下，我们开发数据库相关的应用，主要分为三种code first、db first、model first

我只用过前二种，

code first，代码优先，数据库都是根据实体类生成，所有的关系，可以是逻辑关联，也可以是物理关联。
DB First: 数据库优先，直接设计表结构，用设计工具生成表，设计主键，外键、索引，关联关系等。

FreeSql.Generator 命令行方式

通过几行命令，就可实现生成项目中通用的代码结构，不需要复制一段代码后修改，加快开发速度，减少重复劳动，少用一根头发。

由于每个人的项目结构，代码位置各不相同，对于ORM来说，不同的业务逻辑各不相同，所以该项目没有相应的模板，相信使用过Razor的同学一定能实现自己的模板。

1-2年前，我和一个学长也写过代码生成器，这里分享一下当时做项目时的一些模板，https://github.com/i542873057/SJNScaffolding/tree/master/SJNScaffolding.RazorPage/Templates,该项目是基于 . NET Core+Razor Page,由于已离职，所以没有继续维护，这些模板都和ABP相关，当时提取了一些通用的功能，单表操作，可以直接生成前后端功能，只需要在word中按统一的格式写好数据字典的文档，直接复制到系统，即可根据空格，定义类型等方式解析字段。

回到FreeSql.Generator 命令行

对于此工具的使用可参考 https://github.com/dotnetcore/FreeSql/wiki/DbFirst
源码位置 https://github.com/dotnetcore/FreeSql/tree/master/Extensions/FreeSql.Generator
前提是本地安装了.net core 3.1 的sdk.

怎么使用呢。

安装 dotnet-tool 生成实体类

dotnet tool install -g FreeSql.Generator

新建目录，在地址栏输入 cmd 快速打开命令窗口，输入命令：

FreeSql.Generator --help

我们可以看到

C:\Users\igeekfan\Desktop\code>FreeSql.Generator --help  ____     ____   __  / __/ ____ ___ ___ / __/ ___ _ / /  / _/ / __// -_)/ -_) _\ \ / _ `/ / /  /_/ /_/ \__/ \__/ /___/ \_, / /_/         /_/ # Github # https://github.com/2881099/FreeSql v1.5.0 使用 FreeSql 快速生成数据库的实体类 更新工具：dotnet tool update -g FreeSql.Generator # 快速开始 # > FreeSql.Generator -Razor 1 -NameOptions 0,0,0,0 -NameSpace MyProject -DB "MySql,Data Source=127.0.0.1;..."  -Razor 1     * 选择模板：实体类+特性  -Razor 2     * 选择模板：实体类+特性+导航属性  -Razor "d:\diy.cshtml" * 自定义模板文件  -NameOptions    * 总共4个布尔值，分别对应：        # 首字母大写        # 首字母大写,其他小写        # 全部小写        # 下划线转驼峰  -NameSpace    * 命名空间  -DB "MySql,Data Source=127.0.0.1;Port=3306;User ID=root;Password=root;Initial Catalog=数据库;Charset=utf8;SslMode=none;Max pool size=2"  -DB "SqlServer,Data Source=.;Integrated Security=True;Initial Catalog=数据库;Pooling=true;Max Pool Size=2"  -DB "PostgreSQL,Host=192.168.164.10;Port=5432;Username=postgres;Password=123456;Database=数据库;Pooling=true;Maximum Pool Size=2"  -DB "Oracle,user id=user1;password=123456;data source=//127.0.0.1:1521/XE;Pooling=true;Max Pool Size=2"  -DB "Sqlite,Data Source=document.db"  -DB "Dameng,server=127.0.0.1;port=5236;user id=2user;password=123456789;database=2user;poolsize=2"        Dameng 是国产达梦数据库  -Filter     Table+View+StoreProcedure        默认生成：表+视图+存储过程        如果不想生成视图和存储过程 -Filter View+StoreProcedure  -Match     正则表达式，只生成匹配的表，如：dbo\.TB_.+  -FileName     文件名，默认：{name}.cs  -Output     保存路径，默认为当前 shell 所在目录        推荐在实体类目录创建 gen.bat，双击它重新所有实体类

更新命令行

dotnet tool update -g FreeSql.Generator

这里lin-cms-dotnetcore这个项目来测试。

数据库表名是下划线，字段也是下划线方式。
-Razor 指定第一个模板
-NameOptions 0,0,0,1 最后一个1，代表下划线转驼峰，满足C#命名规则
-NameSpace 指定了命名空间 LinCms.Core.Entities
-DB 就是数据库的相关配置
mysql 本地地址 127.0.0.1 3306端口用户名 root 密码123456 数据库 lin-cms
-Match book 这样就能只生成book，支持正则表达式，如 -Math lin_user 就会生成以lin_user开头的表。如dbo.TB_.+，会生成以TB开头的表。即只生成匹配的表

执行此命令。

FreeSql.Generator -Razor 1 -NameOptions 0,0,0,1 -NameSpace LinCms.Core.Entities -DB "MySql,Data Source=127.0.0.1;Port=3306;User ID=root;Password=123456;Initial Catalog=lincms;Charset=utf8;SslMode=none;Max pool size=2"

这时候代码已经生成了

其中一个代码生成如下。这些类是partial ，熟悉C#的同学，应该知道，类的定义使用此关键字，我们能在不同的地方为该类扩展。以防止重新同步数据库的结构时，丢失改动的字段。

namespace LinCms.Core.Entities {	[JsonObject(MemberSerialization.OptIn), Table(Name = "book")]	public partial class Book {		/// <summary>		/// 主键Id		/// </summary>		[JsonProperty, Column(Name = "id", IsPrimary = true, IsIdentity = true)]		public long Id { get; set; }		[JsonProperty, Column(Name = "author", DbType = "varchar(20)")]		public string Author { get; set; } = string.Empty;		[JsonProperty, Column(Name = "image", DbType = "varchar(50)")]		public string Image { get; set; } = string.Empty;  //更多xxx	}}

最终效果图如下

此时会生成二个文件
__重新生成.bat,下次重新点击他就能重新生成实体类了。

FreeSql.Generator -Razor "__razor.cshtml.txt" -NameOptions 1,1,0,1 -NameSpace MyProject -DB "MySql,Data Source=127.0.0.1;Port=3306;User ID=root;Password=123456;Initial Catalog=lincms;Charset=utf8;SslMode=none;Max pool size=2" -FileName "{name}.cs"

上面的命令-Razor 指定了这个txt文件 __razor.cshtml.txt

我们可以定义自己的模板，以生成符合自已业务的的代码，从而实现快速开发。

我们可以看下模板中的文件内容,他就是asp.net下的mvc 结构下的razor后端模板渲染,把这个.txt后缀去掉，就很明了了。对于asp.net mvc的razor，我们可以将控制器下方法的值替换掉cshtml中的值。这个过程是有一个类库在帮我们实现的,叫RazorEngine，不过那个是.net framework下的实践。.NET Framework 下的RazorEngine代码生成介绍

@using FreeSql.DatabaseModel;@{var gen = Model as RazorModel;Func<string, string> GetAttributeString = attr => {	if (string.IsNullOrEmpty(attr)) return null;	return string.Concat(", ", attr.Trim('[', ']'));};Func<DbColumnInfo, string> GetDefaultValue = col => { if (col.CsType == typeof(string)) return " = string.Empty;"; return "";};}//xxxnamespace @gen.NameSpace {@if (string.IsNullOrEmpty(gen.table.Comment) == false) {	@:/// <summary>	@:/// @gen.table.Comment.Replace("\r\n", "\n").Replace("\n", "\r\n		/// ")	@:/// </summary>}	[JsonObject(MemberSerialization.OptIn)@GetAttributeString(gen.GetTableAttribute())]	public partial class @gen.GetCsName(gen.FullTableName) {	@foreach (var col in gen.columns) {		if (string.IsNullOrEmpty(col.Coment) == false) {		@:/// <summary>		@:/// @col.Coment.Replace("\r\n", "\n").Replace("\n", "\r\n		/// ")		@:/// </summary>		}		@:@("[JsonProperty" + GetAttributeString(gen.GetColumnAttribute(col)) + "]")		@:public @gen.GetCsType(col) @gen.GetCsName(col.Name) { get; set; }@GetDefaultValue(col)@:	}	}@gen.GetMySqlEnumSetDefine()}

RazorEngine.NetCore

到了.NET Core时代，我看了下FreeSql.Generator用的这个类库RazorEngine.NetCore，实现动态操作cshtml，生成需要的文本。

Razor Engine是基于微软Razor解析的模板引擎，它允许你使用Razor语法构建动态模板，你只需要使用Engine的静态方法，Engine.Razor.RunCompile等。

创建一个控制台应用，然后安装包。

Install-Package RazorEngine.NetCore

using RazorEngine;using RazorEngine.Templating; // For extension methods.string template = "Hello @Model.Name, welcome to RazorEngine!";var result = Engine.Razor.RunCompile(template, "templateKey", null, new { Name = "World" });Console.WriteLine(result);

输出如下内容

Hello World, welcome to RazorEngine!

此处使用的RunCompile方法是扩展方法，您需要引用RazorEngine.Templating命名空间。

The "templateKey" 保持唯一值，比如使用guid值。字符串，并且你可以根据此字符串key重新运行缓存的模板。

如果再次根据此key，可使用原本的模板。

var result = Engine.Razor.Run("templateKey", null, new { Name = "Max" });

会输出如下内容

Hello Max, welcome to RazorEngine!

上面中的RunCompile第三个参数，传null,因为我们第四个参数使用的是匿名类，

根目录创建一个HelloWord.cshtml，要选择属性，->如果较新则复制内容，

Hello @Model.Name, welcome to RazorEngine!

控制台如下代码。

string templateFilePath = "HelloWorld.cshtml";var templateFile = File.ReadAllText(templateFilePath);string templateFileResult = Engine.Razor.RunCompile(templateFile, Guid.NewGuid().ToString(), null, new{ Name = "World"});Console.WriteLine(templateFileResult);

控制台输出

Hello World, welcome to RazorEngine!

using System;namespace OvOv.Razor{ public class CopyRightUserInfo {  public string UserName { get; set; }  public string EmailAddress { get; set; }  public DateTime CreateTime { get; set; }  public string FileRemark { get; set; } }}

根目录创建一个CopyRightTemplate.cshtml，要选择属性，->如果较新则复制内容，

@{ var gen = Model as OvOv.Razor.CopyRightUserInfo;}//=============================================================// 创建人:   @gen.UserName// 创建时间:   @gen.CreateTime// 邮箱：    @gen.EmailAddress//==============================================================

控制台如下代码。

string copyRightTemplatePath = "CopyRightTemplate.cshtml";var copyRightTemplate = File.ReadAllText(copyRightTemplatePath);string copyRightResult = Engine.Razor.RunCompile(copyRightTemplate, Guid.NewGuid().ToString(), typeof(CopyRightUserInfo), new CopyRightUserInfo{ CreateTime = DateTime.Now, EmailAddress = "710277267@qq.com", UserName = "IGeekFan"});Console.WriteLine(copyRightResult);Console.ReadKey();

控制台输出

//=============================================================// 创建人:   IGeekFan// 创建时间:   2020/6/23 18:14:08// 邮箱：    710277267@qq.com//==============================================================

全放到控制台下，输出如下结果。代码生成器最重要的一点解决了，我们就能实现自己的代码生成器，先构建自己的模板，实现输入（命令行，WPF，WEB端及更多），输出（生成文件）。

以上源码已放到示例代码中 https://github.com/luoyunchong/dotnetcore-examples/blob/master/aspnetcore-freesql/OvOv.Razor/Program.cs

源码解析

首先这是一个控制台应用，Main(string[] args)可接收多个参数。

处理无参数，--help
处理args数组，解析出所有的参数，如果没有设置，则为默认值。（处理一些参数异常问题）最重要的是根据-Razor，选定对应的模板。

ArgsRazor=""//根据-Razor 1 不是2 还是模板的路径，取出的模板文本值。var razorId = Guid.NewGuid().ToString("N");RazorEngine.Engine.Razor.Compile(ArgsRazor, razorId);

根据数据库连接串，取出参数过滤后的表，视图，存储过程
循环数据库的表等，

model为模板中需要的数据
razorId与上文的razorId相同，
sw为生成后的文本保存的值。

var sw = new StringWriter();var model = new RazorModel(fsql, ArgsNameSpace, ArgsNameOptions, tables, table);RazorEngine.Engine.Razor.Run(razorId, sw, null, model);

将sw字符串保存生成类.cs文件（根据参数配置生成文件名）
另外生成一个__重新生成.bat,__razor.cshtml.txt，方便后续用户重新生成实体类。

FreeSql.Tools

这是 FreeSql 衍生出来的辅助工具包，内含生成器等功能；作者：mypeng1985
因为这个不兼容mac,linux,所以作者建议使用dotnet-tool 命令行工具生成实体类，从而支持MAC/Linux系统。对于不是使用FreeSql的开发者，也能使用此工具，你只需要修改对应的模板即可。

使用方式：不多介绍。

https://github.com/2881099/Freesql.tools
分为WPF ，WinForm + DSkin 版本（套网页）
看了下代码，底层生成代码逻辑也是用的RazorEngine .NET Framework 下的RazorEngine代码生成介绍

预览图

FreeSql.Generator命令行代码生成器是如何实现的

星空资讯

2020年6月23日星期二