package com.zy.common.config;
|
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.mapper.EntityWrapper;
|
import com.core.common.Cools;
|
import com.core.common.R;
|
import com.core.common.SpringUtils;
|
import com.zy.asrs.entity.ApiLog;
|
import com.zy.asrs.service.ApiLogService;
|
import com.zy.common.utils.IpTools;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.core.MethodParameter;
|
import org.springframework.http.MediaType;
|
import org.springframework.http.converter.HttpMessageConverter;
|
import org.springframework.http.server.ServerHttpRequest;
|
import org.springframework.http.server.ServerHttpResponse;
|
import org.springframework.http.server.ServletServerHttpRequest;
|
import org.springframework.web.bind.annotation.ControllerAdvice;
|
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.io.BufferedReader;
|
import java.io.InputStreamReader;
|
import java.nio.charset.StandardCharsets;
|
import java.util.Date;
|
|
/**
|
* 返回值处理
|
* Created by vincent on 2022/4/22
|
*/
|
@Slf4j
|
@ControllerAdvice
|
public class ControllerResAdvice implements ResponseBodyAdvice<Object> {
|
|
@Autowired
|
private ApiLogService apiLogService;
|
|
@Override
|
public boolean supports(MethodParameter methodParameter, Class<? extends HttpMessageConverter<?>> aClass) {
|
return true;
|
}
|
|
@Override
|
public Object beforeBodyWrite(Object o, MethodParameter methodParameter, MediaType mediaType, Class<? extends HttpMessageConverter<?>> aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
|
if (serverHttpRequest instanceof ServletServerHttpRequest) {
|
HttpServletRequest request = ((ServletServerHttpRequest) serverHttpRequest).getServletRequest();
|
Object appAuth = request.getAttribute("appAuth");
|
if (appAuth != null && o instanceof R) {
|
String appkey = request.getHeader("appkey");
|
if (Cools.isEmpty(appkey)) {
|
appkey = "-";
|
}
|
Object reqCache = request.getAttribute("cache");
|
boolean success = String.valueOf(((R) o).get("code")).equalsIgnoreCase("200");
|
String uri = request.getRequestURI();
|
if (!skipOpenAsrsHighFrequencyApiLog(uri)) {
|
if (success) {
|
apiLogService.save(
|
String.valueOf(appAuth),
|
uri,
|
appkey,
|
IpTools.gainRealIp(request),
|
reqCache == null ? "" : JSON.toJSONString(reqCache),
|
JSON.toJSONString(o),
|
success
|
);
|
} else {
|
beforeBodyWriteCallApiLogSave(
|
String.valueOf(appAuth),
|
uri,
|
appkey,
|
IpTools.gainRealIp(request),
|
reqCache == null ? "" : JSON.toJSONString(reqCache),
|
JSON.toJSONString(o),
|
success);
|
}
|
}
|
} else if (o instanceof R && isInboundThirdPartyUri(request.getRequestURI())) {
|
String appkey = request.getHeader("appkey");
|
if (Cools.isEmpty(appkey)) {
|
appkey = "-";
|
}
|
Object reqCache = request.getAttribute("cache");
|
boolean success = String.valueOf(((R) o).get("code")).equalsIgnoreCase("200");
|
String uri = request.getRequestURI();
|
String ns = inboundNamespace(uri);
|
if (!skipOpenAsrsHighFrequencyApiLog(uri)) {
|
if (success) {
|
apiLogService.save(
|
ns,
|
uri,
|
appkey,
|
IpTools.gainRealIp(request),
|
reqCache == null ? "" : JSON.toJSONString(reqCache),
|
JSON.toJSONString(o),
|
success
|
);
|
} else {
|
beforeBodyWriteCallApiLogSave(
|
ns,
|
uri,
|
appkey,
|
IpTools.gainRealIp(request),
|
reqCache == null ? "" : JSON.toJSONString(reqCache),
|
JSON.toJSONString(o),
|
success);
|
}
|
}
|
}
|
}
|
return o;
|
}
|
|
/** 电视机/看板等高频查询，不落 api_log（context-path 前缀任意，只匹配路径后缀） */
|
private static final String[] OPEN_ASRS_SKIP_LOG_URI_SUFFIXES = {
|
"/open/asrs/queryTask",
|
"/open/asrs/queryLoc",
|
"/open/asrs/line/charts/hourly",
|
"/open/asrs/line/charts",
|
"/open/asrs/locDetl/statistics",
|
};
|
|
private static boolean skipOpenAsrsHighFrequencyApiLog(String uri) {
|
if (uri == null) {
|
return false;
|
}
|
int cut = uri.indexOf(';');
|
String path = cut > 0 ? uri.substring(0, cut) : uri;
|
for (String suffix : OPEN_ASRS_SKIP_LOG_URI_SUFFIXES) {
|
if (path.endsWith(suffix)) {
|
return true;
|
}
|
}
|
return false;
|
}
|
|
private static boolean isInboundThirdPartyUri(String uri) {
|
if (uri == null) {
|
return false;
|
}
|
return uri.contains("/open/asrs")
|
|| uri.contains("/wcs/openapi/report")
|
|| uri.contains("/wcs/openapi/reassign/loc")
|
|| uri.contains("/rpc/pakin/loc");
|
}
|
|
private static String inboundNamespace(String uri) {
|
if (uri != null && uri.contains("/wcs/openapi/report")) {
|
return "WCS回写";
|
}
|
if (uri != null && uri.contains("/wcs/openapi/reassign/loc")) {
|
return "WCS重分配入库";
|
}
|
if (uri != null && uri.contains("/rpc/pakin/loc")) {
|
return "WCS入库";
|
}
|
return "开放接口";
|
}
|
|
public void beforeBodyWriteCallApiLogSave(String name, String url, String appkey, String ip, String request, String response, boolean success) {
|
if (skipOpenAsrsHighFrequencyApiLog(url)) {
|
return;
|
}
|
ApiLogService apiLogService = SpringUtils.getBean(ApiLogService.class);
|
String memo = response;
|
try{
|
JSONObject reqJson = JSONObject.parseObject(request);
|
memo = reqJson.getString("staNo"); // 提取 staNo
|
} catch (Exception e){
|
|
}
|
ApiLog apiLog = apiLogService.selectOne(new EntityWrapper<ApiLog>()
|
.eq("namespace", name)
|
.eq("memo", memo)
|
.eq("result", success? 1:0)
|
.orderBy("create_time", false)
|
);
|
|
if (!Cools.isEmpty(apiLog)){
|
long parseLong = Long.parseLong(apiLog.getTimestamp());
|
if (new Date().getTime()-parseLong<5*1000*60){
|
return;
|
}
|
}
|
// 保存接口日志
|
apiLogService.save(
|
name,
|
url,
|
appkey,
|
ip,
|
request,
|
response,
|
success,
|
memo
|
);
|
}
|
|
public static String json(HttpServletRequest request) {
|
try {
|
BufferedReader streamReader = new BufferedReader( new InputStreamReader(request.getInputStream(), StandardCharsets.UTF_8));
|
StringBuilder sb = new StringBuilder();
|
String inputStr;
|
while ((inputStr = streamReader.readLine()) != null) {
|
sb.append(inputStr);
|
}
|
return sb.toString();
|
} catch (Exception e) {
|
e.printStackTrace();
|
return "";
|
}
|
}
|
|
}
|
