1
0

ServiceCollectionExtensions.cs 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126
  1. using System;
  2. using Microsoft.Extensions.DependencyInjection;
  3. using Microsoft.Extensions.Hosting;
  4. using FreeSql;
  5. using Admin.Core.Common.Configs;
  6. using Admin.Core.Common.Helpers;
  7. using Admin.Core.Model;
  8. using Admin.Core.Common.Auth;
  9. namespace Admin.Core.Db
  10. {
  11. public static class ServiceCollectionExtensions
  12. {
  13. /// <summary>
  14. /// 添加数据库
  15. /// </summary>
  16. /// <param name="services"></param>
  17. /// <param name="env"></param>
  18. public async static void AddDb(this IServiceCollection services, IHostEnvironment env)
  19. {
  20. var dbConfig = new ConfigHelper().Get<DbConfig>("dbconfig", env.EnvironmentName);
  21. //创建数据库
  22. if (dbConfig.CreateDb)
  23. {
  24. await DbHelper.CreateDatabase(dbConfig);
  25. }
  26. #region FreeSql
  27. var freeSqlBuilder = new FreeSqlBuilder()
  28. .UseConnectionString(dbConfig.Type, dbConfig.ConnectionString)
  29. .UseAutoSyncStructure(dbConfig.SyncStructure)
  30. .UseLazyLoading(false)
  31. .UseNoneCommandParameter(true);
  32. #region 监听所有命令
  33. if (dbConfig.MonitorCommand)
  34. {
  35. freeSqlBuilder.UseMonitorCommand(cmd => { }, (cmd, traceLog) =>
  36. {
  37. //Console.WriteLine($"{cmd.CommandText}\n{traceLog}\r\n");
  38. Console.WriteLine($"{cmd.CommandText}\r\n");
  39. });
  40. }
  41. #endregion
  42. var fsql = freeSqlBuilder.Build();
  43. #region 初始化数据库
  44. //同步结构
  45. if (dbConfig.SyncStructure)
  46. {
  47. DbHelper.SyncStructure(fsql);
  48. }
  49. //同步数据
  50. if (dbConfig.SyncData)
  51. {
  52. await DbHelper.SyncData(fsql);
  53. }
  54. #endregion
  55. //生成数据包
  56. if (dbConfig.GenerateData && !dbConfig.CreateDb && !dbConfig.SyncData)
  57. {
  58. await DbHelper.GenerateSimpleJsonData(fsql);
  59. }
  60. #region 监听Curd操作
  61. if (dbConfig.Curd)
  62. {
  63. fsql.Aop.CurdBefore += (s, e) =>
  64. {
  65. Console.WriteLine($"{e.Sql}\r\n");
  66. };
  67. }
  68. #endregion
  69. #region 审计数据
  70. //计算服务器时间
  71. //var serverTime = fsql.Select<T>().Limit(1).First(a => DateTime.local);
  72. //var timeOffset = DateTime.UtcNow.Subtract(serverTime);
  73. fsql.Aop.AuditValue += (s, e) =>
  74. {
  75. var user = services.BuildServiceProvider().GetService<IUser>();
  76. if (e.AuditValueType == FreeSql.Aop.AuditValueType.Insert)
  77. {
  78. switch (e.Property.Name)
  79. {
  80. case "CreatedUserId":
  81. e.Value = user?.Id;
  82. break;
  83. case "CreatedUserName":
  84. e.Value = user?.Name;
  85. break;
  86. //case "CreatedTime":
  87. // e.Value = DateTime.Now.Subtract(timeOffset);
  88. // break;
  89. }
  90. }
  91. else if (e.AuditValueType == FreeSql.Aop.AuditValueType.Update)
  92. {
  93. switch (e.Property.Name)
  94. {
  95. case "ModifiedUserId":
  96. e.Value = user?.Id;
  97. break;
  98. case "ModifiedUserName":
  99. e.Value = user?.Name;
  100. break;
  101. //case "ModifiedTime":
  102. // e.Value = DateTime.Now.Subtract(timeOffset);
  103. // break;
  104. }
  105. }
  106. };
  107. #endregion
  108. services.AddSingleton(fsql);
  109. services.AddFreeRepository(filter => filter.Apply<IEntitySoftDelete>("SoftDelete", a => a.IsDeleted == false));
  110. services.AddScoped<IUnitOfWork>(sp => sp.GetRequiredService<IFreeSql>().CreateUnitOfWork());
  111. #endregion
  112. }
  113. }
  114. }