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

FreeSql

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

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

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

我只用过前二种，

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

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

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 "";
};
}
//xxx
namespace @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);