using System.Reflection;
|
using Castle.DynamicProxy;
|
using KH.WMS.Core.Attributes;
|
using KH.WMS.Core.Logging;
|
using Microsoft.Extensions.Logging;
|
|
namespace KH.WMS.Core.AOP;
|
|
/// <summary>
|
/// 拦截器基类
|
/// </summary>
|
public class LoggingInterceptor : IInterceptor
|
{
|
private readonly ILoggerService _logger;
|
|
|
public LoggingInterceptor(ILoggerService logger)
|
{
|
_logger = logger;
|
}
|
|
public void Intercept(IInvocation invocation)
|
{
|
string methodName = $"{invocation.TargetType.Name}.{invocation.Method.Name}";
|
MethodInfo method = invocation.Method;
|
|
try
|
{
|
// 从方法上获取特性
|
LogInterceptorAttribute? _attribute = invocation.MethodInvocationTarget.GetCustomAttributes(typeof(LogInterceptorAttribute), false)
|
.FirstOrDefault() as LogInterceptorAttribute;
|
|
// 如果方法上没有特性，尝试从类上获取
|
if (_attribute == null)
|
{
|
_attribute = invocation.TargetType.GetCustomAttributes(typeof(LogInterceptorAttribute), false)
|
.FirstOrDefault() as LogInterceptorAttribute;
|
}
|
if (_attribute == null)
|
return;
|
|
// 如果没有特性，使用默认值
|
var logLevel = _attribute?.LogLevel ?? LogLevel.Information;
|
|
_logger.LogInfo("[方法开始] -> {MethodName}",
|
methodName);
|
// 执行方法
|
invocation.Proceed();
|
|
if (_attribute == null)
|
return;
|
|
var message = $"[方法结束] <- {methodName}";
|
|
_logger.LogInfo(message);
|
}
|
catch (Exception ex)
|
{
|
// 异常处理
|
OnException(invocation, methodName, ex);
|
}
|
finally
|
{
|
// 清理逻辑
|
OnFinally(invocation, methodName);
|
}
|
|
}
|
|
/// <summary>
|
/// 异常处理
|
/// </summary>
|
protected void OnException(IInvocation invocation, string methodName, Exception ex)
|
{
|
}
|
|
/// <summary>
|
/// 清理逻辑
|
/// </summary>
|
protected void OnFinally(IInvocation invocation, string methodName)
|
{
|
}
|
}
|
