TemplatePro/BZPT.SqlSugarRepository/DBContext.cs

using Castle.Core.Logging;
using Microsoft.Extensions.Logging;
using NPlatform.Filters;
using NPlatform.Infrastructure.IdGenerators;
using NPOI.SS.Formula.Functions;
using SqlSugar;
using System.ComponentModel.DataAnnotations;
using System.ComponentModel.DataAnnotations.Schema;
using System.Configuration;
using System.Runtime.CompilerServices;

namespace BZPT.Repositories
{
    public partial class DBContext : SqlSugarClient
    {
        private readonly Microsoft.Extensions.Logging.ILogger<DBContext> _logger;
        private readonly string _connectionString;
        private readonly DbType _dbProvider;
        private readonly int _timeOut;

        public DBContext(string connectString, DbType dbProvider, Microsoft.Extensions.Logging.ILogger<DBContext> logger, 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;
                            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;
                            entity.DbTableName = attr?.Name;
                        }
                    }
                }
            }, dbAction)
        {
            _logger = logger;
            _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);
                    var logger = GetLogger(db);
                    logger?.LogTrace($"time:{db.Ado.SqlExecutionTime.ToString()},sql:{strSql}");
                }
                catch (Exception ex)
                {
                    // 处理异常，例如记录错误日志
                }
            };

            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 strSql = UtilMethods.GetNativeSql(exp.Sql, exp.Parametres as SugarParameter[]);
                    var logger = GetLogger(db);
                    logger?.LogDebug($"time:{db.Ado.SqlExecutionTime.ToString()},sql:{strSql}");
                }
                catch (Exception ex)
                {
                    // 处理异常，例如记录错误日志
                }
            };

            db.Aop.OnExecutingChangeSql = (sql, pars) =>
            {
#if DEBUG
                try
                {
                    var strSql = UtilMethods.GetNativeSql(sql, pars);
                    var logger = GetLogger(db);
                    logger?.LogDebug($"time:{db.Ado.SqlExecutionTime.ToString()},sql:{strSql}");
                }
                catch (Exception ex)
                {
                    // 处理异常，例如记录错误日志
                }
#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  

                // Write logic
            };

            db.Aop.OnLogExecuted = (sql, p) =>
            {
                // 执行时间超过1秒
                if (db.Ado.SqlExecutionTime.TotalSeconds > 1)
                {
                    // 代码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 获取上层方法的信息
                }
                // 相当于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 拿到所有实体对象
                }
            };
        };

        private static Microsoft.Extensions.Logging.ILogger<DBContext> GetLogger(SqlSugarClient db)
        {
            // 这里需要根据实际情况实现从 db 中获取 logger 的逻辑
            // 由于 SqlSugarClient 本身不直接提供获取 logger 的方法，可能需要其他方式传递 logger
            // 这里简单返回 null，实际使用时需要完善
            return null;
        }
    }
}