SysDatabaseService.cs 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776
  1. // Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
  2. //
  3. // 本项目主要遵循 MIT 许可证和 Apache 许可证(版本 2.0)进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
  4. //
  5. // 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任!
  6. using Npgsql;
  7. namespace Admin.NET.Core.Service;
  8. /// <summary>
  9. /// 系统数据库管理服务 🧩
  10. /// </summary>
  11. [ApiDescriptionSettings(Order = 250)]
  12. public class SysDatabaseService : IDynamicApiController, ITransient
  13. {
  14. private readonly ISqlSugarClient _db;
  15. private readonly IViewEngine _viewEngine;
  16. private readonly CodeGenOptions _codeGenOptions;
  17. public SysDatabaseService(ISqlSugarClient db,
  18. IViewEngine viewEngine,
  19. IOptions<CodeGenOptions> codeGenOptions)
  20. {
  21. _db = db;
  22. _viewEngine = viewEngine;
  23. _codeGenOptions = codeGenOptions.Value;
  24. }
  25. /// <summary>
  26. /// 获取库列表 🔖
  27. /// </summary>
  28. /// <returns></returns>
  29. [DisplayName("获取库列表")]
  30. public List<VisualDb> GetList()
  31. {
  32. return App.GetOptions<DbConnectionOptions>().ConnectionConfigs.Select(u => new VisualDb { ConfigId = u.ConfigId.ToString(), DbNickName = u.DbNickName }).ToList();
  33. }
  34. /// <summary>
  35. /// 获取可视化库表结构 🔖
  36. /// </summary>
  37. /// <returns></returns>
  38. [DisplayName("获取可视化库表结构")]
  39. public VisualDbTable GetVisualDbTable()
  40. {
  41. var visualTableList = new List<VisualTable>();
  42. var visualColumnList = new List<VisualColumn>();
  43. var columnRelationList = new List<ColumnRelation>();
  44. var dbOptions = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.First(u => u.ConfigId.ToString() == SqlSugarConst.MainConfigId);
  45. // 遍历所有实体获取所有库表结构
  46. var random = new Random();
  47. var entityTypes = App.EffectiveTypes.Where(u => !u.IsInterface && !u.IsAbstract && u.IsClass && u.IsDefined(typeof(SugarTable), false)).ToList();
  48. foreach (var entityType in entityTypes)
  49. {
  50. var entityInfo = _db.EntityMaintenance.GetEntityInfoNoCache(entityType);
  51. var visualTable = new VisualTable
  52. {
  53. TableName = entityInfo.DbTableName,
  54. TableComents = entityInfo.TableDescription + entityInfo.DbTableName,
  55. X = random.Next(5000),
  56. Y = random.Next(5000)
  57. };
  58. visualTableList.Add(visualTable);
  59. foreach (EntityColumnInfo columnInfo in entityInfo.Columns)
  60. {
  61. var visualColumn = new VisualColumn
  62. {
  63. TableName = columnInfo.DbTableName,
  64. ColumnName = dbOptions.DbSettings.EnableUnderLine ? columnInfo.DbColumnName.ToUnderLine() : columnInfo.DbColumnName,
  65. DataType = columnInfo.PropertyInfo.PropertyType.Name,
  66. DataLength = columnInfo.Length.ToString(),
  67. ColumnDescription = columnInfo.ColumnDescription,
  68. };
  69. visualColumnList.Add(visualColumn);
  70. // 根据导航配置获取表之间关联关系
  71. if (columnInfo.Navigat != null)
  72. {
  73. var name1 = columnInfo.Navigat.GetName();
  74. var name2 = columnInfo.Navigat.GetName2();
  75. var targetColumnName = string.IsNullOrEmpty(name2) ? "Id" : name2;
  76. var relation = new ColumnRelation
  77. {
  78. SourceTableName = columnInfo.DbTableName,
  79. SourceColumnName = dbOptions.DbSettings.EnableUnderLine ? name1.ToUnderLine() : name1,
  80. Type = columnInfo.Navigat.GetNavigateType() == NavigateType.OneToOne ? "ONE_TO_ONE" : "ONE_TO_MANY",
  81. TargetTableName = dbOptions.DbSettings.EnableUnderLine ? columnInfo.DbColumnName.ToUnderLine() : columnInfo.DbColumnName,
  82. TargetColumnName = dbOptions.DbSettings.EnableUnderLine ? targetColumnName.ToUnderLine() : targetColumnName
  83. };
  84. columnRelationList.Add(relation);
  85. }
  86. }
  87. }
  88. return new VisualDbTable { VisualTableList = visualTableList, VisualColumnList = visualColumnList, ColumnRelationList = columnRelationList };
  89. }
  90. /// <summary>
  91. /// 获取字段列表 🔖
  92. /// </summary>
  93. /// <param name="tableName">表名</param>
  94. /// <param name="configId">ConfigId</param>
  95. /// <returns></returns>
  96. [DisplayName("获取字段列表")]
  97. public List<DbColumnOutput> GetColumnList(string tableName, string configId = SqlSugarConst.MainConfigId)
  98. {
  99. var db = _db.AsTenant().GetConnectionScope(configId);
  100. return string.IsNullOrWhiteSpace(tableName) ? new List<DbColumnOutput>() : db.DbMaintenance.GetColumnInfosByTableName(tableName, false).Adapt<List<DbColumnOutput>>();
  101. }
  102. /// <summary>
  103. /// 获取数据库数据类型列表 🔖
  104. /// </summary>
  105. /// <param name="configId"></param>
  106. /// <returns></returns>
  107. [DisplayName("获取数据库数据类型列表")]
  108. public List<string> GetDbTypeList(string configId = SqlSugarConst.MainConfigId)
  109. {
  110. var db = _db.AsTenant().GetConnectionScope(configId);
  111. return db.DbMaintenance.GetDbTypes().OrderBy(u => u).ToList();
  112. }
  113. /// <summary>
  114. /// 增加列 🔖
  115. /// </summary>
  116. /// <param name="input"></param>
  117. [ApiDescriptionSettings(Name = "AddColumn"), HttpPost]
  118. [DisplayName("增加列")]
  119. public void AddColumn(DbColumnInput input)
  120. {
  121. var column = new DbColumnInfo
  122. {
  123. ColumnDescription = input.ColumnDescription,
  124. DbColumnName = input.DbColumnName,
  125. IsIdentity = input.IsIdentity == 1,
  126. IsNullable = input.IsNullable == 1,
  127. IsPrimarykey = input.IsPrimarykey == 1,
  128. Length = input.Length,
  129. DecimalDigits = input.DecimalDigits,
  130. DataType = input.DataType
  131. };
  132. var db = _db.AsTenant().GetConnectionScope(input.ConfigId);
  133. db.DbMaintenance.AddColumn(input.TableName, column);
  134. // 默认值直接添加报错
  135. if (!string.IsNullOrWhiteSpace(input.DefaultValue))
  136. {
  137. db.DbMaintenance.AddDefaultValue(input.TableName, column.DbColumnName, input.DefaultValue);
  138. }
  139. db.DbMaintenance.AddColumnRemark(input.DbColumnName, input.TableName, input.ColumnDescription);
  140. if (column.IsPrimarykey) db.DbMaintenance.AddPrimaryKey(input.TableName, input.DbColumnName);
  141. }
  142. /// <summary>
  143. /// 删除列 🔖
  144. /// </summary>
  145. /// <param name="input"></param>
  146. [ApiDescriptionSettings(Name = "DeleteColumn"), HttpPost]
  147. [DisplayName("删除列")]
  148. public void DeleteColumn(DeleteDbColumnInput input)
  149. {
  150. var db = _db.AsTenant().GetConnectionScope(input.ConfigId);
  151. db.DbMaintenance.DropColumn(input.TableName, input.DbColumnName);
  152. }
  153. /// <summary>
  154. /// 编辑列 🔖
  155. /// </summary>
  156. /// <param name="input"></param>
  157. [ApiDescriptionSettings(Name = "UpdateColumn"), HttpPost]
  158. [DisplayName("编辑列")]
  159. public void UpdateColumn(UpdateDbColumnInput input)
  160. {
  161. var db = _db.AsTenant().GetConnectionScope(input.ConfigId);
  162. // 前端未修改列名时,不进行重命名操作,避免报错
  163. if (input.OldColumnName != input.ColumnName)
  164. {
  165. db.DbMaintenance.RenameColumn(input.TableName, input.OldColumnName, input.ColumnName);
  166. }
  167. if (!string.IsNullOrWhiteSpace(input.DefaultValue))
  168. {
  169. db.DbMaintenance.AddDefaultValue(input.TableName, input.ColumnName, input.DefaultValue);
  170. }
  171. //if (db.DbMaintenance.IsAnyColumnRemark(input.ColumnName, input.TableName))
  172. //{
  173. // db.DbMaintenance.DeleteColumnRemark(input.ColumnName, input.TableName);
  174. //}
  175. db.DbMaintenance.AddColumnRemark(input.ColumnName, input.TableName, string.IsNullOrWhiteSpace(input.Description) ? input.ColumnName : input.Description);
  176. }
  177. /// <summary>
  178. /// 移动列位置 🔖
  179. /// </summary>
  180. /// <param name="input"></param>
  181. [ApiDescriptionSettings(Name = "MoveColumn"), HttpPost]
  182. [DisplayName("移动列")]
  183. public void MoveColumn(MoveDbColumnInput input)
  184. {
  185. var db = _db.AsTenant().GetConnectionScope(input.ConfigId);
  186. var dbMaintenance = db.DbMaintenance;
  187. var dbType = db.CurrentConnectionConfig.DbType;
  188. var columns = dbMaintenance.GetColumnInfosByTableName(input.TableName, false);
  189. var targetColumn = columns.FirstOrDefault(c =>
  190. c.DbColumnName.Equals(input.ColumnName, StringComparison.OrdinalIgnoreCase));
  191. if (targetColumn == null)
  192. throw new Exception($"列 {input.ColumnName} 在表 {input.TableName} 中不存在");
  193. switch (dbType)
  194. {
  195. case SqlSugar.DbType.MySql:
  196. MoveColumnInMySQL(db, input.TableName, input.ColumnName, input.AfterColumnName);
  197. break;
  198. default:
  199. throw new NotSupportedException($"暂不支持 {dbType} 数据库的列移动操作");
  200. }
  201. }
  202. /// <summary>
  203. /// 获取列定义
  204. /// </summary>
  205. /// <param name="db"></param>
  206. /// <param name="tableName"></param>
  207. /// <param name="columnName"></param>
  208. /// <param name="noDefault"></param>
  209. /// <returns></returns>
  210. /// <exception cref="Exception"></exception>
  211. private string GetColumnDefinitionInMySQL(ISqlSugarClient db, string tableName, string columnName, bool noDefault = false)
  212. {
  213. var columnDef = db.Ado.SqlQuery<dynamic>(
  214. $"SHOW FULL COLUMNS FROM `{tableName}` WHERE Field = '{columnName}'"
  215. ).FirstOrDefault();
  216. if (columnDef == null)
  217. throw new Exception($"Column {columnName} not found");
  218. var definition = new StringBuilder();
  219. definition.Append($"`{columnName}` "); // 列名
  220. definition.Append($"{columnDef.Type} "); // 数据类型
  221. // 处理约束条件
  222. definition.Append(columnDef.Null == "YES" ? "NULL " : "NOT NULL ");
  223. if (columnDef.Default != null && !noDefault)
  224. definition.Append($"DEFAULT '{columnDef.Default}' ");
  225. if (!string.IsNullOrEmpty(columnDef.Extra))
  226. definition.Append($"{columnDef.Extra} ");
  227. if (!string.IsNullOrEmpty(columnDef.Comment))
  228. definition.Append($"COMMENT '{columnDef.Comment.Replace("'", "''")}'");
  229. return definition.ToString();
  230. }
  231. /// <summary>
  232. /// MySQL 列移动实现
  233. /// </summary>
  234. /// <param name="db"></param>
  235. /// <param name="tableName"></param>
  236. /// <param name="columnName"></param>
  237. /// <param name="afterColumnName"></param>
  238. private void MoveColumnInMySQL(ISqlSugarClient db, string tableName, string columnName, string afterColumnName)
  239. {
  240. var definition = GetColumnDefinitionInMySQL(db, tableName, columnName);
  241. var sql = new StringBuilder();
  242. sql.Append($"ALTER TABLE `{tableName}` MODIFY COLUMN {definition}");
  243. if (string.IsNullOrEmpty(afterColumnName))
  244. sql.Append(" FIRST");
  245. else
  246. sql.Append($" AFTER `{afterColumnName}`");
  247. db.Ado.ExecuteCommand(sql.ToString());
  248. }
  249. /// <summary>
  250. /// 获取表列表 🔖
  251. /// </summary>
  252. /// <param name="configId">ConfigId</param>
  253. /// <returns></returns>
  254. [DisplayName("获取表列表")]
  255. public List<DbTableInfo> GetTableList(string configId = SqlSugarConst.MainConfigId)
  256. {
  257. var db = _db.AsTenant().GetConnectionScope(configId);
  258. return db.DbMaintenance.GetTableInfoList(false);
  259. }
  260. /// <summary>
  261. /// 增加表 🔖
  262. /// </summary>
  263. /// <param name="input"></param>
  264. [ApiDescriptionSettings(Name = "AddTable"), HttpPost]
  265. [DisplayName("增加表")]
  266. public void AddTable(DbTableInput input)
  267. {
  268. if (input.DbColumnInfoList == null || !input.DbColumnInfoList.Any())
  269. throw Oops.Oh(ErrorCodeEnum.db1000);
  270. if (input.DbColumnInfoList.GroupBy(u => u.DbColumnName).Any(u => u.Count() > 1))
  271. throw Oops.Oh(ErrorCodeEnum.db1002);
  272. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == input.ConfigId);
  273. var db = _db.AsTenant().GetConnectionScope(input.ConfigId);
  274. var typeBuilder = db.DynamicBuilder().CreateClass(input.TableName, new SugarTable() { TableName = input.TableName, TableDescription = input.Description });
  275. input.DbColumnInfoList.ForEach(u =>
  276. {
  277. var dbColumnName = config!.DbSettings.EnableUnderLine ? u.DbColumnName.Trim().ToUnderLine() : u.DbColumnName.Trim();
  278. // 虚拟类都默认string类型,具体以列数据类型为准
  279. typeBuilder.CreateProperty(dbColumnName, typeof(string), new SugarColumn()
  280. {
  281. IsPrimaryKey = u.IsPrimarykey == 1,
  282. IsIdentity = u.IsIdentity == 1,
  283. ColumnDataType = u.DataType,
  284. Length = u.Length,
  285. IsNullable = u.IsNullable == 1,
  286. DecimalDigits = u.DecimalDigits,
  287. ColumnDescription = u.ColumnDescription,
  288. DefaultValue = u.DefaultValue,
  289. });
  290. });
  291. db.CodeFirst.InitTables(typeBuilder.BuilderType());
  292. }
  293. /// <summary>
  294. /// 删除表 🔖
  295. /// </summary>
  296. /// <param name="input"></param>
  297. [ApiDescriptionSettings(Name = "DeleteTable"), HttpPost]
  298. [DisplayName("删除表")]
  299. public void DeleteTable(DeleteDbTableInput input)
  300. {
  301. var db = _db.AsTenant().GetConnectionScope(input.ConfigId);
  302. db.DbMaintenance.DropTable(input.TableName);
  303. }
  304. /// <summary>
  305. /// 编辑表 🔖
  306. /// </summary>
  307. /// <param name="input"></param>
  308. [ApiDescriptionSettings(Name = "UpdateTable"), HttpPost]
  309. [DisplayName("编辑表")]
  310. public void UpdateTable(UpdateDbTableInput input)
  311. {
  312. var db = _db.AsTenant().GetConnectionScope(input.ConfigId);
  313. db.DbMaintenance.RenameTable(input.OldTableName, input.TableName);
  314. try
  315. {
  316. if (db.DbMaintenance.IsAnyTableRemark(input.TableName))
  317. db.DbMaintenance.DeleteTableRemark(input.TableName);
  318. if (!string.IsNullOrWhiteSpace(input.Description))
  319. db.DbMaintenance.AddTableRemark(input.TableName, input.Description);
  320. }
  321. catch (NotSupportedException ex)
  322. {
  323. throw Oops.Oh(ex.ToString());
  324. }
  325. }
  326. /// <summary>
  327. /// 创建实体 🔖
  328. /// </summary>
  329. /// <param name="input"></param>
  330. [ApiDescriptionSettings(Name = "CreateEntity"), HttpPost]
  331. [DisplayName("创建实体")]
  332. public void CreateEntity(CreateEntityInput input)
  333. {
  334. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == input.ConfigId);
  335. input.Position = string.IsNullOrWhiteSpace(input.Position) ? "Admin.NET.Application" : input.Position;
  336. input.EntityName = string.IsNullOrWhiteSpace(input.EntityName) ? (config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(input.TableName, null) : input.TableName) : input.EntityName;
  337. string[] dbColumnNames = Array.Empty<string>();
  338. // Entity.cs.vm中是允许创建没有基类的实体的,所以这里也要做出相同的判断
  339. if (!string.IsNullOrWhiteSpace(input.BaseClassName))
  340. {
  341. Assembly assembly = Assembly.Load("Admin.NET.Core");
  342. Type type = assembly.GetType($"Admin.NET.Core.{input.BaseClassName}");
  343. if (type is null)
  344. throw Oops.Oh("基类集合配置不存在此类型");
  345. dbColumnNames = CodeGenUtil.GetPropertyInfoArray(type)?.Select(p => p.Name).ToArray();
  346. if (dbColumnNames is null || dbColumnNames is { Length: 0 })
  347. throw Oops.Oh("基类中不存在任何字段");
  348. }
  349. var templatePath = GetEntityTemplatePath();
  350. var targetPath = GetEntityTargetPath(input);
  351. var db = _db.AsTenant().GetConnectionScope(input.ConfigId);
  352. DbTableInfo dbTableInfo = db.DbMaintenance.GetTableInfoList(false).FirstOrDefault(u => u.Name == input.TableName || u.Name == input.TableName.ToLower()) ?? throw Oops.Oh(ErrorCodeEnum.db1001);
  353. List<DbColumnInfo> dbColumnInfos = db.DbMaintenance.GetColumnInfosByTableName(input.TableName, false);
  354. dbColumnInfos.ForEach(u =>
  355. {
  356. if (u.DbColumnName.ToUpper() == u.DbColumnName)
  357. {
  358. //字段全是大写的, 这种情况下生成的代码会有问题(即对 DOB 这样的字段,生成的前端代码为 dOB, 而数据序列化到前端又成了 dob,导致bug),因此抛出异常,不允许。
  359. throw new Exception($"错误:{u.DbColumnName} 字段全是大写字母,这样生成的代码会有bug!请更改为大写字母开头的驼峰式命名!");
  360. }
  361. u.PropertyName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, dbColumnNames) : u.DbColumnName; // 转下划线后的列名需要再转回来
  362. u.DataType = CodeGenUtil.ConvertDataType(u, config.DbType);
  363. });
  364. if (_codeGenOptions.BaseEntityNames.Contains(input.BaseClassName, StringComparer.OrdinalIgnoreCase))
  365. dbColumnInfos = dbColumnInfos.Where(u => !dbColumnNames.Contains(u.PropertyName, StringComparer.OrdinalIgnoreCase)).ToList();
  366. var tContent = File.ReadAllText(templatePath);
  367. var tResult = _viewEngine.RunCompileFromCached(tContent, new
  368. {
  369. NameSpace = $"{input.Position}.Entity",
  370. input.TableName,
  371. input.EntityName,
  372. BaseClassName = string.IsNullOrWhiteSpace(input.BaseClassName) ? "" : $": {input.BaseClassName}",
  373. input.ConfigId,
  374. dbTableInfo.Description,
  375. TableField = dbColumnInfos
  376. });
  377. File.WriteAllText(targetPath, tResult, Encoding.UTF8);
  378. }
  379. /// <summary>
  380. /// 创建种子数据 🔖
  381. /// </summary>
  382. /// <param name="input"></param>
  383. [ApiDescriptionSettings(Name = "CreateSeedData"), HttpPost]
  384. [DisplayName("创建种子数据")]
  385. public async Task CreateSeedData(CreateSeedDataInput input)
  386. {
  387. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == input.ConfigId);
  388. input.Position = string.IsNullOrWhiteSpace(input.Position) ? "Admin.NET.Core" : input.Position;
  389. var templatePath = GetSeedDataTemplatePath();
  390. var db = _db.AsTenant().GetConnectionScope(input.ConfigId);
  391. var tableInfo = db.DbMaintenance.GetTableInfoList(false).First(u => u.Name == input.TableName); // 表名
  392. List<DbColumnInfo> dbColumnInfos = db.DbMaintenance.GetColumnInfosByTableName(input.TableName, false); // 所有字段
  393. IEnumerable<EntityInfo> entityInfos = await GetEntityInfos();
  394. Type entityType = null;
  395. foreach (var item in entityInfos)
  396. {
  397. if (tableInfo.Name.ToLower() != (config.DbSettings.EnableUnderLine ? item.DbTableName.ToUnderLine() : item.DbTableName).ToLower()) continue;
  398. entityType = item.Type;
  399. break;
  400. }
  401. if (entityType == null) throw Oops.Oh(ErrorCodeEnum.db1003);
  402. input.EntityName = entityType.Name;
  403. input.SeedDataName = entityType.Name + "SeedData";
  404. if (!string.IsNullOrWhiteSpace(input.Suffix)) input.SeedDataName += input.Suffix;
  405. // 查询所有数据
  406. var query = db.QueryableByObject(entityType);
  407. // 优先用创建时间排序
  408. DbColumnInfo orderField = dbColumnInfos.FirstOrDefault(u => u.DbColumnName.ToLower() == "create_time" || u.DbColumnName.ToLower() == "createtime");
  409. if (orderField != null) query = query.OrderBy(orderField.DbColumnName);
  410. // 再使用第一个主键排序
  411. query = query.OrderBy(dbColumnInfos.First(u => u.IsPrimarykey).DbColumnName);
  412. var records = ((IEnumerable)await query.ToListAsync()).ToDynamicList();
  413. // 过滤已存在的数据
  414. if (input.FilterExistingData && records.Any())
  415. {
  416. // 获取实体类型-所有种数据数据类型
  417. var entityTypes = App.EffectiveTypes.Where(u => !u.IsInterface && !u.IsAbstract && u.IsClass && u.IsDefined(typeof(SugarTable), false) && u.FullName.EndsWith("." + input.EntityName))
  418. .Where(u => !u.GetCustomAttributes<IgnoreTableAttribute>().Any())
  419. .ToList();
  420. if (entityTypes.Count == 1) // 只有一个实体匹配才能过滤
  421. {
  422. // 获取实体的主键对应的属性名称
  423. var pkInfo = entityTypes[0].GetProperties().FirstOrDefault(u => u.GetCustomAttribute<SugarColumn>()?.IsPrimaryKey == true);
  424. if (pkInfo != null)
  425. {
  426. var seedDataTypes = App.EffectiveTypes
  427. .Where(u => !u.IsInterface && !u.IsAbstract && u.IsClass && u.GetInterfaces().Any(
  428. i => i.HasImplementedRawGeneric(typeof(ISqlSugarEntitySeedData<>)) && i.GenericTypeArguments[0] == entityTypes[0]
  429. )
  430. )
  431. .ToList();
  432. // 可能会重名的种子数据不作为过滤项
  433. string doNotFilterFullName1 = $"{input.Position}.SeedData.{input.SeedDataName}";
  434. string doNotFilterFullName2 = $"{input.Position}.{input.SeedDataName}"; // Core中的命名空间没有SeedData
  435. PropertyInfo idPropertySeedData = records[0].GetType().GetProperty("Id");
  436. for (int i = seedDataTypes.Count - 1; i >= 0; i--)
  437. {
  438. string fullName = seedDataTypes[i].FullName;
  439. if ((fullName == doNotFilterFullName1) || (fullName == doNotFilterFullName2)) continue;
  440. // 删除重复数据
  441. var instance = Activator.CreateInstance(seedDataTypes[i]);
  442. var hasDataMethod = seedDataTypes[i].GetMethod("HasData");
  443. var seedData = ((IEnumerable)hasDataMethod?.Invoke(instance, null))?.Cast<object>();
  444. if (seedData == null) continue;
  445. List<object> recordsToRemove = new();
  446. foreach (var record in records)
  447. {
  448. object recordId = pkInfo.GetValue(record);
  449. if (seedData.Select(d1 => idPropertySeedData.GetValue(d1)).Any(dataId => recordId != null && dataId != null && recordId.Equals(dataId)))
  450. {
  451. recordsToRemove.Add(record);
  452. }
  453. }
  454. foreach (var itemToRemove in recordsToRemove)
  455. {
  456. records.Remove(itemToRemove);
  457. }
  458. }
  459. }
  460. }
  461. }
  462. // 检查有没有 System.Text.Json.Serialization.JsonIgnore 的属性
  463. // var jsonIgnoreProperties = entityType.GetProperties().Where(p => (p.GetAttribute<System.Text.Json.Serialization.JsonIgnoreAttribute>() != null ||
  464. // p.GetAttribute<JsonIgnoreAttribute>() != null) && p.GetAttribute<SugarColumn>() != null).ToList();
  465. // var jsonIgnoreInfo = new List<List<JsonIgnoredPropertyData>>();
  466. // if (jsonIgnoreProperties.Count > 0)
  467. // {
  468. // int recordIndex = 0;
  469. // foreach (var r in (IEnumerable)records)
  470. // {
  471. // List<JsonIgnoredPropertyData> record = new();
  472. // foreach (var item in jsonIgnoreProperties)
  473. // {
  474. // object v = item.GetValue(r);
  475. // string strValue = "null";
  476. // if (v != null)
  477. // {
  478. // strValue = v.ToString();
  479. // if (v.GetType() == typeof(string))
  480. // strValue = "\"" + strValue + "\"";
  481. // else if (v.GetType() == typeof(DateTime))
  482. // strValue = "DateTime.Parse(\"" + ((DateTime)v).ToString("yyyy-MM-dd HH:mm:ss") + "\")";
  483. // }
  484. // record.Add(new JsonIgnoredPropertyData { RecordIndex = recordIndex, Name = item.Name, Value = strValue });
  485. // }
  486. // recordIndex++;
  487. // jsonIgnoreInfo.Add(record);
  488. // }
  489. // }
  490. // 获取所有字段信息
  491. var propertyList = entityType.GetProperties().Where(x => false == (x.GetCustomAttribute<SugarColumn>()?.IsIgnore ?? false)).ToList();
  492. for (var i = 0; i < propertyList.Count; i++)
  493. {
  494. if (propertyList[i].Name != nameof(EntityBaseId.Id) || !(propertyList[i].GetCustomAttribute<SugarColumn>()?.IsPrimaryKey ?? true)) continue;
  495. var temp = propertyList[i];
  496. for (var j = i; j > 0; j--) propertyList[j] = propertyList[j - 1];
  497. propertyList[0] = temp;
  498. }
  499. // 拼接数据
  500. var recordList = records.Select(obj => string.Join(", ", propertyList.Select(prop =>
  501. {
  502. var propType = Nullable.GetUnderlyingType(prop.PropertyType) ?? prop.PropertyType;
  503. object value = prop.GetValue(obj);
  504. if (value == null) value = "null";
  505. else if (propType == typeof(string))
  506. {
  507. value = $"@\"{value}\"";
  508. }
  509. else if (propType.IsEnum)
  510. {
  511. value = $"{propType.Name}.{value}";
  512. }
  513. else if (propType == typeof(bool))
  514. {
  515. value = (bool)value ? "true" : "false";
  516. }
  517. else if (propType == typeof(DateTime))
  518. {
  519. value = $"DateTime.Parse(\"{((DateTime)value):yyyy-MM-dd HH:mm:ss.fff}\")";
  520. }
  521. return $"{prop.Name}={value}";
  522. }))).ToList();
  523. var tContent = await File.ReadAllTextAsync(templatePath);
  524. var data = new
  525. {
  526. NameSpace = $"{input.Position}.SeedData",
  527. EntityNameSpace = entityType.Namespace,
  528. input.TableName,
  529. input.EntityName,
  530. input.SeedDataName,
  531. input.ConfigId,
  532. tableInfo.Description,
  533. // JsonIgnoreInfo = jsonIgnoreInfo,
  534. RecordList = recordList
  535. };
  536. var tResult = await _viewEngine.RunCompileAsync(tContent, data, builderAction: builder =>
  537. {
  538. builder.AddAssemblyReferenceByName("System.Linq");
  539. builder.AddAssemblyReferenceByName("System.Collections");
  540. builder.AddUsing("System.Collections.Generic");
  541. builder.AddUsing("System.Linq");
  542. });
  543. var targetPath = GetSeedDataTargetPath(input);
  544. await File.WriteAllTextAsync(targetPath, tResult, Encoding.UTF8);
  545. }
  546. /// <summary>
  547. /// 获取库表信息
  548. /// </summary>
  549. /// <returns></returns>
  550. private async Task<IEnumerable<EntityInfo>> GetEntityInfos()
  551. {
  552. var entityInfos = new List<EntityInfo>();
  553. var type = typeof(SugarTable);
  554. var types = new List<Type>();
  555. if (_codeGenOptions.EntityAssemblyNames != null)
  556. {
  557. foreach (var asm in _codeGenOptions.EntityAssemblyNames.Select(Assembly.Load))
  558. {
  559. types.AddRange(asm.GetExportedTypes().ToList());
  560. }
  561. }
  562. Type[] cosType = types.Where(o => IsMyAttribute(Attribute.GetCustomAttributes(o, true))).ToArray();
  563. foreach (var c in cosType)
  564. {
  565. var sugarAttribute = c.GetCustomAttributes(type, true)?.FirstOrDefault();
  566. var des = c.GetCustomAttributes(typeof(DescriptionAttribute), true);
  567. var description = "";
  568. if (des.Length > 0)
  569. {
  570. description = ((DescriptionAttribute)des[0]).Description;
  571. }
  572. entityInfos.Add(new EntityInfo()
  573. {
  574. EntityName = c.Name,
  575. DbTableName = sugarAttribute == null ? c.Name : ((SugarTable)sugarAttribute).TableName,
  576. TableDescription = description,
  577. Type = c
  578. });
  579. }
  580. return await Task.FromResult(entityInfos);
  581. bool IsMyAttribute(Attribute[] o)
  582. {
  583. return o.Any(a => a.GetType() == type);
  584. }
  585. }
  586. /// <summary>
  587. /// 获取实体模板文件路径
  588. /// </summary>
  589. /// <returns></returns>
  590. private static string GetEntityTemplatePath()
  591. {
  592. var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "template");
  593. return Path.Combine(templatePath, "Entity.cs.vm");
  594. }
  595. /// <summary>
  596. /// 获取种子数据模板文件路径
  597. /// </summary>
  598. /// <returns></returns>
  599. private static string GetSeedDataTemplatePath()
  600. {
  601. var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "template");
  602. return Path.Combine(templatePath, "SeedData.cs.vm");
  603. }
  604. /// <summary>
  605. /// 设置生成实体文件路径
  606. /// </summary>
  607. /// <param name="input"></param>
  608. /// <returns></returns>
  609. private static string GetEntityTargetPath(CreateEntityInput input)
  610. {
  611. var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, input.Position, "Entity");
  612. //if (!Directory.Exists(backendPath))
  613. //{
  614. // var pluginsPath = App.GetConfig<string[]>("AppSettings:ExternalAssemblies");
  615. // foreach (var pluginPath in pluginsPath)
  616. // {
  617. // backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, pluginPath, input.Position, "Entity");
  618. // if (Directory.Exists(backendPath))
  619. // break;
  620. // }
  621. //}
  622. if (!Directory.Exists(backendPath))
  623. Directory.CreateDirectory(backendPath);
  624. return Path.Combine(backendPath, input.EntityName + ".cs");
  625. }
  626. /// <summary>
  627. /// 设置生成种子数据文件路径
  628. /// </summary>
  629. /// <param name="input"></param>
  630. /// <returns></returns>
  631. private static string GetSeedDataTargetPath(CreateSeedDataInput input)
  632. {
  633. var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, input.Position, "SeedData");
  634. if (!Directory.Exists(backendPath))
  635. Directory.CreateDirectory(backendPath);
  636. return Path.Combine(backendPath, input.SeedDataName + ".cs");
  637. }
  638. /// <summary>
  639. /// 备份数据库(PostgreSQL)🔖
  640. /// </summary>
  641. /// <returns></returns>
  642. [HttpPost, NonUnify]
  643. [DisplayName("备份数据库(PostgreSQL)")]
  644. public async Task<IActionResult> BackupDatabase()
  645. {
  646. if (_db.CurrentConnectionConfig.DbType != SqlSugar.DbType.PostgreSQL)
  647. throw Oops.Oh("只支持 PostgreSQL 数据库 😁");
  648. var npgsqlConn = new NpgsqlConnectionStringBuilder(_db.CurrentConnectionConfig.ConnectionString);
  649. if (npgsqlConn == null || string.IsNullOrWhiteSpace(npgsqlConn.Host) || string.IsNullOrWhiteSpace(npgsqlConn.Username) || string.IsNullOrWhiteSpace(npgsqlConn.Password) || string.IsNullOrWhiteSpace(npgsqlConn.Database))
  650. throw Oops.Oh("PostgreSQL 数据库配置错误");
  651. // 确保备份目录存在
  652. var backupDirectory = Path.Combine(Directory.GetCurrentDirectory(), "backups");
  653. Directory.CreateDirectory(backupDirectory);
  654. // 构建备份文件名
  655. string backupFileName = $"backup_{DateTime.Now:yyyyMMddHHmmss}.sql";
  656. string backupFilePath = Path.Combine(backupDirectory, backupFileName);
  657. // 启动pg_dump进程进行备份
  658. // 设置密码:export PGPASSWORD='xxxxxx'
  659. var bash = $"-U {npgsqlConn.Username} -h {npgsqlConn.Host} -p {npgsqlConn.Port} -E UTF8 -F c -b -v -f {backupFilePath} {npgsqlConn.Database}";
  660. var startInfo = new ProcessStartInfo
  661. {
  662. FileName = "pg_dump",
  663. Arguments = bash,
  664. UseShellExecute = false,
  665. RedirectStandardOutput = true,
  666. RedirectStandardError = true,
  667. CreateNoWindow = true,
  668. EnvironmentVariables =
  669. {
  670. ["PGPASSWORD"] = npgsqlConn.Password
  671. }
  672. };
  673. //_logger.LogInformation("备份数据库:pg_dump " + bash);
  674. //try
  675. //{
  676. using (var backupProcess = Process.Start(startInfo))
  677. {
  678. await backupProcess.WaitForExitAsync();
  679. //var output = await backupProcess.StandardOutput.ReadToEndAsync();
  680. //var error = await backupProcess.StandardError.ReadToEndAsync();
  681. // 检查备份是否成功
  682. if (backupProcess.ExitCode != 0)
  683. {
  684. throw Oops.Oh($"备份失败:ExitCode({backupProcess.ExitCode})");
  685. }
  686. }
  687. // _logger.LogInformation($"备份成功:{backupFilePath}");
  688. //}
  689. //catch (Exception ex)
  690. //{
  691. // _logger.LogError(ex, $"备份失败:");
  692. // throw;
  693. //}
  694. // 若备份成功则提供下载链接
  695. return new FileStreamResult(new FileStream(backupFilePath, FileMode.Open), "application/octet-stream")
  696. {
  697. FileDownloadName = backupFileName
  698. };
  699. }
  700. }