xiaochanghai
2 weeks ago
parent
b327491715
commit
bb28ad8ba9
Binary file not shown.
Binary file not shown.
@ -1,180 +0,0 @@ |
||||
using Serilog; |
||||
using SqlSugar; |
||||
using StackExchange.Profiling; |
||||
using Tiobon.Core.LogHelper; |
||||
using Tiobon.Core.Model.Entity; |
||||
using Tiobon.Core.Model.Models.RootTkey; |
||||
using Tiobon.Core.Model.Tenants; |
||||
|
||||
namespace Tiobon.Core.Common.DB.Aop; |
||||
|
||||
public static class SqlSugarAop |
||||
{ |
||||
public static void OnLogExecuting(ISqlSugarClient sqlSugarScopeProvider, string user, string table, string operate, string sql, SugarParameter[] p, ConnectionConfig config) |
||||
{ |
||||
try |
||||
{ |
||||
MiniProfiler.Current.CustomTiming($"ConnId:[{config.ConfigId}] SQL：", GetParas(p) + "【SQL语句】：" + sql); |
||||
|
||||
if (!AppSettings.app(new string[] { "AppSettings", "SqlAOP", "Enabled" }).ObjToBool()) return; |
||||
|
||||
if (AppSettings.app(new string[] { "AppSettings", "SqlAOP", "LogToConsole", "Enabled" }).ObjToBool() || |
||||
AppSettings.app(new string[] { "AppSettings", "SqlAOP", "LogToFile", "Enabled" }).ObjToBool() || |
||||
AppSettings.app(new string[] { "AppSettings", "SqlAOP", "LogToDB", "Enabled" }).ObjToBool()) |
||||
{ |
||||
using (LogContextExtension.Create.SqlAopPushProperty(sqlSugarScopeProvider)) |
||||
{ |
||||
Log.Information("------------------ \r\n User:[{User}] Table:[{Table}] Operate:[{Operate}] ConnId:[{ConnId}]【SQL语句】: \r\n {Sql}", |
||||
user, table, operate, config.ConfigId, UtilMethods.GetNativeSql(sql, p)); |
||||
} |
||||
} |
||||
} |
||||
catch (Exception e) |
||||
{ |
||||
Log.Error("Error occured OnLogExcuting:" + e); |
||||
} |
||||
} |
||||
|
||||
public static void DataExecuting(object oldValue, DataFilterModel entityInfo) |
||||
{ |
||||
if (entityInfo.EntityValue is BasePoco rootEntity) |
||||
if (rootEntity.Id == 0) |
||||
rootEntity.Id = SnowFlakeSingle.Instance.NextId(); |
||||
|
||||
if (entityInfo.EntityValue is BaseEntity baseEntity) |
||||
{ |
||||
// 新增操作 |
||||
if (entityInfo.OperationType == DataFilterType.InsertByObject) |
||||
if (baseEntity.CreateTime == DateTime.MinValue) |
||||
baseEntity.CreateTime = DateTime.Now; |
||||
|
||||
if (entityInfo.OperationType == DataFilterType.UpdateByObject) |
||||
baseEntity.ModifyTime = DateTime.Now; |
||||
|
||||
if (App.User?.ID > 0) |
||||
{ |
||||
if (baseEntity is ITenantEntity tenant && App.User.TenantId > 0) |
||||
{ |
||||
if (tenant.TenantId == 0) |
||||
tenant.TenantId = App.User.TenantId; |
||||
} |
||||
|
||||
switch (entityInfo.OperationType) |
||||
{ |
||||
case DataFilterType.UpdateByObject: |
||||
baseEntity.ModifyId = App.User.ID; |
||||
baseEntity.ModifyBy = App.User.Name; |
||||
break; |
||||
case DataFilterType.InsertByObject: |
||||
if (baseEntity.CreateBy.IsNullOrEmpty() || baseEntity.CreateId is null or <= 0) |
||||
{ |
||||
baseEntity.CreateId = App.User.ID; |
||||
baseEntity.CreateBy = App.User.Name; |
||||
} |
||||
|
||||
break; |
||||
} |
||||
} |
||||
} |
||||
else |
||||
{ |
||||
//兼容以前的表 |
||||
//这里要小心 在AOP里用反射 数据量多性能就会有问题 |
||||
//要么都统一使用基类 |
||||
//要么考虑老的表没必要兼容老的表 |
||||
// |
||||
|
||||
var getType = entityInfo.EntityValue.GetType(); |
||||
int userId = Convert.ToInt32(App.User.ID); |
||||
switch (entityInfo.OperationType) |
||||
{ |
||||
case DataFilterType.InsertByObject: |
||||
var dyCreateBy = getType.GetProperty("CreateBy"); |
||||
var dyCreateTime = getType.GetProperty("CreateTime"); |
||||
BasePoco ent = entityInfo.EntityValue as BasePoco; |
||||
if (ent != null && ent.Id == 0) |
||||
ent.Id = SnowFlakeSingle.Instance.NextId(); |
||||
if (App.User?.ID > 0 && dyCreateBy != null && dyCreateBy.GetValue(entityInfo.EntityValue) == null) |
||||
{ |
||||
try |
||||
{ |
||||
dyCreateBy.SetValue(entityInfo.EntityValue, Convert.ToInt32(App.User.ID)); |
||||
|
||||
} |
||||
catch (Exception) |
||||
{ |
||||
dyCreateBy.SetValue(entityInfo.EntityValue, App.User.ID); |
||||
|
||||
} |
||||
} |
||||
|
||||
if ((dyCreateTime != null && dyCreateTime.GetValue(entityInfo.EntityValue) is null) || (dyCreateTime != null && dyCreateTime.GetValue(entityInfo.EntityValue) != null && (DateTime)dyCreateTime.GetValue(entityInfo.EntityValue) == DateTime.MinValue)) |
||||
dyCreateTime.SetValue(entityInfo.EntityValue, DateTime.Now); |
||||
|
||||
break; |
||||
case DataFilterType.UpdateByObject: |
||||
if (entityInfo.PropertyName == "UpdateBy") |
||||
{ |
||||
var UpdateBy = getType.GetProperty("UpdateBy"); |
||||
if (App.User?.ID > 0 && UpdateBy != null) |
||||
{ |
||||
try |
||||
{ |
||||
UpdateBy.SetValue(entityInfo.EntityValue, Convert.ToInt32(App.User.ID)); |
||||
|
||||
} |
||||
catch (Exception) |
||||
{ |
||||
UpdateBy.SetValue(entityInfo.EntityValue, App.User.ID); |
||||
|
||||
} |
||||
} |
||||
} |
||||
else if (entityInfo.PropertyName == "UpdateTime") |
||||
{ |
||||
|
||||
var dyModifyTime = getType.GetProperty("UpdateTime"); |
||||
|
||||
if (dyModifyTime != null) |
||||
dyModifyTime.SetValue(entityInfo.EntityValue, DateTime.Now); |
||||
} |
||||
//else if (entityInfo.PropertyName == "UpdateTime") |
||||
//{ |
||||
|
||||
// HttpRequest request = UserContext.Context.Request; |
||||
// var api = request.Path.ObjToString().TrimEnd('/').ToLower(); |
||||
//} |
||||
//else if (entityInfo.PropertyName == "UpdateTime") |
||||
//{ |
||||
|
||||
// var dyModifyTime = getType.GetProperty("UpdateTime"); |
||||
|
||||
// if (dyModifyTime != null) |
||||
// dyModifyTime.SetValue(entityInfo.EntityValue, DateTime.Now); |
||||
//} |
||||
break; |
||||
} |
||||
} |
||||
} |
||||
|
||||
private static string GetWholeSql(SugarParameter[] paramArr, string sql) |
||||
{ |
||||
foreach (var param in paramArr) |
||||
{ |
||||
sql = sql.Replace(param.ParameterName, $@"'{param.Value.ObjToString()}'"); |
||||
} |
||||
|
||||
return sql; |
||||
} |
||||
|
||||
private static string GetParas(SugarParameter[] pars) |
||||
{ |
||||
string key = "【SQL参数】："; |
||||
foreach (var param in pars) |
||||
{ |
||||
key += $"{param.ParameterName}:{param.Value}\n"; |
||||
} |
||||
|
||||
return key; |
||||
} |
||||
} |
||||
@ -1,13 +0,0 @@ |
||||
using SqlSugar; |
||||
using System.Reflection; |
||||
|
||||
namespace Tiobon.Core.Common.DB.Extension; |
||||
|
||||
public static class DbEntityException |
||||
{ |
||||
public static object GetEntityTenant(this Type type) |
||||
{ |
||||
var tenant = type.GetCustomAttribute<TenantAttribute>(); |
||||
return tenant?.configId; |
||||
} |
||||
} |
||||
@ -1,44 +0,0 @@ |
||||
using SqlSugar; |
||||
using System.Reflection; |
||||
using System.Reflection.Emit; |
||||
|
||||
namespace Tiobon.Core.Common.DB.Extension; |
||||
|
||||
public static class DynamicBuildException |
||||
{ |
||||
private static List<CustomAttributeBuilder> GetEntityAttr(this DynamicBuilder builder) |
||||
{ |
||||
FieldInfo fieldInfo = builder.GetType().GetField("entityAttr", BindingFlags.Instance | BindingFlags.NonPublic); |
||||
List<CustomAttributeBuilder> entityAttr = (List<CustomAttributeBuilder>) fieldInfo.GetValue(builder); |
||||
return entityAttr; |
||||
} |
||||
|
||||
private static CustomAttributeBuilder CreateIndex(SugarIndexAttribute indexAttribute) |
||||
{ |
||||
Type type = typeof(SugarIndexAttribute); |
||||
var constructorTypes = new List<Type>() {typeof(string)}; |
||||
for (int i = 0; i < indexAttribute.IndexFields.Count; i++) |
||||
{ |
||||
constructorTypes.AddRange(new[] {typeof(string), typeof(OrderByType)}); |
||||
} |
||||
|
||||
constructorTypes.Add(typeof(bool)); |
||||
|
||||
var values = new List<object>() {indexAttribute.IndexName}; |
||||
foreach (var indexField in indexAttribute.IndexFields) |
||||
{ |
||||
values.AddRange(new object[] {indexField.Key, indexField.Value}); |
||||
} |
||||
|
||||
values.Add(indexAttribute.IsUnique); |
||||
return new CustomAttributeBuilder(type.GetConstructor(constructorTypes.ToArray())!, values.ToArray()); |
||||
} |
||||
|
||||
public static DynamicProperyBuilder CreateIndex(this DynamicProperyBuilder builder, SugarIndexAttribute indexAttribute) |
||||
{ |
||||
var classBuilder = builder.baseBuilder; |
||||
var entityAttr = classBuilder.GetEntityAttr(); |
||||
entityAttr.Add(CreateIndex(indexAttribute)); |
||||
return builder; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue