123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145 |
- using Admin.Core.Common.Auth;
- using Admin.Core.Common.BaseModel;
- using Admin.Core.Common.Configs;
- using Admin.Core.Model.Admin;
- using Admin.Core.Repository.Admin.Tenant.Dto;
- using FreeSql;
- using Microsoft.Extensions.DependencyInjection;
- using System;
- namespace Admin.Core.Repository
- {
- public static class IdleBusExtesions
- {
- /// <summary>
- /// 创建FreeSql实例
- /// </summary>
- /// <param name="user"></param>
- /// <param name="appConfig"></param>
- /// <param name="dbConfig"></param>
- /// <param name="tenant"></param>
- /// <returns></returns>
- private static IFreeSql CreateFreeSql(IUser user, AppConfig appConfig, DbConfig dbConfig, CreateFreeSqlTenantDto tenant)
- {
- var freeSqlBuilder = new FreeSqlBuilder()
- .UseConnectionString(tenant.DbType.Value, tenant.ConnectionString)
- .UseAutoSyncStructure(false)
- .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<IEntitySoftDelete>("SoftDelete", a => a.IsDeleted == false);
- //配置实体
- DbHelper.ConfigEntity(fsql, appConfig);
- #region 监听Curd操作
- if (dbConfig.Curd)
- {
- fsql.Aop.CurdBefore += (s, e) =>
- {
- Console.WriteLine($"{e.Sql}\r\n");
- };
- }
- #endregion 监听Curd操作
- #region 审计数据
- //计算服务器时间
- var serverTime = fsql.Select<DualEntity>().Limit(1).First(a => DateTime.UtcNow);
- var timeOffset = DateTime.UtcNow.Subtract(serverTime);
- fsql.Aop.AuditValue += (s, e) =>
- {
- DbHelper.AuditValue(e, timeOffset, user);
- };
- #endregion 审计数据
- return fsql;
- }
- /// <summary>
- /// 获得FreeSql实例
- /// </summary>
- /// <param name="ib"></param>
- /// <param name="serviceProvider"></param>
- /// <returns></returns>
- public static IFreeSql GetFreeSql(this IdleBus<IFreeSql> ib, IServiceProvider serviceProvider)
- {
- var user = serviceProvider.GetRequiredService<IUser>();
- var appConfig = serviceProvider.GetRequiredService<AppConfig>();
- var tenantId = user.TenantId;
- if (appConfig.Tenant && user.DataIsolationType == DataIsolationType.OwnDb && tenantId.HasValue)
- {
- var tenantName = "tenant_" + tenantId.ToString();
- var exists = ib.Exists(tenantName);
- if (!exists)
- {
- var dbConfig = serviceProvider.GetRequiredService<DbConfig>();
- //查询租户数据库信息
- var masterDb = serviceProvider.GetRequiredService<IFreeSql>();
- var tenantRepository = masterDb.GetRepositoryBase<TenantEntity>();
- var tenant = tenantRepository.Select.DisableGlobalFilter("Tenant").WhereDynamic(tenantId).ToOne<CreateFreeSqlTenantDto>();
- var timeSpan = tenant.IdleTime.HasValue && tenant.IdleTime.Value > 0 ? TimeSpan.FromMinutes(tenant.IdleTime.Value) : TimeSpan.MaxValue;
- ib.TryRegister(tenantName, () => CreateFreeSql(user, appConfig, dbConfig, tenant), timeSpan);
- }
- return ib.Get(tenantName);
- }
- else
- {
- var freeSql = serviceProvider.GetRequiredService<IFreeSql>();
- return freeSql;
- }
- }
- /// <summary>
- /// 获得租户FreeSql实例
- /// </summary>
- /// <param name="ib"></param>
- /// <param name="serviceProvider"></param>
- /// <param name="tenantId"></param>
- /// <returns></returns>
- public static IFreeSql GetTenantFreeSql(this IdleBus<IFreeSql> ib, IServiceProvider serviceProvider, long? tenantId = null)
- {
- if (tenantId.HasValue)
- {
- var user = serviceProvider.GetRequiredService<IUser>();
- var appConfig = serviceProvider.GetRequiredService<AppConfig>();
- var tenantName = "tenant_" + tenantId.ToString();
- var exists = ib.Exists(tenantName);
- if (!exists)
- {
- var dbConfig = serviceProvider.GetRequiredService<DbConfig>();
- //查询租户数据库信息
- var masterDb = serviceProvider.GetRequiredService<IFreeSql>();
- var tenantRepository = masterDb.GetRepositoryBase<TenantEntity>();
- var tenant = tenantRepository.Select.DisableGlobalFilter("Tenant").WhereDynamic(tenantId).ToOne<CreateFreeSqlTenantDto>();
- var timeSpan = tenant.IdleTime.HasValue && tenant.IdleTime.Value > 0 ? TimeSpan.FromMinutes(tenant.IdleTime.Value) : TimeSpan.MaxValue;
- ib.TryRegister(tenantName, () => CreateFreeSql(user, appConfig, dbConfig, tenant), timeSpan);
- }
- return ib.Get(tenantName);
- }
- return null;
- }
- }
- }
|