DBServiceCollectionExtensions.cs 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
  1. using Admin.Core.Common.Auth;
  2. using Admin.Core.Common.BaseModel;
  3. using Admin.Core.Common.Configs;
  4. using Admin.Core.Common.Dbs;
  5. using Admin.Core.Common.Helpers;
  6. using Admin.Core.Model.Admin;
  7. using Admin.Core.Repository;
  8. using FreeSql;
  9. using Microsoft.Extensions.DependencyInjection;
  10. using Microsoft.Extensions.Hosting;
  11. using System;
  12. using System.Threading.Tasks;
  13. namespace Admin.Core.Db
  14. {
  15. public static class DBServiceCollectionExtensions
  16. {
  17. /// <summary>
  18. /// 添加数据库
  19. /// </summary>
  20. /// <param name="services"></param>
  21. /// <param name="env"></param>
  22. public async static Task AddDbAsync(this IServiceCollection services, IHostEnvironment env)
  23. {
  24. services.AddScoped<MyUnitOfWorkManager>();
  25. var dbConfig = new ConfigHelper().Get<DbConfig>("dbconfig", env.EnvironmentName);
  26. //创建数据库
  27. if (dbConfig.CreateDb)
  28. {
  29. await DbHelper.CreateDatabaseAsync(dbConfig);
  30. }
  31. #region FreeSql
  32. var freeSqlBuilder = new FreeSqlBuilder()
  33. .UseConnectionString(dbConfig.Type, dbConfig.ConnectionString)
  34. .UseAutoSyncStructure(false)
  35. .UseLazyLoading(false)
  36. .UseNoneCommandParameter(true);
  37. #region 监听所有命令
  38. if (dbConfig.MonitorCommand)
  39. {
  40. freeSqlBuilder.UseMonitorCommand(cmd => { }, (cmd, traceLog) =>
  41. {
  42. //Console.WriteLine($"{cmd.CommandText}\n{traceLog}\r\n");
  43. Console.WriteLine($"{cmd.CommandText}\r\n");
  44. });
  45. }
  46. #endregion 监听所有命令
  47. var fsql = freeSqlBuilder.Build();
  48. fsql.GlobalFilter.Apply<IEntitySoftDelete>("SoftDelete", a => a.IsDeleted == false);
  49. //配置实体
  50. var appConfig = new ConfigHelper().Get<AppConfig>("appconfig", env.EnvironmentName);
  51. DbHelper.ConfigEntity(fsql, appConfig);
  52. #region 初始化数据库
  53. //同步结构
  54. if (dbConfig.SyncStructure)
  55. {
  56. DbHelper.SyncStructure(fsql, dbConfig: dbConfig, appConfig: appConfig);
  57. }
  58. var user = services.BuildServiceProvider().GetService<IUser>();
  59. #region 审计数据
  60. //计算服务器时间
  61. var serverTime = fsql.Select<DualEntity>().Limit(1).First(a => DateTime.UtcNow);
  62. var timeOffset = DateTime.UtcNow.Subtract(serverTime);
  63. DbHelper.TimeOffset = timeOffset;
  64. fsql.Aop.AuditValue += (s, e) =>
  65. {
  66. DbHelper.AuditValue(e, timeOffset, user);
  67. };
  68. #endregion 审计数据
  69. //同步数据
  70. if (dbConfig.SyncData)
  71. {
  72. await DbHelper.SyncDataAsync(fsql, dbConfig, appConfig);
  73. }
  74. #endregion 初始化数据库
  75. //生成数据包
  76. if (dbConfig.GenerateData && !dbConfig.CreateDb && !dbConfig.SyncData)
  77. {
  78. await DbHelper.GenerateSimpleJsonDataAsync(fsql, appConfig);
  79. }
  80. #region 监听Curd操作
  81. if (dbConfig.Curd)
  82. {
  83. fsql.Aop.CurdBefore += (s, e) =>
  84. {
  85. Console.WriteLine($"{e.Sql}\r\n");
  86. };
  87. }
  88. #endregion 监听Curd操作
  89. if (appConfig.Tenant)
  90. {
  91. fsql.GlobalFilter.Apply<ITenant>("Tenant", a => a.TenantId == user.TenantId);
  92. }
  93. #endregion FreeSql
  94. services.AddSingleton(fsql);
  95. //导入多数据库
  96. if (null != dbConfig.Dbs)
  97. {
  98. foreach (var multiDb in dbConfig.Dbs)
  99. {
  100. switch (multiDb.Name)
  101. {
  102. case nameof(MySqlDb):
  103. var mdb = CreateMultiDbBuilder(multiDb).Build<MySqlDb>();
  104. services.AddSingleton(mdb);
  105. break;
  106. default:
  107. break;
  108. }
  109. }
  110. }
  111. }
  112. /// <summary>
  113. /// 创建多数据库构建器
  114. /// </summary>
  115. /// <param name="multiDb"></param>
  116. /// <returns></returns>
  117. private static FreeSqlBuilder CreateMultiDbBuilder(MultiDb multiDb)
  118. {
  119. return new FreeSqlBuilder()
  120. .UseConnectionString(multiDb.Type, multiDb.ConnectionString)
  121. .UseAutoSyncStructure(false)
  122. .UseLazyLoading(false)
  123. .UseNoneCommandParameter(true);
  124. }
  125. }
  126. }