94 lines
3.5 KiB
C#
94 lines
3.5 KiB
C#
using IPTools.Core;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.AspNetCore.Mvc.Controllers;
|
|
using Microsoft.AspNetCore.Mvc.Filters;
|
|
using Microsoft.Extensions.Logging;
|
|
using Yi.Framework.Infrastructure.AspNetCore;
|
|
using Yi.Framework.Infrastructure.CurrentUsers;
|
|
using Yi.Framework.Infrastructure.Ddd.Repositories;
|
|
using Yi.Framework.Infrastructure.Helper;
|
|
|
|
namespace Yi.Framework.Module.OperLogManager
|
|
{
|
|
public class GlobalOperLogAttribute : ActionFilterAttribute
|
|
{
|
|
private ILogger<GlobalOperLogAttribute> _logger;
|
|
private IRepository<OperationLogEntity> _repository;
|
|
private ICurrentUser _currentUser;
|
|
//注入一个日志服务
|
|
public GlobalOperLogAttribute(ILogger<GlobalOperLogAttribute> logger, IRepository<OperationLogEntity> repository, ICurrentUser currentUser)
|
|
{
|
|
_logger = logger;
|
|
_repository = repository;
|
|
_currentUser = currentUser;
|
|
}
|
|
|
|
public override async void OnResultExecuted(ResultExecutedContext context)
|
|
{
|
|
//判断标签是在方法上
|
|
if (context.ActionDescriptor is not ControllerActionDescriptor controllerActionDescriptor) return;
|
|
|
|
//查找标签，获取标签对象
|
|
OperLogAttribute? operLogAttribute = controllerActionDescriptor.MethodInfo.GetCustomAttributes(inherit: true)
|
|
.FirstOrDefault(a => a.GetType().Equals(typeof(OperLogAttribute))) as OperLogAttribute;
|
|
//空对象直接返回
|
|
if (operLogAttribute is null) return;
|
|
|
|
////获取控制器名
|
|
//string controller = context.RouteData.Values["Controller"].ToString();
|
|
|
|
////获取方法名
|
|
//string action = context.RouteData.Values["Action"].ToString();
|
|
|
|
//获取Ip
|
|
string ip = context.HttpContext.GetClientIp();
|
|
|
|
//根据ip获取地址
|
|
|
|
var ipTool = IpTool.Search(ip);
|
|
string location = ipTool.Province + " " + ipTool.City;
|
|
|
|
//日志服务插入一条操作记录即可
|
|
|
|
var logEntity = new OperationLogEntity();
|
|
logEntity.Id = SnowflakeHelper.NextId;
|
|
logEntity.OperIp = ip;
|
|
//logEntity.OperLocation = location;
|
|
logEntity.OperType = operLogAttribute.OperType;
|
|
logEntity.Title = operLogAttribute.Title;
|
|
logEntity.RequestMethod = context.HttpContext.Request.Method;
|
|
logEntity.Method = context.HttpContext.Request.Path.Value;
|
|
logEntity.OperLocation = location;
|
|
|
|
|
|
logEntity.OperUser = _currentUser.UserName;
|
|
if (operLogAttribute.IsSaveResponseData)
|
|
{
|
|
if (context.Result is ContentResult result && result.ContentType == "application/json")
|
|
{
|
|
logEntity.RequestResult = result.Content?.Replace("\r\n", "").Trim();
|
|
}
|
|
if (context.Result is JsonResult result2)
|
|
{
|
|
logEntity.RequestResult = result2.Value?.ToString();
|
|
}
|
|
|
|
if (context.Result is ObjectResult result3)
|
|
{
|
|
logEntity.RequestResult = JsonHelper.ObjToStr(result3.Value);
|
|
}
|
|
|
|
}
|
|
|
|
if (operLogAttribute.IsSaveRequestData)
|
|
{
|
|
//logEntity.RequestParam = context.HttpContext.GetRequestValue(logEntity.RequestMethod);
|
|
}
|
|
|
|
await _repository.InsertAsync(logEntity);
|
|
|
|
|
|
}
|
|
}
|
|
}
|