using DS.Module.Core;
using DS.Module.Core.Extensions;
using DS.Module.UserModule;
using Microsoft.Extensions.DependencyInjection;
using NLog;
using SqlSugar;
using System.Linq.Dynamic.Core;
using LogLevel = NLog.LogLevel;
namespace DS.Module.SqlSugar;
///
/// SqlSugar 启动服务
///
public static class SqlsugarInstall
{
// ///
// /// 数据库连接字符串
// ///
// private static string DbInfo = AppSetting.Configuration["ConnectionStrings:DbInfo"];
//
// ///
// /// 数据库类型
// ///
// private static string DbType = AppSetting.Configuration["ConnectionStrings:DbType"];
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
// private readonly IUser user;
// private ILogger _logger;
///
/// 将模块服务添加到依赖注入服务容器中
///
/// 依赖注入服务容器
///
public static IServiceCollection AddSqlsugarInstall(this IServiceCollection services)
{
if (services == null) throw new ArgumentNullException(nameof(services));
Console.WriteLine("开始加载sqlsugar模块");
Console.WriteLine("数据库配置id:" + AppSetting.app(new string[] { "DBInfo", "DefaultDbConnId" }));
var dbs = AppSetting.app("DBInfo", "DBS");
//默认数据库
List dbList = new List();
DbConfig defaultdb = new DbConfig()
{
ConnId = AppSetting.app(new string[] { "DBInfo", "DefaultDbConnId" }),
Connection = AppSetting.app(new string[] { "DBInfo", "DefaultDbString" }),
DbType = AppSetting.app(new string[] { "DBInfo", "DefaultDbType" }).ObjToInt()
};
dbList.Add(defaultdb);
//业务数据库集合
foreach (var item in dbs)
{
dbList.Add(item);
}
List connectConfigList = new List();
foreach (var item in dbList)
{
//防止数据库重复,导致的事务异常
if (connectConfigList.Any(a => a.ConfigId == (dynamic)item.ConnId || a.ConnectionString == item.Connection))
{
continue;
}
connectConfigList.Add(new ConnectionConfig()
{
ConnectionString = item.Connection,
DbType = (DbType)item.DbType,
IsAutoCloseConnection = true,
ConfigId = item.ConnId,
InitKeyType = InitKeyType.Attribute,
MoreSettings = new ConnMoreSettings()
{
IsAutoRemoveDataCache = true //自动清理缓存
},
// 自定义特性
ConfigureExternalServices = new ConfigureExternalServices()
{
}
});
}
//全局上下文生效
SqlSugarScope sqlSugar = new SqlSugarScope(connectConfigList,
db =>
{
/*
* 默认只会配置到第一个数据库,这里按照官方文档进行多数据库/多租户文档的说明进行循环配置
*/
foreach (var c in connectConfigList)
{
var dbProvider = db.GetConnectionScope((string)c.ConfigId);
var user = services.GetService();
//单例参数配置,所有上下文生效
dbProvider.Ado.CommandTimeOut = 30;
dbProvider.Aop.OnLogExecuting = (sql, pars) =>
{
Logger.Log(LogLevel.Info,
DateTime.Now.ToString() + "\r\n" +
UtilMethods.GetSqlString(c.DbType, sql, pars));
};
dbProvider.Aop.DataExecuting = (oldValue, entityInfo) =>
{
// 新增操作
if (entityInfo.OperationType == DataFilterType.InsertByObject)
{
if (entityInfo.PropertyName == "Id")
{
if (entityInfo.EntityColumnInfo.PropertyInfo.PropertyType == typeof(string))
{
entityInfo.SetValue(GuidHelper.GetSnowflakeId());
}
}
if (entityInfo.PropertyName == "AddTime")
entityInfo.SetValue(DateTime.Now);
if (user != null)
{
if (!user.GetTenantId().ToString().IsNullOrEmpty())
{
if (entityInfo.PropertyName == "TenantId")
entityInfo.SetValue(user.GetTenantId());
}
if (!user.GetCompanyId().ToString().IsNullOrEmpty())
{
if (entityInfo.PropertyName == "CORPID")
entityInfo.SetValue(user.GetCompanyId());
}
}
// if (entityInfo.PropertyName == "TenantId")
// entityInfo.SetValue(user.GetTenantId());
if (entityInfo.PropertyName == "AddBy" && user != null)
entityInfo.SetValue(user.UserId);
if (entityInfo.PropertyName == "Deleted")
entityInfo.SetValue(false);
if (entityInfo.PropertyName == "GID")
{
if (entityInfo.EntityColumnInfo.PropertyInfo.PropertyType == typeof(Guid))
{
entityInfo.SetValue(Guid.NewGuid());
}
}
if (entityInfo.PropertyName == "CreateDate")
entityInfo.SetValue(DateTime.Now);
if (user != null && !user.GetUserGID().ToString().IsNullOrEmpty())
{
if (entityInfo.PropertyName == "CreateID")
entityInfo.SetValue(Guid.Parse(user.GetUserGID()));
}
}
// 更新操作
if (entityInfo.OperationType == DataFilterType.UpdateByObject)
{
if (entityInfo.PropertyName == "UpdateTime")
entityInfo.SetValue(DateTime.Now);
if (entityInfo.PropertyName == "UpdateBy" && user != null)
entityInfo.SetValue(user.UserId);
if (entityInfo.PropertyName == "ModifyDate")
entityInfo.SetValue(DateTime.Now);
if (user != null && !user.GetUserGID().ToString().IsNullOrEmpty())
{
if (entityInfo.PropertyName == "ModifyID")
entityInfo.SetValue(Guid.Parse(user.GetUserGID()));
}
}
};
dbProvider.Aop.OnError = (exp) => //执行SQL 错误事件
{
Logger.Error(DateTime.Now.ToString() + "\r\n" + exp.Sql + "\r\n" + exp.Parametres);
};
// var basePath = AppContext.BaseDirectory;
// var servicesDllFile = Path.Combine(basePath, "DS.WMS.Core.dll");
// //从程序集拿到实体type集合
// Type[] types = Assembly
// .LoadFrom(servicesDllFile) //如果 .dll报错,可以换成 xxx.exe 有些生成的是exe
// .GetTypes().Where(it => it.FullName.Contains("DS.WMS.Core.") && it.GetCustomAttributes(typeof(SugarTable), true)?.FirstOrDefault() != null) //命名空间过滤,当然你也可以写其他条件过滤
// .ToArray(); //断点调试一下是不是需要的Type,不是需要的在进行过滤
// //全局过滤器
// // var superAdminViewAllData = Convert.ToBoolean(App.GetOptions().SuperAdminViewAllData);
// foreach (var entityType in types)
// {
// // 配置多租户全局过滤器
// if (!entityType.GetProperty("TenantId").IsNull())
// {
// //判断实体类中包含TenantId属性
// //构建动态Lambda
// var lambda = DynamicExpressionParser.ParseLambda
// (new[] { Expression.Parameter(entityType, "it") },
// typeof(bool), $"{nameof(DBEntityTenant.TenantId)} == @0 ",
// user.GetTenantId());
// // dbProvider.QueryFilter.Add(new TableFilterItem