You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
BookingHeChuan/Myshipping.Core/SqlSugar/SqlSugarSetup.cs

482 lines
23 KiB
C#

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

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")
{
var CreatedUserName = ((dynamic)entityInfo.EntityValue).CreatedUserName;
if (string.IsNullOrEmpty(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")
{
var UpdatedUserId = ((dynamic)entityInfo.EntityValue).UpdatedUserId;
if (UpdatedUserId == null || UpdatedUserId == 0)
entityInfo.SetValue(App.User?.FindFirst(ClaimConst.CLAINM_USERID)?.Value);
}
if (entityInfo.PropertyName == "UpdatedUserName")
{
var UpdatedUserName = ((dynamic)entityInfo.EntityValue).UpdatedUserName;
if (string.IsNullOrEmpty(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;
}
}