From 18c51d40be82435289ba3be6bd5f8e15fdf786f7 Mon Sep 17 00:00:00 2001
From: cl <1442464845@qq.com>
Date: 星期六, 04 四月 2026 00:26:56 +0800
Subject: [PATCH] mqtt
---
src/main/java/com/zy/common/config/ControllerResAdvice.java | 141 ++++++++++++++++++++++++++++++++++++++++++-----
1 files changed, 126 insertions(+), 15 deletions(-)
diff --git a/src/main/java/com/zy/common/config/ControllerResAdvice.java b/src/main/java/com/zy/common/config/ControllerResAdvice.java
index 08b22d1..ae7410a 100644
--- a/src/main/java/com/zy/common/config/ControllerResAdvice.java
+++ b/src/main/java/com/zy/common/config/ControllerResAdvice.java
@@ -1,8 +1,12 @@
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;
@@ -17,6 +21,10 @@
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;
/**
* 杩斿洖鍊煎鐞�
@@ -39,25 +47,128 @@
if (serverHttpRequest instanceof ServletServerHttpRequest) {
HttpServletRequest request = ((ServletServerHttpRequest) serverHttpRequest).getServletRequest();
Object appAuth = request.getAttribute("appAuth");
- if (appAuth != null) {
- if (o instanceof R) {
- String appkey = request.getHeader("appkey");
- if (!Cools.isEmpty(appkey)) {
- // 淇濆瓨鎺ュ彛鏃ュ織
- apiLogService.save(
- String.valueOf(appAuth),
- request.getRequestURI(),
- appkey,
- IpTools.gainRealIp(request),
- JSON.toJSONString(request.getParameterMap()),
- JSON.toJSONString(o),
- Boolean.TRUE
- );
- }
+ 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");
+ if (success) {
+ apiLogService.save(
+ String.valueOf(appAuth),
+ request.getRequestURI(),
+ appkey,
+ IpTools.gainRealIp(request),
+ reqCache == null ? "" : JSON.toJSONString(reqCache),
+ JSON.toJSONString(o),
+ success
+ );
+ } else {
+ beforeBodyWriteCallApiLogSave(
+ String.valueOf(appAuth),
+ request.getRequestURI(),
+ 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 ns = inboundNamespace(request.getRequestURI());
+ if (success) {
+ apiLogService.save(
+ ns,
+ request.getRequestURI(),
+ appkey,
+ IpTools.gainRealIp(request),
+ reqCache == null ? "" : JSON.toJSONString(reqCache),
+ JSON.toJSONString(o),
+ success
+ );
+ } else {
+ beforeBodyWriteCallApiLogSave(
+ ns,
+ request.getRequestURI(),
+ appkey,
+ IpTools.gainRealIp(request),
+ reqCache == null ? "" : JSON.toJSONString(reqCache),
+ JSON.toJSONString(o),
+ success);
}
}
}
return o;
}
+ private static boolean isInboundThirdPartyUri(String uri) {
+ if (uri == null) {
+ return false;
+ }
+ return uri.contains("/open/asrs") || uri.contains("/wcs/openapi/report");
+ }
+
+ private static String inboundNamespace(String uri) {
+ if (uri != null && uri.contains("/wcs/openapi/report")) {
+ return "WCS鍥炲啓";
+ }
+ return "寮�鏀炬帴鍙�";
+ }
+
+ public void beforeBodyWriteCallApiLogSave(String name, String url, String appkey, String ip, String request, String response, boolean success) {
+ 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 "";
+ }
+ }
+
}
--
Gitblit v1.9.1