218 lines
9.0 KiB
C#
218 lines
9.0 KiB
C#
using BZPT.Domains.Application;
|
||
using BZPT.Domains.Entity.Sys;
|
||
using Castle.Core.Logging;
|
||
using Microsoft.Extensions.Logging;
|
||
using Mysqlx.Expr;
|
||
using Newtonsoft.Json;
|
||
using NPlatform.Filters;
|
||
using NPlatform.Infrastructure.IdGenerators;
|
||
using NPOI.HPSF;
|
||
using NPOI.SS.Formula.Functions;
|
||
using Serilog.Context;
|
||
using Serilog.Events;
|
||
using SqlSugar;
|
||
using System.ComponentModel.DataAnnotations;
|
||
using System.ComponentModel.DataAnnotations.Schema;
|
||
using System.Configuration;
|
||
using System.Data.SqlTypes;
|
||
using System.Runtime.CompilerServices;
|
||
|
||
namespace BZPT.Repositories
|
||
{
|
||
public partial class DBContext : SqlSugarClient
|
||
{
|
||
private readonly string _connectionString;
|
||
private readonly DbType _dbProvider;
|
||
private readonly int _timeOut;
|
||
|
||
public DBContext(string connectString, DbType dbProvider, string ConfigId = "default", int timeOut = 180)
|
||
: base(new ConnectionConfig()
|
||
{
|
||
DbType = dbProvider,
|
||
ConnectionString = connectString,
|
||
IsAutoCloseConnection = true,
|
||
ConfigId = ConfigId,
|
||
ConfigureExternalServices = new ConfigureExternalServices()
|
||
{
|
||
EntityService = (property, column) =>
|
||
{
|
||
var attributes = property.GetCustomAttributes(true);
|
||
if (attributes.Any(it => it is KeyAttribute))
|
||
{
|
||
column.IsPrimarykey = true;
|
||
}
|
||
if (attributes.Any(it => it is NotMappedAttribute))
|
||
{
|
||
column.IsIgnore = true;
|
||
}
|
||
if (attributes.Any(t => t is ColumnAttribute))
|
||
{
|
||
var attr = attributes.First(it => it is ColumnAttribute) as ColumnAttribute;
|
||
if(!string.IsNullOrEmpty(attr?.Name))
|
||
column.DbColumnName = attr?.Name;
|
||
}
|
||
},
|
||
EntityNameService = (type, entity) =>
|
||
{
|
||
var attributes = type.GetCustomAttributes(true);
|
||
if (attributes.Any(it => it is TableAttribute))
|
||
{
|
||
var attr = attributes.First(it => it is TableAttribute) as TableAttribute;
|
||
if(!string.IsNullOrEmpty(attr?.Name))
|
||
entity.DbTableName = attr?.Name;
|
||
}
|
||
}
|
||
}
|
||
}, dbAction)
|
||
{
|
||
_connectionString = connectString;
|
||
_dbProvider = dbProvider;
|
||
_timeOut = timeOut;
|
||
}
|
||
|
||
public static Action<SqlSugarClient> dbAction = db =>
|
||
{
|
||
// SQL执行完
|
||
db.Aop.OnLogExecuted = (sql, pars) =>
|
||
{
|
||
try
|
||
{
|
||
var strSql = UtilMethods.GetNativeSql(sql, pars);
|
||
//Serilog.Log.Logger.Information($"time:{db.Ado.SqlExecutionTime.ToString()},sql:{strSql}");
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
// 处理异常，例如记录错误日志
|
||
Serilog.Log.Logger.Error(ex, "sql执行错误");
|
||
MessageApplication.SendErrorMessage("sql执行错误：" + ex?.ToString());
|
||
}
|
||
};
|
||
|
||
db.Aop.OnLogExecuting = (sql, pars) =>
|
||
{
|
||
// 获取原生SQL推荐 5.1.4.63 性能OK
|
||
// UtilMethods.GetNativeSql(sql, pars)
|
||
|
||
// 获取无参数化SQL 影响性能只适合调试
|
||
// UtilMethods.GetSqlString(DbType.SqlServer, sql, pars)
|
||
};
|
||
|
||
db.Aop.OnError = (exp) =>
|
||
{
|
||
try
|
||
{
|
||
var sql = exp.Sql;
|
||
MessageApplication.SendErrorMessage("sql执行错误：" + exp.ToString());
|
||
if (exp.Parametres != null)
|
||
{
|
||
foreach (var p in (exp.Parametres as SugarParameter[]))
|
||
{
|
||
switch (p.DbType)
|
||
{
|
||
case System.Data.DbType.String:
|
||
case System.Data.DbType.Date:
|
||
case System.Data.DbType.DateTime:
|
||
sql = sql.Replace(p.ParameterName, $"'{p.Value}'");
|
||
break;
|
||
default:
|
||
sql = sql.Replace(p.ParameterName, $"{p.Value}");
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
Serilog.Log.Logger.Error($"time:{exp?.ToString()}{db.Ado.SqlExecutionTime.ToString()},sql:{sql}");
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
// 处理异常，例如记录错误日志
|
||
Serilog.Log.Logger.Error($"time:{exp?.Message}");
|
||
}
|
||
};
|
||
|
||
db.Aop.OnExecutingChangeSql = (sql, pars) =>
|
||
{
|
||
#if DEBUG
|
||
try
|
||
{
|
||
var strSql = UtilMethods.GetNativeSql(sql, pars);
|
||
Serilog.Log.Logger.Debug($"time:{db.Ado.SqlExecutionTime.ToString()},sql:{strSql}");
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Serilog.Log.Logger.Error($"OnExecutingChangeSql:{ex.ToString()}");
|
||
// 处理异常，例如记录错误日志
|
||
}
|
||
#endif
|
||
return new KeyValuePair<string, SugarParameter[]>(sql, pars);
|
||
};
|
||
|
||
db.Aop.OnDiffLogEvent = it =>
|
||
{
|
||
// 操作前记录 包含： 字段描述 列名 值 表名 表描述
|
||
var editBeforeData = it.BeforeData;
|
||
// 操作后记录 包含： 字段描述 列名 值 表名 表描述
|
||
var editAfterData = it.AfterData;
|
||
var sql = it.Sql;
|
||
var parameter = it.Parameters;
|
||
var data = it.BusinessData; // 这边会显示你传进来的对象
|
||
var time = it.Time;
|
||
var diffType = it.DiffType; // enum insert 、update and delete
|
||
Serilog.Log.Logger.Information($"diffType:{diffType},editBeforeData:{editBeforeData},editAfterData:{editAfterData},sql:{sql},params:{JsonConvert.SerializeObject(parameter)}");
|
||
// Write logic
|
||
};
|
||
|
||
db.Aop.OnLogExecuted = (sql, p) =>
|
||
{
|
||
// 执行时间超过1秒
|
||
if (db.Ado.SqlExecutionTime.TotalSeconds > 2)
|
||
{
|
||
// 代码CS文件名
|
||
var fileName = db.Ado.SqlStackTrace.FirstFileName;
|
||
// 代码行数
|
||
var fileLine = db.Ado.SqlStackTrace.FirstLine;
|
||
// 方法名
|
||
var FirstMethodName = db.Ado.SqlStackTrace.FirstMethodName;
|
||
// db.Ado.SqlStackTrace.MyStackTraceList[1].xxx 获取上层方法的信息
|
||
|
||
LogContext.PushProperty("CostTime", db.Ado.SqlExecutionTime.TotalSeconds);
|
||
Serilog.Log.Logger.Warning($"SqlExecutionTime:{sql},params:{JsonConvert.SerializeObject(p)},fileName:{fileName},fileLine:{fileLine},MethodName:{FirstMethodName}");
|
||
}
|
||
// 相当于EF的 PrintToMiniProfiler
|
||
};
|
||
|
||
//db.Aop.DataExecuting = (oldValue, entityInfo) =>
|
||
//{
|
||
// /*** 列级别事件 ：更新的每一列都会进事件 ***/
|
||
// if (entityInfo.PropertyName == "UpdateTime" && entityInfo.OperationType == DataFilterType.UpdateByObject)
|
||
// {
|
||
// entityInfo.SetValue(DateTime.Now); // 修改UpdateTime字段
|
||
|
||
// /*当前列获取特性*/
|
||
// // 5.1.3.23 +
|
||
// // entityInfo.IsAnyAttribute<特性>()
|
||
// // entityInfo.GetAttribute<特性>()
|
||
// }
|
||
|
||
// /*** 行级别事件 ：更新一条记录只进一次 ***/
|
||
// if (entityInfo.EntityColumnInfo.IsPrimarykey)
|
||
// {
|
||
// // entityInfo.EntityValue 拿到单条实体对象
|
||
// }
|
||
|
||
// /*** 根据当前列修改另一列 ***/
|
||
// // if(当前列逻辑==XXX)
|
||
// // var properyDate = entityInfo.EntityValue.GetType().GetProperty("Date");
|
||
// // if(properyDate!=null)
|
||
// // properyDate.SetValue(entityInfo.EntityValue,1);
|
||
|
||
// // 可以写多个IF
|
||
|
||
// /*** 删除生效 （只有行级事件） ***/
|
||
// if (entityInfo.OperationType == DataFilterType.DeleteByObject)
|
||
// {
|
||
// // entityInfo.EntityValue 拿到所有实体对象
|
||
// }
|
||
//};
|
||
};
|
||
}
|
||
} |