想要用代码写代码,肯定是绕不开反射的。反射的概念相比都不陌生,只是应用多少就因人而异,今天分享一个代码生成器的思路,仅供参考,不要过分依赖哦。
思路分析
众所周知,利用反射可以在程序运行时获取到任一对象的类型、属性、参数、方法等,并加以调用,利用这些获取到的可以在程序运行时追加各种自定义的功能。以CRUD为例,我们可以利用反射获取到所有的Model并编写代码模板,最终达成用代码生成代码的结果。思路既然能走通,开搞开搞~
编码阶段
首先要确定生成哪些代码。CRUD每天都在做,但大部分都一样,这就导致每天有些时间都在板砖,毫无意义。所以我们需要自动生成全库的CRUD代码。
首先要写一个CRUD的模板代码,考虑到ORM框架太多,这里就以SqlSugar为例:
[HttpGet] public async Task<IActionResult> GetList(int index = 1, int size = 15) { RefAsync<int> count = 0; return PageMsg(await _sqlHelper.DB .Queryable<Models.AD>() .OrderBy(x => x.Weight) .ToPageListAsync(index, size, count), count.Value); } [HttpPost, Role("添加广告")] public async Task<IActionResult> Add([FromForm] Models.AD ad) { Models.AD add = await _sqlHelper.DB .Insertable(ad) .RemoveDataCache() .ExecuteReturnEntityAsync(); return Ok(add); } [HttpPut, Role("修改广告")] public async Task<IActionResult> Cag([FromForm] Models.AD ad) { int result = await _sqlHelper.DB .Updateable(ad) .RemoveDataCache() .ExecuteCommandAsync(); return YesOrNo(result > 0); } [HttpDelete] public async Task<IActionResult> Delete([FromForm] int id) { int result = await _sqlHelper.DB .Deleteable<Models.AD>() .Where(x => x.ID == id) .ExecuteCommandAsync(); return YesOrNo(result > 0); }
能看出这是一组操作AD类的CRUD,接下来就是需要把上面的模板封装起来,AD这种表的名称作为变量输入即可,如果想控制代码文件的路径,也可以作为参数传入。完整版:
public void Make(string tableName, string path) { if (!Directory.Exists(path)) Directory.CreateDirectory(path); using FileStream fs = new FileStream($"{path}/{tableName}Controller.cs", FileMode.Append); System.Text.StringBuilder sb = new System.Text.StringBuilder(); sb.Append("using System.Threading.Tasks;"); sb.Append("\n"); sb.Append("using XXXX.Services.DB;"); sb.Append("\n"); sb.Append("using Microsoft.AspNetCore.Authorization;"); sb.Append("\n"); sb.Append("using Microsoft.AspNetCore.Mvc;"); sb.Append("\n"); sb.Append("using SqlSugar;"); sb.Append("\n"); sb.Append("namespace XXXX.Controllers.v1"); sb.Append("\n"); sb.Append("{"); sb.Append("\n"); sb.Append(" [ApiController, Route(\"v1/[controller]/[action]\"), Authorize]"); sb.Append("\n"); sb.Append($" public class {tableName}Controller : BaseController"); sb.Append("\n"); sb.Append(" {"); sb.Append("\n"); sb.Append(" private readonly SqlHelper _sqlHelper;"); sb.Append("\n"); sb.Append($" public {tableName}Controller(SqlHelper sqlHelper)"); sb.Append("\n"); sb.Append(" {"); sb.Append("\n"); sb.Append(" _sqlHelper = sqlHelper;"); sb.Append("\n"); sb.Append(" }"); sb.Append("\n"); sb.Append(" [HttpGet]"); sb.Append("\n"); sb.Append($" public async Task<IActionResult> GetList(int index = 1, int size = 15)"); sb.Append("\n"); sb.Append(" {"); sb.Append("\n"); sb.Append($" RefAsync<int> count = 0;"); sb.Append("\n"); sb.Append(" return Ok(new{"); sb.Append("\n"); sb.Append($" rows = await _sqlHelper.DB.Queryable<Models.{tableName}>().ToPageListAsync(index, size, count),"); sb.Append("\n"); sb.Append(" total = count.Value"); sb.Append("\n"); sb.Append(" });"); sb.Append("\n"); sb.Append(" }"); sb.Append("\n"); sb.Append(" [HttpPost]"); sb.Append("\n"); sb.Append($" public async Task<IActionResult> Add([FromForm] Models.{tableName} {tableName.ToLower()})"); sb.Append("\n"); sb.Append(" {"); sb.Append("\n"); sb.Append($" Models.{tableName} add = await _sqlHelper.DB.Insertable({tableName.ToLower()}).RemoveDataCache().ExecuteReturnEntityAsync();"); sb.Append("\n"); sb.Append(" return Ok(add);"); sb.Append("\n"); sb.Append(" }"); sb.Append("\n"); sb.Append(" [HttpPut]"); sb.Append("\n"); sb.Append($" public async Task<IActionResult> Cag([FromForm] Models.{tableName} {tableName.ToLower()})"); sb.Append("\n"); sb.Append(" {"); sb.Append("\n"); sb.Append($" int result = await _sqlHelper.DB.Updateable({tableName.ToLower()}).RemoveDataCache().ExecuteCommandAsync();"); sb.Append("\n"); sb.Append(" return YesOrNo(result>0);"); sb.Append("\n"); sb.Append(" }"); sb.Append("\n"); sb.Append(" [HttpDelete]"); sb.Append("\n"); sb.Append($" public async Task<IActionResult> Cag([FromForm] int id)"); sb.Append("\n"); sb.Append(" {"); sb.Append("\n"); sb.Append($" int result = await _sqlHelper.DB.Deleteable<Models.{tableName}>().Where(x => x.ID == id).ExecuteCommandAsync();"); sb.Append("\n"); sb.Append(" return YesOrNo(result>0);"); sb.Append("\n"); sb.Append(" }"); sb.Append("\n"); sb.Append(" }"); sb.Append("\n"); sb.Append("}"); using StreamWriter sw = new StreamWriter(fs); sw.WriteLine(sb.ToString()); }
为了方便理解,刻意美化了一下。
OK,代码模板搞好了,该从哪里拿到所有的Model呢?思路是先加载程序集,然后找到存放Model的命名空间,然后找到命名空间下面所有符合条件的Class,然后就可以拿到具体的名称,从而调用代码模板进行文件生成,像这样:
var assembly = Assembly.Load("程序集名称"); var types = assembly.ExportedTypes.Where(a => a.FullName.Contains("Model所处的命名空间")).ToList(); //指定 string path = $"{Directory.GetCurrentDirectory()}/autoCode"; foreach (var item in types) { string tableName = item.FullName.Split('.')[^1]; Make(tableName, path); }
接下来直接运行,走完以后的结果是这样的:
随便打开一个文件,是这样的:
效果还是不错的。
如果需要更加精细的控制,可以预先在Model设置特性,然后通过判断特性是否存在来决定如何生成代码,举个栗子:
var assembly = Assembly.Load("程序集名称"); var types = assembly.ExportedTypes.Where(a => a.FullName.Contains("Model所处的命名空间")).ToList(); //指定 string path = $"{Directory.GetCurrentDirectory()}/autoCode"; foreach (var item in types) { //获取所有公开属性 foreach (var prop in item.GetProperties()) { MyAttribute attr = method.GetCustomAttribute(typeof(MyAttribute), true) as MyAttribute; if(attr is null) { //进入这里代表该属性没有附加 MyAttribute } } }
当然,还可以获取MyAttribute的属性拿过来进行判断等等。而实际上很多ORM框架也是用类似手段实现Code First、DB First,不过复杂程度比较高。
这样批量生成代码有点像活字印刷,虽然可以解决大量机械重复的工作,但灵活度还是不够高,人脑还是无法替代的。