parent
ea5c6fcb81
commit
b53fe0f7d6
@ -0,0 +1,41 @@
|
|||||||
|
package cn.teammodel.config;
|
||||||
|
|
||||||
|
import cn.teammodel.config.intercepter.EnhanceRequestServletFilter;
|
||||||
|
import cn.teammodel.config.intercepter.UploadApiLogInterceptor;
|
||||||
|
import lombok.RequiredArgsConstructor;
|
||||||
|
import org.jetbrains.annotations.NotNull;
|
||||||
|
import org.springframework.boot.web.servlet.FilterRegistrationBean;
|
||||||
|
import org.springframework.context.annotation.Bean;
|
||||||
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
import org.springframework.core.Ordered;
|
||||||
|
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
|
||||||
|
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* WebMvc配置 : 静态文件、拦截器等
|
||||||
|
* @author winter
|
||||||
|
* @create 2024-01-15 17:06
|
||||||
|
*/
|
||||||
|
@Configuration
|
||||||
|
@RequiredArgsConstructor
|
||||||
|
public class WebMvcConfig implements WebMvcConfigurer {
|
||||||
|
private final UploadApiLogInterceptor uploadApiLogInterceptor;
|
||||||
|
/**
|
||||||
|
* 拦截器配置
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public void addInterceptors(@NotNull InterceptorRegistry registry) {
|
||||||
|
registry.addInterceptor(uploadApiLogInterceptor);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 注册过滤器
|
||||||
|
@Bean
|
||||||
|
public FilterRegistrationBean<EnhanceRequestServletFilter> filterRegistrationBean(){
|
||||||
|
FilterRegistrationBean<EnhanceRequestServletFilter> filterRegistrationBean=new FilterRegistrationBean<>();
|
||||||
|
filterRegistrationBean.setFilter(new EnhanceRequestServletFilter());
|
||||||
|
filterRegistrationBean.addUrlPatterns("/*");
|
||||||
|
//order的数值越小 则优先级越高,这里直接使用的最高优先级
|
||||||
|
filterRegistrationBean.setOrder(Ordered.HIGHEST_PRECEDENCE);
|
||||||
|
return filterRegistrationBean;
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,134 @@
|
|||||||
|
package cn.teammodel.config.intercepter;
|
||||||
|
|
||||||
|
import cn.hutool.crypto.SecureUtil;
|
||||||
|
import cn.hutool.extra.servlet.ServletUtil;
|
||||||
|
import cn.hutool.jwt.JWTUtil;
|
||||||
|
import cn.teammodel.manager.NotificationService;
|
||||||
|
import com.fasterxml.jackson.databind.JsonNode;
|
||||||
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||||
|
import com.fasterxml.jackson.databind.node.ObjectNode;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import okhttp3.*;
|
||||||
|
import org.apache.commons.lang3.ObjectUtils;
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
import org.springframework.web.servlet.HandlerInterceptor;
|
||||||
|
|
||||||
|
import javax.annotation.Resource;
|
||||||
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
import javax.servlet.http.HttpServletResponse;
|
||||||
|
import java.io.BufferedReader;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStreamReader;
|
||||||
|
import java.nio.charset.StandardCharsets;
|
||||||
|
import java.time.Instant;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 上传接口调用日志拦截器
|
||||||
|
* @author winter
|
||||||
|
* @create 2024-03-18 10:00
|
||||||
|
*/
|
||||||
|
@Component
|
||||||
|
@Slf4j
|
||||||
|
public class UploadApiLogInterceptor implements HandlerInterceptor {
|
||||||
|
@Resource
|
||||||
|
private NotificationService notificationService;
|
||||||
|
|
||||||
|
private final ObjectMapper mapper = new ObjectMapper();
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
|
||||||
|
String ip = ServletUtil.getClientIP(request);
|
||||||
|
String client = "";
|
||||||
|
String tokenSha = "";
|
||||||
|
String id = "";
|
||||||
|
String name = "";
|
||||||
|
String school = "";
|
||||||
|
String scope = "";
|
||||||
|
String referer = request.getHeader("Referer");
|
||||||
|
Long requestTime = Instant.now().toEpochMilli();
|
||||||
|
// 获取 json 请求参数(这里读了后，后面就再也读不到了，需要处理)
|
||||||
|
BufferedReader streamReader = new BufferedReader(new InputStreamReader(request.getInputStream(), StandardCharsets.UTF_8));
|
||||||
|
StringBuilder responseStrBuilder = new StringBuilder();
|
||||||
|
String inputStr;
|
||||||
|
while ((inputStr = streamReader.readLine()) != null)
|
||||||
|
responseStrBuilder.append(inputStr);
|
||||||
|
|
||||||
|
JsonNode jsonNode = mapper.readTree(responseStrBuilder.toString());
|
||||||
|
String xAuthToken = request.getHeader("x-auth-AuthToken");
|
||||||
|
// 如果有 authorization
|
||||||
|
String authorization = request.getHeader("Authorization");
|
||||||
|
if (StringUtils.isNotBlank(authorization)) {
|
||||||
|
authorization = authorization.replace("Bearer ", "");
|
||||||
|
List<String> roles = (List<String>) JWTUtil.parseToken(authorization).getPayload("roles");
|
||||||
|
if (ObjectUtils.isNotEmpty(roles)) {
|
||||||
|
client = roles.get(0);
|
||||||
|
}
|
||||||
|
tokenSha = SecureUtil.sha1("Bearer " + authorization);
|
||||||
|
}
|
||||||
|
String xAuthIdToken = request.getHeader("X-Auth-IdToken");
|
||||||
|
if (StringUtils.isNotBlank(xAuthIdToken)) {
|
||||||
|
id = (String) JWTUtil.parseToken(xAuthIdToken).getPayload("sub");
|
||||||
|
name = (String) JWTUtil.parseToken(xAuthIdToken).getPayload("name");
|
||||||
|
if (StringUtils.isNotBlank(tokenSha)) {
|
||||||
|
tokenSha = SecureUtil.sha1(xAuthIdToken);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
String xAuthSchool = request.getHeader("X-Auth-School");
|
||||||
|
if (StringUtils.isNotBlank(xAuthSchool)) {
|
||||||
|
school = xAuthSchool;
|
||||||
|
}
|
||||||
|
if (StringUtils.isNotBlank(xAuthToken)) {
|
||||||
|
id = (String) JWTUtil.parseToken(xAuthToken).getPayload("sub");
|
||||||
|
name = (String) JWTUtil.parseToken(xAuthToken).getPayload("name");
|
||||||
|
school = (String) JWTUtil.parseToken(xAuthToken).getPayload("azp");
|
||||||
|
scope = (String) JWTUtil.parseToken(xAuthToken).getPayload("scope");
|
||||||
|
if (StringUtils.isNotBlank(tokenSha)) {
|
||||||
|
tokenSha = SecureUtil.sha1(xAuthToken);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ObjectNode root = mapper.createObjectNode();
|
||||||
|
root.put("ip", ip);
|
||||||
|
root.put("time", requestTime);
|
||||||
|
root.put("id", id);
|
||||||
|
root.put("name", name);
|
||||||
|
root.set("param", jsonNode);
|
||||||
|
root.put("school", school);
|
||||||
|
root.put("client", client);
|
||||||
|
root.put("tid", tokenSha);
|
||||||
|
root.put("scope", scope);
|
||||||
|
root.put("path", request.getContextPath() + request.getRequestURI());
|
||||||
|
root.put("host", request.getServerName());
|
||||||
|
root.put("p", "appraisal");
|
||||||
|
|
||||||
|
// 发送请求
|
||||||
|
OkHttpClient okHttpClient = new OkHttpClient();
|
||||||
|
String requestData = root.toString();
|
||||||
|
RequestBody requestBody = RequestBody.create(MediaType.parse("application/json"), requestData);
|
||||||
|
Request okRequest = new Request.Builder()
|
||||||
|
.url("https://teammodelosfunction.chinacloudsites.cn/api/http-log")
|
||||||
|
.post(requestBody)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
okHttpClient.newCall(okRequest).enqueue(new Callback() {
|
||||||
|
@Override
|
||||||
|
public void onFailure(Call call, IOException e) {
|
||||||
|
log.error("UploadApiLogIntercepter error" ) ;
|
||||||
|
notificationService.send("日志上传告警: 请求发送失败，检查请求发送客户端");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onResponse(Call call, Response response) throws IOException {
|
||||||
|
if (!response.isSuccessful()) {
|
||||||
|
log.error("UploadApiLogIntercepter error" ) ;
|
||||||
|
notificationService.send("日志上传告警: 请求响应异常，检查 API 源是否正常");
|
||||||
|
} else {
|
||||||
|
log.info("UploadApiLogIntercepter success");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue