springAOP切面获取入参和出参

最新推荐文章于 2024-09-24 18:03:04 发布

大panda

最新推荐文章于 2024-09-24 18:03:04 发布

阅读量9.7k

点赞数

分类专栏： spring+hibernate

本文链接：https://blog.csdn.net/i6725545/article/details/94397579

版权

spring+hibernate 专栏收录该内容

15 篇文章 1 订阅

订阅专栏

老版及参考

https://blog.csdn.net/weixin_33859504/article/details/86997374

新版****

package core.interceptor;

import com.alibaba.fastjson.JSON;
import lombok.extern.slf4j.Slf4j;
import neusoft.kqtrain.core.controller.common.BaseUriHead;
import org.apache.commons.lang.StringUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.lang.annotation.Annotation;
import java.lang.reflect.Method;

/**
 * 切面获取入参和出参
 */
@Component
@Aspect
@Slf4j
public class CustomAspect {

    @Pointcut("@within(org.springframework.stereotype.Controller) " +
            "|| @within(org.springframework.web.bind.annotation.RestController)")
    public void pointcut() {

    }

    @Around("pointcut()")
    public Object handle(ProceedingJoinPoint joinPoint) throws Throwable {
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        String uri = request.getRequestURI().replace(request.getContextPath(), "");
        Object result = joinPoint.proceed();
        if (uri.indexOf(BaseUriHead.API) != -1 || uri.indexOf(BaseUriHead.APPLETS) != -1) {
            String inPutParam = preHandle(joinPoint, request);
            String outPutParam = postHandle(result);
            String ip = getRemoteHost(request);
            log.info("remoteIP:{},uri:{},inPutParam:{},outPutParam:{}", ip, uri, inPutParam, outPutParam);
        }
        return result;
    }

    private String preHandle(ProceedingJoinPoint joinPoint, HttpServletRequest request) {
        Signature signature = joinPoint.getSignature();
        MethodSignature methodSignature = (MethodSignature) signature;
        Method targetMethod = methodSignature.getMethod();
        Annotation[] annotations = targetMethod.getAnnotations();
        StringBuffer sb = new StringBuffer();
        String token = request.getHeader("token");
        if (StringUtils.isNotBlank(token)) {
            sb.append("heander-token:" + token + ",");
        }
        for (Annotation annotation : annotations) {
            log.info("1-{}", annotation.annotationType().toString());
            if (annotation.annotationType().toString().indexOf("org.springframework.web.bind.annotation") == -1) {
                continue;
            }
            sb.append(JSON.toJSONString(request.getParameterMap()));
        }
        return sb.toString();
    }

    /**
     * 返回数据
     *
     * @param retVal
     * @return
     */
    private String postHandle(Object retVal) {
        if (null == retVal) {
            return "";
        }
        return JSON.toJSONString(retVal);
    }

    private String getRemoteHost(HttpServletRequest request) {
        String unknown = "unknown";
        String ip = request.getHeader("x-forwarded-for");
        if (StringUtils.isBlank(ip) || unknown.equalsIgnoreCase(ip)) {
            ip = request.getHeader("Proxy-Client-IP");
        }
        if (StringUtils.isBlank(ip) || unknown.equalsIgnoreCase(ip)) {
            ip = request.getHeader("WL-Proxy-Client-IP");
        }
        if (StringUtils.isBlank(ip) || unknown.equalsIgnoreCase(ip)) {
            ip = request.getRemoteAddr();
        }
        return "0:0:0:0:0:0:0:1".equals(ip) ? "127.0.0.1" : ip;
    }

}

``