using System.Diagnostics;
|
using System.Runtime.CompilerServices;
|
using KH.WMS.Core.Logging.LogEnums;
|
|
namespace KH.WMS.Core.Logging;
|
|
/// <summary>
|
/// 日志模块自动识别器
|
/// </summary>
|
public static class LogModuleDetector
|
{
|
/// <summary>
|
/// 模块命名空间映射配置
|
/// </summary>
|
private static readonly Dictionary<string, LogModule> NamespaceModuleMapping = new(StringComparer.OrdinalIgnoreCase)
|
{
|
// 基础设施层
|
["KH.WMS.Core.Api"] = LogModule.Api,
|
["KH.WMS.Core.Database"] = LogModule.Database,
|
["KH.WMS.Core.Cache"] = LogModule.Cache,
|
["KH.WMS.Core.Authentication"] = LogModule.Auth,
|
["KH.WMS.Core.Security"] = LogModule.Auth,
|
["KH.WMS.Core.Middleware"] = LogModule.Core,
|
["KH.WMS.Core.BackgroundServices"] = LogModule.Job,
|
["KH.WMS.Core.Logging"] = LogModule.Core,
|
|
// WMS 业务层 - 入库管理
|
["KH.WMS.Core.Inbound"] = LogModule.WMS_Inbound,
|
["KH.WMS.Core.Receiving"] = LogModule.WMS_Inbound,
|
["KH.WMS.Core.PutAway"] = LogModule.WMS_Inbound,
|
["KH.WMS.Core.Quality"] = LogModule.WMS_Inbound,
|
|
// WMS 业务层 - 出库管理
|
["KH.WMS.Core.Outbound"] = LogModule.WMS_Outbound,
|
["KH.WMS.Core.Picking"] = LogModule.WMS_Picking,
|
["KH.WMS.Core.Packing"] = LogModule.WMS_Outbound,
|
["KH.WMS.Core.Shipping"] = LogModule.WMS_Outbound,
|
["KH.WMS.Core.Loading"] = LogModule.WMS_Loading,
|
|
// WMS 业务层 - 库存管理
|
["KH.WMS.Core.Inventory"] = LogModule.WMS_Inventory,
|
["KH.WMS.Core.Stock"] = LogModule.WMS_Inventory,
|
["KH.WMS.Core.Stocktaking"] = LogModule.WMS_Counting,
|
["KH.WMS.Core.Counting"] = LogModule.WMS_Counting,
|
|
// WMS 业务层 - 基础数据
|
["WMKH.WMS.CoreS.Owner"] = LogModule.WMS_Owner,
|
["KH.WMS.Core.Product"] = LogModule.WMS_Product,
|
["KH.WMS.Core.Sku"] = LogModule.WMS_Product,
|
["KH.WMS.Core.Warehouse"] = LogModule.WMS_Warehouse,
|
["KH.WMS.Core.Zone"] = LogModule.WMS_Zone,
|
["KH.WMS.Core.Location"] = LogModule.WMS_Location,
|
|
// WMS 业务层 - 其他
|
["KH.WMS.Core.Transfer"] = LogModule.WMS_Transfer,
|
["KH.WMS.Core.Move"] = LogModule.WMS_Transfer,
|
["KH.WMS.Core.Wave"] = LogModule.WMS_Wave,
|
["KH.WMS.Core.Unloading"] = LogModule.WMS_Unloading,
|
["KH.WMS.Core.Processing"] = LogModule.WMS_Processing,
|
["KH.WMS.Core.Replenishment"] = LogModule.WMS_Replenishment,
|
["KH.WMS.Core.Strategy"] = LogModule.WMS_Strategy,
|
["KH.WMS.Core.Order"] = LogModule.WMS_Order,
|
["KH.WMS.Core.Transport"] = LogModule.WMS_Transport,
|
["KH.WMS.Core.Billing"] = LogModule.WMS_Billing,
|
["KH.WMS.Core.Report"] = LogModule.WMS_Report,
|
|
// TODO
|
// 通用业务层
|
["Application.Services"] = LogModule.Business,
|
["Domain.Services"] = LogModule.Business,
|
["Core.Services"] = LogModule.Business,
|
|
// API 层
|
["Controllers"] = LogModule.Api,
|
["Api"] = LogModule.Api,
|
["Controllers.Api"] = LogModule.Api,
|
|
// 外部服务
|
["External"] = LogModule.External,
|
["ThirdParty"] = LogModule.External,
|
["Integration"] = LogModule.External,
|
|
// 文件处理
|
["File"] = LogModule.File,
|
["Storage"] = LogModule.File,
|
["Upload"] = LogModule.File,
|
["Download"] = LogModule.File,
|
};
|
|
/// <summary>
|
/// 从调用栈中自动识别模块
|
/// </summary>
|
/// <param name="skipFrames">跳过的帧数（默认跳过 LoggerService 内部调用）</param>
|
/// <returns>识别出的模块，默认返回 Core</returns>
|
public static LogModule DetectModule(int skipFrames = 2)
|
{
|
try
|
{
|
var stackTrace = new StackTrace(skipFrames, false);
|
|
for (int i = 0; i < stackTrace.FrameCount; i++)
|
{
|
var frame = stackTrace.GetFrame(i);
|
if (frame == null) continue;
|
|
var namespaceName = frame.GetMethod()?.DeclaringType?.Namespace;
|
if (string.IsNullOrEmpty(namespaceName)) continue;
|
|
// 跳过日志系统本身的命名空间
|
if (namespaceName.StartsWith("KH.WMS.Core.Logging", StringComparison.OrdinalIgnoreCase) ||
|
namespaceName.StartsWith("Microsoft.Extensions.Logging", StringComparison.OrdinalIgnoreCase))
|
{
|
continue;
|
}
|
|
// 查找匹配的模块
|
var module = FindModuleByNamespace(namespaceName);
|
if (module != LogModule.Core)
|
{
|
return module;
|
}
|
}
|
|
return LogModule.Core;
|
}
|
catch
|
{
|
return LogModule.Core;
|
}
|
}
|
|
/// <summary>
|
/// 根据命名空间查找对应的模块
|
/// </summary>
|
private static LogModule FindModuleByNamespace(string namespaceName)
|
{
|
// 完全匹配
|
if (NamespaceModuleMapping.TryGetValue(namespaceName, out var module))
|
{
|
return module;
|
}
|
|
// 前缀匹配
|
foreach (var mapping in NamespaceModuleMapping)
|
{
|
if (namespaceName.StartsWith(mapping.Key, StringComparison.OrdinalIgnoreCase))
|
{
|
return mapping.Value;
|
}
|
}
|
|
// 默认返回业务模块
|
if (namespaceName.Contains("Service", StringComparison.OrdinalIgnoreCase) ||
|
namespaceName.Contains("Domain", StringComparison.OrdinalIgnoreCase) ||
|
namespaceName.Contains("Application", StringComparison.OrdinalIgnoreCase))
|
{
|
return LogModule.Business;
|
}
|
|
return LogModule.Core;
|
}
|
|
/// <summary>
|
/// 从类型推断模块
|
/// </summary>
|
public static LogModule DetectModuleFromType(Type type)
|
{
|
if (type == null) return LogModule.Core;
|
|
var namespaceName = type.Namespace;
|
return string.IsNullOrEmpty(namespaceName) ? LogModule.Core : FindModuleByNamespace(namespaceName);
|
}
|
|
/// <summary>
|
/// 添加自定义命名空间映射
|
/// </summary>
|
/// <param name="namespacePrefix">命名空间前缀</param>
|
/// <param name="module">对应的模块</param>
|
public static void AddMapping(string namespacePrefix, LogModule module)
|
{
|
NamespaceModuleMapping[namespacePrefix] = module;
|
}
|
|
/// <summary>
|
/// 批量添加命名空间映射
|
/// </summary>
|
public static void AddMappings(Dictionary<string, LogModule> mappings)
|
{
|
foreach (var mapping in mappings)
|
{
|
NamespaceModuleMapping[mapping.Key] = mapping.Value;
|
}
|
}
|
}
|
