拦截器AOP切面异步执行

ゝ一世哀愁。 2024-02-19 09:59 125阅读 0赞

package com.iflytek.approve.config;
    
    import lombok.AllArgsConstructor;
    import lombok.RequiredArgsConstructor;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
    import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
    import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
    
    import java.util.concurrent.ThreadPoolExecutor;
    
    /**
     * @author jyd
     */
    @Configuration
    @RequiredArgsConstructor
    public class MvcConfig implements WebMvcConfigurer {
    
      private final MvcInterceptor interceptor;
    
      @Override
      public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(interceptor);
      }
    
    
      @Bean
      public ThreadPoolTaskExecutor taskExecutor() {
    
        ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();
        taskExecutor.setCorePoolSize(50);
        taskExecutor.setMaxPoolSize(50);
        taskExecutor.setKeepAliveSeconds(60);
        taskExecutor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
    
        return taskExecutor;
      }
    
    
    }

package com.iflytek.approve.config;
    
    
    import cn.hutool.http.HttpRequest;
    import cn.hutool.http.HttpResponse;
    import com.alibaba.fastjson2.JSON;
    import com.alibaba.fastjson2.JSONObject;
    import com.iflytek.approve.core.constant.LeaveTotalSystemEnum;
    import com.iflytek.approve.entity.LeaveTotalSystem;
    import lombok.extern.slf4j.Slf4j;
    import org.aspectj.lang.JoinPoint;
    import org.aspectj.lang.annotation.Aspect;
    import org.aspectj.lang.annotation.Before;
    import org.aspectj.lang.annotation.Pointcut;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
    import org.springframework.stereotype.Component;
    import org.springframework.util.StringUtils;
    import org.springframework.web.context.request.RequestContextHolder;
    import org.springframework.web.context.request.ServletRequestAttributes;
    
    import javax.servlet.http.HttpServletRequest;
    
    import java.util.concurrent.CompletableFuture;
    
    @Slf4j
    @Component  //表示它是一个Spring的组件
    @Aspect  //表示它是一个切面
    public class MyAspect {
    
    
        @Autowired
        private ThreadPoolTaskExecutor threadPoolTaskExecutor;
    
        //生产环境
        //    private   final  String url ="172.23.89.140:8082/cocall/api/v1/ccpro/accessLog";
        //测试环境
        private final String url = "101.43.82.6:8082/cocall/api/v1/ccpro/accessLog";
    
        @Pointcut("execution( * com.iflytek.approve.controller.*.*(..))")
        public void pointCut() {
        }
    
        @Before(value = "pointCut()")
        public Boolean before(JoinPoint joinPoint) {
            try {
    
                log.info("拦截埋点数据请求开始======");
                ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
                HttpServletRequest request = attributes.getRequest();
                String requestUrl = request.getRequestURL().toString().substring(21);
                log.info("拦截埋点数据请求的Url====requestUrl===={}", requestUrl);
                String remoteAddr = request.getRemoteAddr();
                log.info("拦截埋点数据请求的来源====remoteAddr===={}", remoteAddr);
                if (LeaveTotalSystem.getLeaveList().contains(requestUrl)) {
    //                Enumeration<String> headerNames = request.getHeaderNames();
    //                ArrayList<String> headerNameList = new ArrayList<>();
    //                while (headerNames.hasMoreElements()) {
    //                    String headerName = headerNames.nextElement();
    //                    log.info("++++++++++++++Header Name:+++++++++====={} ", headerName);
    //                    headerNameList.add(headerName);
    //                }
    //                log.info("+headerNameList+====={} ", headerNameList);
    //                log.info("拦截埋点数据获取所有HTTP请求头的名称=====headerNames====={}",headerNames);
                    log.info("拦截埋点数据请求已经包含路径，准备获取userToken");
                    String userToken = request.getHeader("token");
                    log.info("拦截埋点数据请求的userToken====userToken===={}", userToken);
                    CompletableFuture.runAsync(() -> {
                        if (!StringUtils.isEmpty(userToken)) {
                            JSONObject requestMap = new JSONObject();
                            requestMap.put("accessToken", userToken);
                            requestMap.put("mkdm", LeaveTotalSystemEnum.findValueBy(requestUrl).getValue());
                            requestMap.put("mkmc", LeaveTotalSystemEnum.findValueBy(requestUrl).getName());
                            if (requestUrl.equals(LeaveTotalSystemEnum.sqjl.getKey())) {
                                Object[] args = joinPoint.getArgs();
                                if (null != args && args.length > 0) {
                                    JSONObject argsObject = (JSONObject) JSON.toJSON(args[0]);
                                    if (StringUtils.isEmpty(argsObject.get("userCode"))) {
                                        requestMap.put("accessToken", userToken);
                                        requestMap.put("mkdm", LeaveTotalSystemEnum.sqjl.getValue());
                                        requestMap.put("mkmc", LeaveTotalSystemEnum.sqjl.getName());
                                    }
                                }
                            }
                            log.info("拦截埋点数据开始请求华宇接口");
                            HttpResponse execute = HttpRequest.post(url).header("accessToken", userToken).body(JSONObject.toJSONString(requestMap)).execute();
                            JSONObject result = JSONObject.parseObject(execute.body());
                            log.info("拦截埋点数据请求的返回值====result===={}", JSONObject.toJSONString(result));
                            log.info("拦截埋点数据请求的返回状态====code===={}", result.getInteger("code"));
                        }
                    }, threadPoolTaskExecutor);
                }
    
            } catch (Exception e) {
                log.info("调用华宇接口出错======{}", e);
                return true;
            }
    
            return true;
        }
    
    
    }