|
|
using Furion;
|
|
|
using Furion.Logging.Extensions;
|
|
|
using Myshipping.Core.Entity;
|
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
|
using SqlSugar;
|
|
|
using System;
|
|
|
using System.Collections.Generic;
|
|
|
using System.ComponentModel.DataAnnotations;
|
|
|
using System.ComponentModel.DataAnnotations.Schema;
|
|
|
using System.Linq;
|
|
|
using System.Linq.Dynamic.Core;
|
|
|
using System.Linq.Expressions;
|
|
|
|
|
|
namespace Myshipping.Core;
|
|
|
|
|
|
public static class SqlSugarSetup
|
|
|
{
|
|
|
//public static void SqlSugarConfigure(this IServiceCollection services)
|
|
|
//{
|
|
|
// #region 配置sqlsuagr
|
|
|
// List<ConnectionConfig> connectConfigList = new List<ConnectionConfig>();
|
|
|
// //数据库序号从0开始,默认数据库为0
|
|
|
// var config = App.GetOptions<ConnectionStringsOptions>();
|
|
|
// //默认数据库
|
|
|
// connectConfigList.Add(new ConnectionConfig
|
|
|
// {
|
|
|
// ConnectionString = config.DefaultDbString,
|
|
|
// DbType = (DbType)Convert.ToInt32(Enum.Parse(typeof(DbType), config.DefaultDbType)),
|
|
|
// IsAutoCloseConnection = true,
|
|
|
// ConfigId = config.DefaultDbNumber,
|
|
|
// InitKeyType = InitKeyType.Attribute,
|
|
|
// MoreSettings = new ConnMoreSettings()
|
|
|
// {
|
|
|
// IsAutoRemoveDataCache = true//自动清理缓存
|
|
|
|
|
|
// },
|
|
|
// ConfigureExternalServices = new ConfigureExternalServices()
|
|
|
// {
|
|
|
// EntityNameService = (type, entity) =>
|
|
|
// {
|
|
|
// var attributes = type.GetCustomAttributes(true);
|
|
|
// if (attributes.Any(it => it is TableAttribute))
|
|
|
// {
|
|
|
// entity.DbTableName = (attributes.First(it => it is TableAttribute) as TableAttribute).Name;
|
|
|
// }
|
|
|
// },
|
|
|
// EntityService = (type, column) =>
|
|
|
// {
|
|
|
// var attributes = type.GetCustomAttributes(true);
|
|
|
// if (attributes.Any(it => it is KeyAttribute))// by attribute set primarykey
|
|
|
// {
|
|
|
// column.IsPrimarykey = true; //有哪些特性可以看 1.2 特性明细
|
|
|
// }
|
|
|
// if (attributes.Any(it => it is ColumnAttribute))
|
|
|
// {
|
|
|
// column.DbColumnName = (attributes.First(it => it is ColumnAttribute) as ColumnAttribute).Name;
|
|
|
// }
|
|
|
// }
|
|
|
// }
|
|
|
|
|
|
// });
|
|
|
// if (config.DbConfigs == null)
|
|
|
// config.DbConfigs = new List<DbConfig>();
|
|
|
// //业务数据库集合
|
|
|
// foreach (var item in config.DbConfigs)
|
|
|
// {
|
|
|
// //防止数据库重复，导致的事务异常
|
|
|
// if (connectConfigList.Any(a => a.ConfigId == (dynamic)item.DbNumber || a.ConnectionString == item.DbString))
|
|
|
//{
|
|
|
// continue;
|
|
|
//}
|
|
|
// connectConfigList.Add(new ConnectionConfig
|
|
|
// {
|
|
|
// ConnectionString = item.DbString,
|
|
|
// DbType = (DbType)Convert.ToInt32(Enum.Parse(typeof(DbType), item.DbType)),
|
|
|
// IsAutoCloseConnection = true,
|
|
|
// ConfigId = item.DbNumber,
|
|
|
// InitKeyType = InitKeyType.Attribute,
|
|
|
// MoreSettings = new ConnMoreSettings()
|
|
|
// {
|
|
|
// IsAutoRemoveDataCache = true//自动清理缓存
|
|
|
// }
|
|
|
// });
|
|
|
// }
|
|
|
// List<Type> types = App.EffectiveTypes.Where(a => !a.IsAbstract && a.IsClass && a.GetCustomAttributes(typeof(SugarTable), true)?.FirstOrDefault() != null).ToList();
|
|
|
// //sugar action
|
|
|
// Action<ISqlSugarClient> configure = db =>
|
|
|
// {
|
|
|
// connectConfigList.ForEach(config => {
|
|
|
// string temp = config.ConfigId;
|
|
|
// var _db = db.AsTenant().GetConnection(temp);
|
|
|
// _db.CurrentConnectionConfig.ConfigureExternalServices = new ConfigureExternalServices()
|
|
|
// {
|
|
|
// DataInfoCacheService = new SqlSugarCache()//配置我们创建的缓存类
|
|
|
// };
|
|
|
// //执行超时时间
|
|
|
// _db.Ado.CommandTimeOut = 30;
|
|
|
// _db.Aop.OnLogExecuting = (sql, pars) =>
|
|
|
// {
|
|
|
// if (sql.StartsWith("SELECT"))
|
|
|
// {
|
|
|
// Console.ForegroundColor = ConsoleColor.Green;
|
|
|
// }
|
|
|
// if (sql.StartsWith("UPDATE") || sql.StartsWith("INSERT"))
|
|
|
// {
|
|
|
// Console.ForegroundColor = ConsoleColor.White;
|
|
|
// }
|
|
|
// if (sql.StartsWith("DELETE"))
|
|
|
// {
|
|
|
// Console.ForegroundColor = ConsoleColor.Blue;
|
|
|
// }
|
|
|
// //App.PrintToMiniProfiler("SqlSugar", "Info", sql + "\r\n" + _db.Utilities.SerializeObject(pars.ToDictionary(it => it.ParameterName, it => it.Value)));
|
|
|
// Console.WriteLine("Sql:" + "\r\n\r\n" + UtilMethods.GetSqlString(db.CurrentConnectionConfig.DbType, sql, pars));
|
|
|
// App.PrintToMiniProfiler("SqlSugar", "Info", UtilMethods.GetSqlString(db.CurrentConnectionConfig.DbType, sql, pars));
|
|
|
// };
|
|
|
|
|
|
|
|
|
// _db.Aop.DataExecuting = (oldValue, entityInfo) =>
|
|
|
// {
|
|
|
// // 新增操作
|
|
|
// if (entityInfo.OperationType == DataFilterType.InsertByObject)
|
|
|
// {
|
|
|
// // 主键(long)-赋值雪花Id
|
|
|
// if (entityInfo.EntityColumnInfo.IsPrimarykey && entityInfo.EntityColumnInfo.PropertyInfo.PropertyType == typeof(long)) {
|
|
|
// var id = ((dynamic)entityInfo.EntityValue).Id;
|
|
|
// if (id == null || id == 0)
|
|
|
// entityInfo.SetValue(Yitter.IdGenerator.YitIdHelper.NextId());
|
|
|
// }
|
|
|
|
|
|
|
|
|
// if (entityInfo.PropertyName == "CreatedTime")
|
|
|
// entityInfo.SetValue(DateTime.Now);
|
|
|
// if (App.User != null)
|
|
|
// {
|
|
|
// if (entityInfo.PropertyName == "TenantId")
|
|
|
// {
|
|
|
// var tenantId = ((dynamic)entityInfo.EntityValue).TenantId;
|
|
|
// if (tenantId == null || tenantId == 0)
|
|
|
// entityInfo.SetValue(App.User.FindFirst(ClaimConst.TENANT_ID)?.Value);
|
|
|
// }
|
|
|
// if (entityInfo.PropertyName == "CreatedUserId") {
|
|
|
// var createUserId = ((dynamic)entityInfo.EntityValue).CreatedUserId;
|
|
|
// if (createUserId == null || createUserId == 0)
|
|
|
// entityInfo.SetValue(App.User.FindFirst(ClaimConst.CLAINM_USERID)?.Value);
|
|
|
// }
|
|
|
|
|
|
// if (entityInfo.PropertyName == "CreatedUserName")
|
|
|
// entityInfo.SetValue(App.User.FindFirst(ClaimConst.CLAINM_NAME)?.Value);
|
|
|
// //if (entityInfo.PropertyName == "CreateOrgId")
|
|
|
// // entityInfo.SetValue(App.User.FindFirst(ClaimConst.OrgId)?.Value);
|
|
|
// }
|
|
|
// }
|
|
|
// // 更新操作
|
|
|
// if (entityInfo.OperationType == DataFilterType.UpdateByObject)
|
|
|
// {
|
|
|
// if (entityInfo.PropertyName == "UpdatedTime")
|
|
|
// entityInfo.SetValue(DateTime.Now);
|
|
|
// if (entityInfo.PropertyName == "UpdatedUserId")
|
|
|
// entityInfo.SetValue(App.User?.FindFirst(ClaimConst.CLAINM_USERID)?.Value);
|
|
|
// if (entityInfo.PropertyName == "UpdatedUserName")
|
|
|
// entityInfo.SetValue(App.User?.FindFirst(ClaimConst.CLAINM_NAME)?.Value);
|
|
|
|
|
|
// }
|
|
|
// };
|
|
|
|
|
|
// //全局过滤器
|
|
|
// var superAdminViewAllData = Convert.ToBoolean(App.GetOptions<SystemSettingsOptions>().SuperAdminViewAllData);
|
|
|
// foreach (var entityType in types)
|
|
|
// {
|
|
|
// // 配置多租户全局过滤器
|
|
|
// if (!entityType.GetProperty(ClaimConst.TENANT_ID).IsEmpty())
|
|
|
// { //判断实体类中包含TenantId属性
|
|
|
// //构建动态Lambda
|
|
|
// var lambda = DynamicExpressionParser.ParseLambda
|
|
|
// (new[] { Expression.Parameter(entityType, "it") },
|
|
|
// typeof(bool), $"{nameof(DBEntityTenant.TenantId)} == @0 or (@1 and @2)",
|
|
|
// GetTenantId(), IsSuperAdmin(), superAdminViewAllData);
|
|
|
// _db.QueryFilter.Add(new TableFilterItem<object>(entityType, lambda)); //将Lambda传入过滤器
|
|
|
// }
|
|
|
// // 配置加删除全局过滤器
|
|
|
// if (!entityType.GetProperty(CommonConst.DELETE_FIELD).IsEmpty())
|
|
|
// { //判断实体类中包含IsDeleted属性
|
|
|
// //构建动态Lambda
|
|
|
// var lambda = DynamicExpressionParser.ParseLambda
|
|
|
// (new[] { Expression.Parameter(entityType, "it") },
|
|
|
// typeof(bool), $"{nameof(DEntityBase.IsDeleted)} == @0",
|
|
|
// false);
|
|
|
// _db.QueryFilter.Add(new TableFilterItem<object>(entityType, lambda)
|
|
|
// {
|
|
|
// IsJoinQuery = true
|
|
|
// }); //将Lambda传入过滤器
|
|
|
// }
|
|
|
// }
|
|
|
// });
|
|
|
// };
|
|
|
// services.AddSqlSugar(connectConfigList, configure);
|
|
|
// #endregion
|
|
|
//}
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
/// SqlsugarScope的配置
|
|
|
/// Scope必须用单例注入
|
|
|
/// 不可以用Action委托注入
|
|
|
/// </summary>
|
|
|
/// <param name="services"></param>
|
|
|
public static void SqlSugarScopeConfigure(this IServiceCollection services)
|
|
|
{
|
|
|
//数据库序号从0开始,默认数据库为0
|
|
|
var config = App.GetOptions<ConnectionStringsOptions>();
|
|
|
|
|
|
//默认数据库
|
|
|
List<DbConfig> dbList = new List<DbConfig>();
|
|
|
DbConfig defaultdb = new DbConfig()
|
|
|
{
|
|
|
DbNumber = config.DefaultDbNumber,
|
|
|
DbString = config.DefaultDbString,
|
|
|
DbType = config.DefaultDbType
|
|
|
};
|
|
|
dbList.Add(defaultdb);
|
|
|
//业务数据库集合
|
|
|
foreach (var item in config.DbConfigs)
|
|
|
{
|
|
|
dbList.Add(item);
|
|
|
}
|
|
|
|
|
|
List<ConnectionConfig> connectConfigList = new List<ConnectionConfig>();
|
|
|
|
|
|
foreach (var item in dbList)
|
|
|
{
|
|
|
//防止数据库重复，导致的事务异常
|
|
|
if (connectConfigList.Any(a => a.ConfigId == (dynamic)item.DbNumber || a.ConnectionString == item.DbString))
|
|
|
{
|
|
|
continue;
|
|
|
}
|
|
|
connectConfigList.Add(new ConnectionConfig()
|
|
|
{
|
|
|
ConnectionString = item.DbString,
|
|
|
DbType = (DbType)Convert.ToInt32(Enum.Parse(typeof(DbType), item.DbType)),
|
|
|
IsAutoCloseConnection = true,
|
|
|
ConfigId = item.DbNumber,
|
|
|
InitKeyType = InitKeyType.Attribute,
|
|
|
MoreSettings = new ConnMoreSettings()
|
|
|
{
|
|
|
IsAutoRemoveDataCache = true//自动清理缓存
|
|
|
|
|
|
},
|
|
|
ConfigureExternalServices = new ConfigureExternalServices()
|
|
|
{
|
|
|
DataInfoCacheService = new SqlSugarCache(),
|
|
|
EntityNameService = (type, entity) =>
|
|
|
{
|
|
|
var attributes = type.GetCustomAttributes(true);
|
|
|
if (attributes.Any(it => it is TableAttribute))
|
|
|
{
|
|
|
entity.DbTableName = (attributes.First(it => it is TableAttribute) as TableAttribute).Name;
|
|
|
}
|
|
|
},
|
|
|
EntityService = (type, column) =>
|
|
|
{
|
|
|
var attributes = type.GetCustomAttributes(true);
|
|
|
if (attributes.Any(it => it is KeyAttribute))// by attribute set primarykey
|
|
|
{
|
|
|
column.IsPrimarykey = true; //有哪些特性可以看 1.2 特性明细
|
|
|
}
|
|
|
if (attributes.Any(it => it is ColumnAttribute))
|
|
|
{
|
|
|
column.DbColumnName = (attributes.First(it => it is ColumnAttribute) as ColumnAttribute).Name;
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
});
|
|
|
}
|
|
|
|
|
|
List<Type> types = App.EffectiveTypes.Where(a => !a.IsAbstract && a.IsClass && a.GetCustomAttributes(typeof(SugarTable), true)?.FirstOrDefault() != null).ToList();
|
|
|
|
|
|
SqlSugarScope sqlSugarScope = new SqlSugarScope(connectConfigList,
|
|
|
//全局上下文生效
|
|
|
db =>
|
|
|
{
|
|
|
/*
|
|
|
* 默认只会配置到第一个数据库，这里按照官方文档进行多数据库/多租户文档的说明进行循环配置
|
|
|
*/
|
|
|
foreach (var c in connectConfigList)
|
|
|
{
|
|
|
var dbProvider = db.GetConnectionScope((string)c.ConfigId);
|
|
|
//执行超时时间
|
|
|
dbProvider.Ado.CommandTimeOut = 30;
|
|
|
|
|
|
dbProvider.Aop.OnLogExecuting = (sql, pars) =>
|
|
|
{
|
|
|
if (sql.StartsWith("SELECT"))
|
|
|
{
|
|
|
Console.ForegroundColor = ConsoleColor.Green;
|
|
|
}
|
|
|
if (sql.StartsWith("UPDATE") || sql.StartsWith("INSERT"))
|
|
|
{
|
|
|
Console.ForegroundColor = ConsoleColor.White;
|
|
|
}
|
|
|
if (sql.StartsWith("DELETE"))
|
|
|
{
|
|
|
Console.ForegroundColor = ConsoleColor.Blue;
|
|
|
}
|
|
|
//Console.WriteLine("Sql:" + "\r\n\r\n" + UtilMethods.GetSqlString(c.DbType, sql, pars));
|
|
|
App.PrintToMiniProfiler("SqlSugar", "Info", UtilMethods.GetSqlString(c.DbType, sql, pars));
|
|
|
$"DB:{c.ConfigId}, Sql:\r\n\r\n { UtilMethods.GetSqlString(c.DbType, sql, pars)}".LogInformation();
|
|
|
|
|
|
|
|
|
};
|
|
|
|
|
|
dbProvider.Aop.DataExecuting = (oldValue, entityInfo) =>
|
|
|
{
|
|
|
// 新增操作
|
|
|
if (entityInfo.OperationType == DataFilterType.InsertByObject)
|
|
|
{
|
|
|
// 主键(long)-赋值雪花Id
|
|
|
if (entityInfo.EntityColumnInfo.IsPrimarykey && entityInfo.EntityColumnInfo.PropertyInfo.PropertyType == typeof(long))
|
|
|
{
|
|
|
var id = ((dynamic)entityInfo.EntityValue).Id;
|
|
|
if (id == null || id == 0)
|
|
|
entityInfo.SetValue(Yitter.IdGenerator.YitIdHelper.NextId());
|
|
|
}
|
|
|
|
|
|
|
|
|
if (entityInfo.PropertyName == "CreatedTime")
|
|
|
entityInfo.SetValue(DateTime.Now);
|
|
|
if (App.User != null)
|
|
|
{
|
|
|
if (entityInfo.PropertyName == "TenantId")
|
|
|
{
|
|
|
var tenantId = ((dynamic)entityInfo.EntityValue).TenantId;
|
|
|
if (tenantId == null || tenantId == 0)
|
|
|
entityInfo.SetValue(App.User.FindFirst(ClaimConst.TENANT_ID)?.Value);
|
|
|
}
|
|
|
if (entityInfo.PropertyName == "TenantName")
|
|
|
{
|
|
|
var tenantName = ((dynamic)entityInfo.EntityValue).TenantName;
|
|
|
if (string.IsNullOrEmpty(tenantName))
|
|
|
entityInfo.SetValue(App.User.FindFirst(ClaimConst.TENANT_NAME)?.Value);
|
|
|
}
|
|
|
if (entityInfo.PropertyName == "CreatedUserId")
|
|
|
{
|
|
|
var createUserId = ((dynamic)entityInfo.EntityValue).CreatedUserId;
|
|
|
if (createUserId == null || createUserId == 0)
|
|
|
entityInfo.SetValue(App.User.FindFirst(ClaimConst.CLAINM_USERID)?.Value);
|
|
|
}
|
|
|
|
|
|
if (entityInfo.PropertyName == "CreatedUserName")
|
|
|
entityInfo.SetValue(App.User.FindFirst(ClaimConst.CLAINM_NAME)?.Value);
|
|
|
}
|
|
|
}
|
|
|
// 更新操作
|
|
|
if (entityInfo.OperationType == DataFilterType.UpdateByObject)
|
|
|
{
|
|
|
if (entityInfo.PropertyName == "UpdatedTime")
|
|
|
entityInfo.SetValue(DateTime.Now);
|
|
|
if (entityInfo.PropertyName == "UpdatedUserId")
|
|
|
entityInfo.SetValue(App.User?.FindFirst(ClaimConst.CLAINM_USERID)?.Value);
|
|
|
if (entityInfo.PropertyName == "UpdatedUserName")
|
|
|
entityInfo.SetValue(App.User?.FindFirst(ClaimConst.CLAINM_NAME)?.Value);
|
|
|
|
|
|
}
|
|
|
};
|
|
|
|
|
|
/*
|
|
|
* 使用 SqlSugarScope 循环配置此项的时候会覆盖整个 ConfigureExternalServices，
|
|
|
* 移动到 New ConnectionConfig中配置
|
|
|
*/
|
|
|
//db.CurrentConnectionConfig.ConfigureExternalServices = new ConfigureExternalServices()
|
|
|
//{
|
|
|
// DataInfoCacheService = new SqlSugarCache()//配置我们创建的缓存类
|
|
|
//};
|
|
|
|
|
|
//全局过滤器
|
|
|
var superAdminViewAllData = Convert.ToBoolean(App.GetOptions<SystemSettingsOptions>().SuperAdminViewAllData);
|
|
|
foreach (var entityType in types)
|
|
|
{
|
|
|
// 配置多租户全局过滤器
|
|
|
if (!entityType.GetProperty(ClaimConst.TENANT_ID).IsEmpty())
|
|
|
{ //判断实体类中包含TenantId属性
|
|
|
//构建动态Lambda
|
|
|
var lambda = DynamicExpressionParser.ParseLambda
|
|
|
(new[] { Expression.Parameter(entityType, "it") },
|
|
|
typeof(bool), $"{nameof(DBEntityTenant.TenantId)} == @0 or (@1 and @2)",
|
|
|
GetTenantId(), IsSuperAdmin(), superAdminViewAllData);
|
|
|
dbProvider.QueryFilter.Add(new TableFilterItem<object>(entityType, lambda)); //将Lambda传入过滤器
|
|
|
}
|
|
|
// 配置加删除全局过滤器
|
|
|
if (!entityType.GetProperty(CommonConst.DELETE_FIELD).IsEmpty())
|
|
|
{ //判断实体类中包含IsDeleted属性
|
|
|
//构建动态Lambda
|
|
|
var lambda = DynamicExpressionParser.ParseLambda
|
|
|
(new[] { Expression.Parameter(entityType, "it") },
|
|
|
typeof(bool), $"{nameof(DEntityBase.IsDeleted)} == @0",
|
|
|
false);
|
|
|
dbProvider.QueryFilter.Add(new TableFilterItem<object>(entityType, lambda)
|
|
|
{
|
|
|
IsJoinQuery = true
|
|
|
}); //将Lambda传入过滤器
|
|
|
}
|
|
|
}
|
|
|
|
|
|
}
|
|
|
});
|
|
|
services.AddSingleton<ISqlSugarClient>(sqlSugarScope);
|
|
|
// 注册 SqlSugar 仓储
|
|
|
services.AddScoped(typeof(SqlSugarRepository<>));
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
/// 获取当前租户id
|
|
|
/// </summary>
|
|
|
/// <returns></returns>
|
|
|
private static object GetTenantId()
|
|
|
{
|
|
|
if (App.User == null) return null;
|
|
|
return App.User.FindFirst(ClaimConst.TENANT_ID)?.Value;
|
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
|
/// 判断是不是超级管理员
|
|
|
/// </summary>
|
|
|
/// <returns></returns>
|
|
|
private static bool IsSuperAdmin()
|
|
|
{
|
|
|
if (App.User == null) return false;
|
|
|
return App.User.FindFirst(ClaimConst.CLAINM_SUPERADMIN)?.Value == AdminType.SuperAdmin.GetHashCode().ToString();
|
|
|
}
|
|
|
/// <summary>
|
|
|
/// 添加 SqlSugar 拓展
|
|
|
/// </summary>
|
|
|
/// <param name="services"></param>
|
|
|
/// <param name="config"></param>
|
|
|
/// <param name="buildAction"></param>
|
|
|
/// <returns></returns>
|
|
|
public static IServiceCollection AddSqlSugar(this IServiceCollection services, ConnectionConfig config, Action<ISqlSugarClient> buildAction = default)
|
|
|
{
|
|
|
var list = new List<ConnectionConfig>();
|
|
|
list.Add(config);
|
|
|
return services.AddSqlSugar(list, buildAction);
|
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
|
/// 添加 SqlSugar 拓展
|
|
|
/// </summary>
|
|
|
/// <param name="services"></param>
|
|
|
/// <param name="configs"></param>
|
|
|
/// <param name="buildAction"></param>
|
|
|
/// <returns></returns>
|
|
|
public static IServiceCollection AddSqlSugar(this IServiceCollection services, List<ConnectionConfig> configs, Action<ISqlSugarClient> buildAction = default)
|
|
|
{
|
|
|
// 注册 SqlSugar 客户端
|
|
|
services.AddScoped<ISqlSugarClient>(u =>
|
|
|
{
|
|
|
var db = new SqlSugarClient(configs);
|
|
|
buildAction?.Invoke(db);
|
|
|
return db;
|
|
|
});
|
|
|
|
|
|
// 注册 SqlSugar 仓储
|
|
|
services.AddScoped(typeof(SqlSugarRepository<>));
|
|
|
return services;
|
|
|
}
|
|
|
}
|