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;
}
}
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;
}
}