zoukankan      html  css  js  c++  java
  • 使用ActionFilterAttribute 记录 WebApi Action 请求和返回结果记录

    通过 重写 OnActionExecutingAsync,来 拦截action的请求消息,当执行OnActionExecutingAsync完成以后才真正进入请求的action中,action运行完后又把控制权给了 OnActionExecutedAsync ,这个管道机制可以使我们用它来轻松实现 权限认证、日志记录 ,跨域以及很多需要对全局或者部分请求做手脚的的功能。

    大概的流程如下:

    通过ActionFilterAttribute ,就能拦截action 处理的所有内容,包括请求提交的参数以及返回值。由于asp.net MVC 与webapi 是两个完全独立的管道:

    MVC由System.Web.Mvc.ActionFilterAttribute 来做action请求的拦截。
    webapi 由 System.Web.Http.Filters.ActionFilterAttribute 来处理。
    因此拦截action请求是完全不相干的两个通道,于此同时,当我们需要注册全局的ActionFilterAttribute 这两个也是分开注册的:

    MVC 直接在System.Web.Mvc.GlobalFilterCollection 这个全局管道里面注册 ActionFilter ,位置在App_Start目录>FilterConfig 类>RegisterGlobalFilters 方法 使用参数filters , filters.Add(new YourMvcAttribute()) 添加你的mvc ActionFilterAttribute 。

    wepi API 在System.Web.Http.Filters 中注册, 在项目的App_Start 目录>WebApiConfig类中>Register 方法中加入使用 config参数, config.Filters.Add(new YourWebApiAttribute()); 添加你的 webapi ActionFilterAttribute

    这样就可以注册你的 ActionFilterAttribute 成为全局的Filter,系统中请求经过Action 之前或之后 都会被你的ActionFilter 拦下来做处理然后在转交下去。

    好了道理已经讲完了,现在开始我自己要实现的 日志记录功能,

    需求是记录所有访问webapi action的(请求地址、内容、访问用户、提交的参数、返回的结果、以及一些客户端的信息)

    由于MVC 框架 提倡契约编程,在你自定义的Attribute 时,需要遵守契约规范, 【YourFilterName】+Attribute ,所以我的filter名字为 OperateTrackAttribute

    //-----------------------------------------------------------------------
    // <copyright file="OperateTrackAttribute.cs" company="FenSiShengHuo, Ltd.">
    //     Copyright (c) 2016 , All rights reserved.
    // </copyright>
    //-----------------------------------------------------------------------
    
    using System;
    using System.IO;
    using System.Linq;
    using System.Text;
    using System.Threading;
    using System.Threading.Tasks;
    using System.Web;
    using System.Web.Http.Controllers;
    using System.Web.Http.Filters;
    
    namespace DotNet.WeChat.API.Attribute
    {
        using DotNet.Business;
        using DotNet.Model;
        using DotNet.Utilities;
        using DotNet.WeChat.API.Models;
    
        /// <summary>
        /// OperateTrackAttribute
        /// 使用ActionFilterAttribute 记录 WebApi Action 请求和返回结果记录
        /// 
        /// 修改纪录
        /// 
        /// 2018-08-03 版本:1.0 JiShiYu 创建文件。
        /// 
        /// <author>
        ///     <name>JiShiYu</name>
        ///     <date>2018-08-03</date>
        /// </author>
        /// </summary>
    
        public class OperateTrackAttribute : ActionFilterAttribute
        {
            /// <summary>
            /// 自定义参数
            /// </summary>
            public string msg { get; set; }
    
            /// <summary>
            /// 无参构造函数
            /// </summary>
            public OperateTrackAttribute()
            {
    
            }
    
            /// <summary>
            /// 初始化时填入类的说明
            /// </summary>
            /// <param name="message"></param>
            public OperateTrackAttribute(string message)
            {
                msg = message;
            }
    
            /// <summary>
            ////// </summary>
            private static readonly string key = "enterTime";
    
            /// <summary>
            /// 通过重写 OnActionExecutingAsync,来拦截action的请求消息,
            /// 当执行OnActionExecutingAsync完成以后才真正进入请求的action中,action运行完后又把控制权给了 OnActionExecutedAsync ,
            /// 这个管道机制可以使我们用它来轻松实现 权限认证、日志记录 ,跨域以及很多需要对全局或者部分请求做手脚的的功能。
            /// </summary>
            /// <param name="actionContext"></param>
            /// <param name="cancellationToken"></param>
            /// <returns></returns>
            public override Task OnActionExecutingAsync(HttpActionContext actionContext, CancellationToken cancellationToken)
            {
                if (SkipLogging(actionContext))//是否该类标记为NoLog
                {
                    return base.OnActionExecutingAsync(actionContext, cancellationToken);
                }
    
                //记录进入请求的时间
                actionContext.Request.Properties[key] = DateTime.Now.ToBinary();
    
                return base.OnActionExecutingAsync(actionContext, cancellationToken);
            }
            /// <summary>
            /// 在请求执行完后 记录请求的数据以及返回数据
            /// </summary>
            /// <param name="actionExecutedContext"></param>
            /// <param name="cancellationToken"></param>
            /// <returns></returns>
            public override Task OnActionExecutedAsync(HttpActionExecutedContext actionExecutedContext, CancellationToken cancellationToken)
            {
                object beginTime = null;
                if (actionExecutedContext.Request.Properties.TryGetValue(key, out beginTime))
                {
                    DateTime time = DateTime.FromBinary(Convert.ToInt64(beginTime));
                    HttpRequest request = HttpContext.Current.Request;
                    string token = request.Headers["token"];
                    BaseUserInfo userInfo = GetUserByHeaders(actionExecutedContext);
                    WepApiActionLog apiActionLog = new WepApiActionLog
                    {
                        Id = Guid.NewGuid().ToString("N"),
                        // 获取action名称
                        ActionName = actionExecutedContext.ActionContext.ActionDescriptor.ActionName,
                        // 获取Controller 名称
                        ControllerName = actionExecutedContext.ActionContext.ActionDescriptor.ControllerDescriptor.ControllerName,
                        // 获取action开始执行的时间
                        EnterTime = time,
                        // 获取执行action的耗时
                        CostTime = (DateTime.Now - time).TotalMilliseconds,
                        UserAgent = request.UserAgent,
                        // 获取用户
                        UserId = userInfo.Id,
                        UserRealName = userInfo.RealName,
                        CompanyId = userInfo.CompanyId,
                        CompanyName = userInfo.CompanyName,
                        // 获取访问的ip
                        UserHostAddress = request.UserHostAddress,
                        UserHostName = request.UserHostName,
                        UrlReferrer = request.UrlReferrer != null ? request.UrlReferrer.AbsoluteUri : "",
                        Browser = request.Browser.Browser + " - " + request.Browser.Version + " - " + request.Browser.Type,
                        // 获取request提交的参数
                        Paramaters = GetRequestValues(actionExecutedContext),
                        // 获取response响应的结果
                        ExecuteResult = GetResponseValues(actionExecutedContext),
                        Comments = msg,
                        RequestUri = request.Url.AbsoluteUri
                    };
                    //using (TrackLogEntities context = new TrackLogEntities())
                    //{
                    //    context.WepApiActionLogs.Add(apiActionLog);
                    //    context.SaveChanges();
                    //}
    
                    NLogHelper.Debug("访问记录:"+ apiActionLog.FastToJson());
                }
    
                return base.OnActionExecutedAsync(actionExecutedContext, cancellationToken);
            }
    
            /// <summary>
            /// 获取当前登录用户
            /// </summary>
            /// <param name="actionExecutedContext"></param>
            /// <returns></returns>
            public static BaseUserInfo GetUserByHeaders(HttpActionExecutedContext actionExecutedContext)
            {
                BaseUserInfo userInfo = new BaseUserInfo();
                var re = actionExecutedContext.Request;
                var headers = re.Headers;
                if (headers.Contains("userInfo"))
                {
                    string userInfoStr = headers.GetValues("userInfoStr").First();
                    userInfo = BaseUserInfo.Deserialize(userInfoStr);
                }
    
                if (userInfo == null)
                {
                    string appKey = string.Empty;
                    if (headers.Contains("appKey"))
                    {
                        appKey = headers.GetValues("appKey").First();
                    }
                    var secret = BaseServicesLicenseManager.GetSecretByKeyByCache(appKey);
                    if (!string.IsNullOrWhiteSpace(secret))
                    {
                        BaseResult result = BaseServicesLicenseManager.CheckService(appKey, secret, true, false, 0, 0);
                        if (result.Status)
                        {
                            // 从接口确定当前调用者
                            BaseUserEntity userEntity = BaseUserManager.GetObjectByCodeByCache(appKey, true);
                            if (userEntity != null)
                            {
                                userInfo = new BaseUserInfo();
                                userInfo.Id = userEntity.Id;
                                userInfo.Code = userEntity.Code;
                                userInfo.UserName = userEntity.UserName;
                                userInfo.NickName = userEntity.NickName;
                                userInfo.RealName = userEntity.RealName;
                                userInfo.CompanyId = userEntity.CompanyId;
                                userInfo.CompanyCode = userEntity.CompanyCode;
                                userInfo.CompanyName = userEntity.CompanyName;
                            }
                        }
                    }
                }
    
                return userInfo;
            }
    
            /// <summary>
            /// 读取request 的提交内容
            /// </summary>
            /// <param name="actionExecutedContext"></param>
            /// <returns></returns>
            public string GetRequestValues(HttpActionExecutedContext actionExecutedContext)
            {
    
                Stream stream = actionExecutedContext.Request.Content.ReadAsStreamAsync().Result;
                Encoding encoding = Encoding.UTF8;
                /*
                    这个StreamReader不能关闭,也不能dispose, 关了就傻逼了
                    因为你关掉后,后面的管道  或拦截器就没办法读取了
                */
                var reader = new StreamReader(stream, encoding);
                string result = reader.ReadToEnd();
                /*
                这里也要注意:   stream.Position = 0;
                当你读取完之后必须把stream的位置设为开始
                因为request和response读取完以后Position到最后一个位置,交给下一个方法处理的时候就会读不到内容了。
                */
                stream.Position = 0;
                return result;
            }
    
            /// <summary>
            /// 读取action返回的result
            /// </summary>
            /// <param name="actionExecutedContext"></param>
            /// <returns></returns>
            public string GetResponseValues(HttpActionExecutedContext actionExecutedContext)
            {
                Stream stream = actionExecutedContext.Response.Content.ReadAsStreamAsync().Result;
                Encoding encoding = Encoding.UTF8;
                /*
                这个StreamReader不能关闭,也不能dispose, 关了就傻逼了
                因为你关掉后,后面的管道  或拦截器就没办法读取了
                */
                var reader = new StreamReader(stream, encoding);
                string result = reader.ReadToEnd();
                /*
                这里也要注意:   stream.Position = 0; 
                当你读取完之后必须把stream的位置设为开始
                因为request和response读取完以后Position到最后一个位置,交给下一个方法处理的时候就会读不到内容了。
                */
                stream.Position = 0;
                return result;
            }
    
            /// <summary>
            /// 判断类和方法头上的特性是否要进行Action拦截
            /// </summary>
            /// <param name="actionContext"></param>
            /// <returns></returns>
            private static bool SkipLogging(HttpActionContext actionContext)
            {
                return actionContext.ActionDescriptor.GetCustomAttributes<NoLogAttribute>().Any() || actionContext.ActionDescriptor.ControllerDescriptor.GetCustomAttributes<NoLogAttribute>().Any();
            }
        }
    }

    如果将webapi 的 OperateTrackAttribute 注册为webapi全局的 ActionFilter 那么我们如果有不想过滤的action 时,可以通过 检查 方法或类顶部特性 来对那些不需要接受拦击的 Controller 和action 顶部添加一个这样的特性来区分开,并通过在filter中检查是被拦截的action或controller 否包含此特性标记,不包含时拦截。

    下面是这个类的写法,一个空的类 继承Attribute,并在类顶部写出该Attribute 使用的范围: 

    //-----------------------------------------------------------------------
    // <copyright file="NoLogAttribute.cs" company="FenSiShengHuo, Ltd.">
    //     Copyright (c) 2016 , All rights reserved.
    // </copyright>
    //-----------------------------------------------------------------------
    
    using System;
    
    namespace DotNet.WeChat.API.Attribute
    {
        /// <summary>
        /// NoLogAttribute
        /// 个空的类 继承Attribute,并在类顶部写出该Attribute 使用的范围
        /// 这样我们的拦截就更灵活了,无论是添加了整个个Controller 的拦截还是全局拦截,只需要在不拦截的 controller 或action头部加上 [NoLog] 
        /// 
        /// 修改纪录
        /// 
        /// 2018-08-03 版本:1.0 JiShiYu 创建文件。
        /// 
        /// <author>
        ///     <name>JiShiYu</name>
        ///     <date>2018-08-03</date>
        /// </author>
        /// </summary>
        [AttributeUsage(AttributeTargets.Method | AttributeTargets.Class, Inherited = true)]
        public class NoLogAttribute : System.Attribute
        {
        }
    }
    

    这样我们的拦截就更灵活了,无论是添加了整个个Controller 的拦截还是全局拦截,只需要在不拦截的 controller 或action头部加上 [NoLog]

    例如

    /// <summary>
    /// 记录该类中的Action内容
    /// </summary>
    [OperateTrack]
    public class TestApiLogController : ApiController
    {
    [HttpPost]
    public object Login(UserInfo user)
    {
    var result = new { data = user, status = true };
    return result;
    }
    /// <summary>
    /// 该类不参与记录
    /// </summary>
    /// <param name="name"></param>
    /// <returns></returns>
    [NoLog]
    public string DontLogMe(string name)
    {
    return name;
    }
    }
    或者
    
    /// <summary>
    /// 该Controller 下的所有action 都不会被全局的OperateTrack Filter 拦截
    /// </summary>
    [NoLog]
    public class UserManagerController : ApiController
    {
    public List<string> GetUsers() {
    return new List<string>() { "tomers","jack"};
    }
    public string GiveUserSomeMoney(int money)
    {
    return money+"";
    }
    }

    我们来测试一下: 提交到/api/TestApiLog/Login 整个action 被标记为拦截

    再看看记录的结果,结果已经记录了

    这样整个记录用户访问记录的拦截器就到此为止了。

    根据原文地址:https://blog.csdn.net/lxrj2008/article/details/79278744 调整

  • 相关阅读:
    09.MyBatis的逆向工程
    08.MyBatis整合Log4j
    07.MyBatis中的关联查询
    06.动态SQL和foreach
    05.MyBtais两种取值符号以及输入参数和输出参数
    04.MyBatis别名的设置和类型转换器
    03.Mybatis优化
    02.Mybatis的动态代理方式实现增删改查
    01.MyBatis入门
    网络安全之网络安全(数通安全)0
  • 原文地址:https://www.cnblogs.com/hnsongbiao/p/9414422.html
Copyright © 2011-2022 走看看