using System; using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.Hosting; using FreeSql; using Admin.Core.Common.Configs; using Admin.Core.Common.Helpers; using Admin.Core.Common.Auth; using Admin.Core.Common.BaseModel; using Admin.Core.Repository; namespace Admin.Core.Db { //public class MyIdleBus : IdleBus //{ // public MyIdleBus() : base(TimeSpan.FromMinutes(30)) { } //} public static class TenantDBServiceCollectionExtensions { /// /// 添加租户数据库 /// /// /// public static void AddTenantDb(this IServiceCollection services, IHostEnvironment env) { services.AddScoped(); var dbConfig = new ConfigHelper().Get("dbconfig", env.EnvironmentName); var appConfig = new ConfigHelper().Get("appconfig", env.EnvironmentName); var user = services.BuildServiceProvider().GetService(); IdleBus ib = new IdleBus(TimeSpan.FromMinutes(1)); ib.TryRegister("tenant_" + user.TenantId.ToString(), () => { #region FreeSql var freeSqlBuilder = new FreeSqlBuilder() .UseConnectionString(dbConfig.Type, dbConfig.ConnectionString) .UseLazyLoading(false) .UseNoneCommandParameter(true); #region 监听所有命令 if (dbConfig.MonitorCommand) { freeSqlBuilder.UseMonitorCommand(cmd => { }, (cmd, traceLog) => { Console.WriteLine($"{cmd.CommandText}\r\n"); }); } #endregion var fsql = freeSqlBuilder.Build(); fsql.GlobalFilter.Apply("SoftDelete", a => a.IsDeleted == false); //共享数据库 if(appConfig.TenantType == TenantType.Share) { fsql.GlobalFilter.ApplyIf("Tenant", () => user.TenantId > 0, a => a.TenantId == user.TenantId); } #region 监听Curd操作 if (dbConfig.Curd) { fsql.Aop.CurdBefore += (s, e) => { Console.WriteLine($"{e.Sql}\r\n"); }; } #endregion #region 审计数据 fsql.Aop.AuditValue += (s, e) => { if (user == null || user.Id <= 0) { return; } if (e.AuditValueType == FreeSql.Aop.AuditValueType.Insert) { switch (e.Property.Name) { case "CreatedUserId": e.Value = user.Id; break; case "CreatedUserName": e.Value = user.Name; break; case "TenantId": e.Value = user.TenantId; break; } } else if (e.AuditValueType == FreeSql.Aop.AuditValueType.Update) { switch (e.Property.Name) { case "ModifiedUserId": e.Value = user.Id; break; case "ModifiedUserName": e.Value = user.Name; break; } } }; #endregion #endregion return fsql; }); services.AddSingleton(ib); } } }