227 lines
9.0 KiB
C#
227 lines
9.0 KiB
C#
using System.Collections.Generic;
|
||
using System.Data;
|
||
using System.IO;
|
||
using System.Reflection;
|
||
using YD_Event.Core.Const;
|
||
using YD_Event.Core.Entities;
|
||
using YD_Event.Core.Enum;
|
||
using YD_Event.Core.Options;
|
||
using Mapster;
|
||
using Newtonsoft.Json;
|
||
using Yitter.IdGenerator;
|
||
|
||
namespace SqlSugar;
|
||
|
||
public static class SqlSugarExtensions
|
||
{
|
||
/// <summary>
|
||
/// 添加 SqlSugar 拓展（适用于单个数据库）
|
||
/// </summary>
|
||
/// <param name="services"></param>
|
||
/// <param name="config">数据库连接对象</param>
|
||
/// <param name="buildAction"></param>
|
||
/// <returns></returns>
|
||
public static IServiceCollection AddSqlSugar(this IServiceCollection services, ConnectionConfigExt config, Action<ISqlSugarClient> buildAction = default)
|
||
{
|
||
return services.AddSqlSugar(new List<ConnectionConfigExt>()
|
||
{
|
||
config
|
||
}, 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<ConnectionConfigExt> configs, Action<SqlSugarClient> buildAction = default)
|
||
{
|
||
SqlSugarScope sqlSugarScope = new SqlSugarScope(configs.Adapt<List<ConnectionConfig>>(), buildAction ?? Aop);
|
||
configs.ForEach(x =>
|
||
{
|
||
Init(sqlSugarScope, x);
|
||
});
|
||
services.AddSingleton<ISqlSugarClient>(sqlSugarScope);//使用单例注入
|
||
services.AddSingleton<ITenant>(sqlSugarScope);
|
||
services.AddUnitOfWork<SqlSugarUnitOfWork>();//事务
|
||
services.AddScoped(typeof(ISqlSugarRepository<>), typeof(SqlSugarRepository<>));//注入泛型仓储
|
||
|
||
return services;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 初始化数据库
|
||
/// </summary>
|
||
/// <param name="client"></param>
|
||
/// <param name="config"></param>
|
||
static void Init(SqlSugarScope client, ConnectionConfigExt config)
|
||
{
|
||
if (!config.EnableInitDb)
|
||
{
|
||
return;
|
||
}
|
||
SqlSugarScopeProvider provider = client.GetConnectionScope(config.ConfigId);
|
||
//创建数据库
|
||
provider.DbMaintenance.CreateDatabase();
|
||
IEnumerable<Type> types = App.EffectiveTypes.Where(x => !x.IsInterface && x.IsClass && !x.IsAbstract && typeof(IEntity).IsAssignableFrom(x));
|
||
provider.CodeFirst.InitTables(types.ToArray());
|
||
|
||
//初始化数据
|
||
InitData(client);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 添加 SqlSugar 服务
|
||
/// </summary>
|
||
/// <param name="services"></param>
|
||
/// <returns></returns>
|
||
public static IServiceCollection AddSqlSugar(this IServiceCollection services)
|
||
{
|
||
var options = App.GetOptions<DbConnectionOptions>();
|
||
ICacheService ormCache = new SqlSugarCache();
|
||
options.Connections.ForEach(x =>
|
||
{
|
||
x.MoreSettings = new ConnMoreSettings()
|
||
{
|
||
//所有 增、删 、改 会自动调用.RemoveDataCache()
|
||
IsAutoRemoveDataCache = true
|
||
};
|
||
//配置缓存
|
||
x.ConfigureExternalServices = new ConfigureExternalServices()
|
||
{
|
||
DataInfoCacheService = ormCache,
|
||
EntityService = ((prop, column) =>
|
||
{
|
||
//如果实体不是主键，并且是可空类型，表列设置为可空(支持string?和string)
|
||
if (column.IsPrimarykey == false && new NullabilityInfoContext()
|
||
.Create(prop).WriteState is NullabilityState.Nullable)
|
||
{
|
||
column.IsNullable = true;
|
||
}
|
||
})
|
||
|
||
};
|
||
|
||
});
|
||
return AddSqlSugar(services, options.Connections, Aop);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 拦截器
|
||
/// </summary>
|
||
private static readonly Action<SqlSugarClient> Aop = (db) =>
|
||
{
|
||
db.Aop.DataExecuting = (_, entityInfo) =>
|
||
{
|
||
switch (entityInfo.OperationType)
|
||
{
|
||
//执行insert时
|
||
case DataFilterType.InsertByObject:
|
||
//自动设置主键
|
||
if (entityInfo.EntityColumnInfo.IsPrimarykey &&
|
||
entityInfo.EntityValue is Entity<long> { Id: 0 } entity)
|
||
{
|
||
entity.Id = YitIdHelper.NextId();
|
||
}
|
||
//如果当前实体继承ICreatedTime就设置创建时间
|
||
if (entityInfo.EntityValue is ICreatedTime createdTime && createdTime.CreatedTime == DateTime.MinValue)
|
||
{
|
||
createdTime.CreatedTime = DateTime.Now;
|
||
}
|
||
if (entityInfo.EntityValue is ICreatedUserId { CreatedUserId: 0 } createdUserId)
|
||
{
|
||
createdUserId.CreatedUserId = App.User.FindFirst(AuthClaimsConst.AuthIdKey)!.Value.Adapt<long>();
|
||
}
|
||
break;
|
||
case DataFilterType.UpdateByObject:
|
||
if (entityInfo.EntityValue is IUpdatedTime updatedTime)
|
||
{
|
||
updatedTime.UpdatedTime = DateTime.Now;
|
||
}
|
||
|
||
break;
|
||
case DataFilterType.DeleteByObject:
|
||
break;
|
||
}
|
||
};
|
||
|
||
// 文档地址：https://www.donet5.com/Home/Doc?typeId=1204
|
||
db.Aop.OnLogExecuting = (sql, parameters) =>
|
||
{
|
||
var originColor = Console.ForegroundColor;
|
||
if (sql.StartsWith("SELECT", StringComparison.OrdinalIgnoreCase))
|
||
Console.ForegroundColor = ConsoleColor.Green;
|
||
if (sql.StartsWith("UPDATE", StringComparison.OrdinalIgnoreCase) || sql.StartsWith("INSERT", StringComparison.OrdinalIgnoreCase))
|
||
Console.ForegroundColor = ConsoleColor.Yellow;
|
||
if (sql.StartsWith("DELETE", StringComparison.OrdinalIgnoreCase))
|
||
Console.ForegroundColor = ConsoleColor.Red;
|
||
Console.WriteLine("【" + DateTime.Now + "——执行SQL】\r\n" + UtilMethods.GetSqlString(db.CurrentConnectionConfig.DbType, sql, parameters) + "\r\n");
|
||
Console.ForegroundColor = originColor;
|
||
//App.PrintToMiniProfiler("SqlSugar", "Info", sql + "\r\n" + db.Utilities.SerializeObject(parameters.ToDictionary(it => it.ParameterName, it => it.Value)));
|
||
};
|
||
db.Aop.OnError = ex =>
|
||
{
|
||
if (ex.Parametres == null) return;
|
||
var originColor = Console.ForegroundColor;
|
||
Console.ForegroundColor = ConsoleColor.DarkRed;
|
||
var pars = db.Utilities.SerializeObject(((SugarParameter[])ex.Parametres).ToDictionary(it => it.ParameterName, it => it.Value));
|
||
Console.WriteLine("【" + DateTime.Now + "——错误SQL】\r\n" + UtilMethods.GetSqlString(db.CurrentConnectionConfig.DbType, ex.Sql, (SugarParameter[])ex.Parametres) + "\r\n");
|
||
Console.ForegroundColor = originColor;
|
||
// App.PrintToMiniProfiler("SqlSugar", "Error", $"{ex.Message}{Environment.NewLine}{ex.Sql}{pars}{Environment.NewLine}");
|
||
};
|
||
|
||
//配置逻辑删除过滤器（查询数据时过滤掉已被标记删除的数据）
|
||
db.QueryFilter.AddTableFilter<ISoftDelete>(it => it.DeleteMark == false);
|
||
};
|
||
|
||
/// <summary>
|
||
/// 初始化基础数据
|
||
/// </summary>
|
||
private static void InitData(SqlSugarScope client)
|
||
{
|
||
List<SysUser> users = new List<SysUser>()
|
||
{
|
||
new SysUser()
|
||
{
|
||
Id = 1,
|
||
Account = "admin",
|
||
Password = "9658b68df5e6208e97ae6c8f4eac6c39",
|
||
Name = "管理员",
|
||
Gender = Gender.Female,
|
||
NickName = "管理员",
|
||
Avatar = "https://oss.okay123.top/oss//2023/07/13/Yln0lzZQLN.jpg",
|
||
CreatedUserId = 1
|
||
},
|
||
new SysUser()
|
||
{
|
||
Id = 50048753934341,
|
||
Account = "test",
|
||
Password = "65caa9533b8d9f336bd07919dd9bf74e",
|
||
Name = "测试",
|
||
Gender = Gender.Female,
|
||
NickName = "测试",
|
||
CreatedUserId = 1
|
||
}
|
||
};
|
||
client.Storageable(users).ToStorage().AsInsertable.ExecuteCommand();
|
||
|
||
string path = Path.Combine(AppContext.BaseDirectory, "InitData");
|
||
var dir = new DirectoryInfo(path);
|
||
var files = dir.GetFiles("*.txt");
|
||
foreach (var file in files)
|
||
{
|
||
using var reader = file.OpenText();
|
||
string s = reader.ReadToEnd();
|
||
var table = JsonConvert.DeserializeObject<DataTable>(s);
|
||
if (table.Rows.Count == 0)
|
||
{
|
||
continue;
|
||
}
|
||
table.TableName = file.Name.Replace(".txt", "");
|
||
|
||
client.Storageable(table).WhereColumns("Id").ToStorage().AsInsertable.ExecuteCommand();
|
||
}
|
||
}
|
||
} |