From 267cb749bd684172451b56690b240e1227935c01 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@gmail.com>
Date: 星期三, 04 二月 2026 14:11:26 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/common/config/IntegrationRecordAdvice.java |   84 ++++++++++++++++++++++++++++++++++++++++++
 zy-acs-manager/src/main/java/com/zy/acs/manager/core/integrate/wms/OpenController.java     |    5 ++
 zy-acs-framework/src/main/java/com/zy/acs/framework/annotations/IntegrationAuth.java       |   21 ++++++++++
 3 files changed, 109 insertions(+), 1 deletions(-)

diff --git a/zy-acs-framework/src/main/java/com/zy/acs/framework/annotations/IntegrationAuth.java b/zy-acs-framework/src/main/java/com/zy/acs/framework/annotations/IntegrationAuth.java
new file mode 100644
index 0000000..f3e458f
--- /dev/null
+++ b/zy-acs-framework/src/main/java/com/zy/acs/framework/annotations/IntegrationAuth.java
@@ -0,0 +1,21 @@
+package com.zy.acs.framework.annotations;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Target({ElementType.TYPE,ElementType.METHOD})
+@Retention(RetentionPolicy.RUNTIME)
+public @interface IntegrationAuth {
+
+	Auth value() default Auth.CHECK;
+
+	String memo() default "";
+
+	public enum Auth{
+		CHECK,
+		NONE
+	}
+
+}
diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/common/config/IntegrationRecordAdvice.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/common/config/IntegrationRecordAdvice.java
new file mode 100644
index 0000000..6a243c4
--- /dev/null
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/common/config/IntegrationRecordAdvice.java
@@ -0,0 +1,84 @@
+package com.zy.acs.manager.common.config;
+
+import com.alibaba.fastjson.JSON;
+import com.zy.acs.framework.common.Cools;
+import com.zy.acs.framework.common.R;
+import com.zy.acs.manager.common.utils.IpTools;
+import com.zy.acs.manager.manager.service.IntegrationRecordService;
+import lombok.extern.slf4j.Slf4j;
+import org.jetbrains.annotations.NotNull;
+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;
+
+@Slf4j
+//@ControllerAdvice
+public class IntegrationRecordAdvice implements ResponseBodyAdvice<Object> {
+
+    @Autowired
+    private IntegrationRecordService integrationRecordService;
+
+    @Override
+    public boolean supports(@NotNull MethodParameter methodParameter, @NotNull Class<? extends HttpMessageConverter<?>> aClass) {
+        return true;
+    }
+
+    @Override
+    public Object beforeBodyWrite(Object o
+            , @NotNull MethodParameter methodParameter
+            , @NotNull MediaType mediaType
+            , @NotNull Class<? extends HttpMessageConverter<?>> aClass
+            , @NotNull ServerHttpRequest serverHttpRequest
+            , @NotNull ServerHttpResponse serverHttpResponse) {
+        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");
+                    Object reqCache = request.getAttribute("cache");
+                    if (!Cools.isEmpty(appkey)) {
+                        // 淇濆瓨鎺ュ彛鏃ュ織
+//                        apiLogService.save(
+//                                String.valueOf(appAuth),
+//                                request.getRequestURI(),
+//                                appkey,
+//                                IpTools.gainRealIp(request),
+//                                reqCache==null?"": JSON.toJSONString(reqCache),
+//                                JSON.toJSONString(o),
+//                                String.valueOf(((R) o).get("code")).equalsIgnoreCase("200")
+//                        );
+                    }
+                }
+            }
+        }
+        return o;
+    }
+
+    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) {
+            log.error(e.getMessage());
+            return "";
+        }
+    }
+
+}
diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/core/integrate/wms/OpenController.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/core/integrate/wms/OpenController.java
index b3bf360..770269d 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/core/integrate/wms/OpenController.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/core/integrate/wms/OpenController.java
@@ -23,6 +23,8 @@
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
+import javax.servlet.http.HttpServletRequest;
+
 /**
  * Created by vincent on 2023/6/12
  */
@@ -41,7 +43,7 @@
 
     @PostMapping("/bus/submit")
     @OperationLog("generate task from open api")
-    public R submit(@RequestBody OpenBusSubmitParam param) {
+    public R submit(@RequestBody OpenBusSubmitParam param, HttpServletRequest request) {
         IntegrationRecord integrationRecord = new IntegrationRecord(
                 null,    // 缂栧彿
                 null,    // 鍚嶇О绌洪棿
@@ -61,6 +63,7 @@
                 null,    // 淇敼鏃堕棿[闈炵┖]
                 null    // 澶囨敞
         );
+        request.setAttribute("integrationRecord", integrationRecord);
 
         mainService.generateBusAndTask(param, null);
         return R.ok("generate tasks success");

--
Gitblit v1.9.1