From 931012792afd7d03b38b4c87b7bdb6235ba86943 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期六, 23 四月 2022 20:59:17 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/asrs/controller/OpenController.java    |   34 ++++++++++------
 src/main/java/com/zy/common/config/ControllerResAdvice.java |   31 ++++++++-------
 2 files changed, 38 insertions(+), 27 deletions(-)

diff --git a/src/main/java/com/zy/asrs/controller/OpenController.java b/src/main/java/com/zy/asrs/controller/OpenController.java
index bc4c409..6faf5f2 100644
--- a/src/main/java/com/zy/asrs/controller/OpenController.java
+++ b/src/main/java/com/zy/asrs/controller/OpenController.java
@@ -15,6 +15,7 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
@@ -45,8 +46,9 @@
     @PostMapping("/order/pakin/new/package/v1")
     @AppAuth(memo = "鎵撳寘涓婄嚎鎺ュ彛")
     public synchronized R pakinOrderPackage(@RequestHeader(required = false) String appkey,
-                                            @RequestBody(required = false) PackParam param) {
-        auth(appkey, param);
+                                            @RequestBody(required = false) PackParam param,
+                                            HttpServletRequest request) {
+        auth(appkey, param, request);
         if (Cools.isEmpty(param)) {
             return R.parse(BaseRes.PARAM);
         }
@@ -61,8 +63,9 @@
      */
     @PostMapping("/order/pakin/new/default/v1")
     public synchronized R pakinOrderCreate(@RequestHeader(required = false) String appkey,
-                                           @RequestBody OpenOrderPakinParam param) {
-        auth(appkey, param);
+                                           @RequestBody OpenOrderPakinParam param,
+                                           HttpServletRequest request) {
+        auth(appkey, param, request);
         if (Cools.isEmpty(param)) {
             return R.parse(BaseRes.PARAM);
         }
@@ -84,8 +87,9 @@
      */
     @PostMapping("/order/pakin/complete/default/v1")
     public synchronized R orderPakinComplete(@RequestHeader(required = false) String appkey,
-                                        @RequestBody(required = false) OpenOrderCompleteParam param) {
-        auth(appkey, param);
+                                             @RequestBody(required = false) OpenOrderCompleteParam param,
+                                             HttpServletRequest request) {
+        auth(appkey, param, request);
         return R.ok().add(openService.pakinOrderComplete(param));
     }
 
@@ -95,8 +99,9 @@
     @PostMapping("/order/pakout/new/default/v1")
     @AppAuth(memo = "娣诲姞璁㈠崟鍑哄簱")
     public synchronized R pakoutOrderCreate(@RequestHeader(required = false) String appkey,
-                                           @RequestBody OpenOrderPakoutParam param) {
-        auth(appkey, param);
+                                            @RequestBody OpenOrderPakoutParam param,
+                                            HttpServletRequest request) {
+        auth(appkey, param, request);
         if (Cools.isEmpty(param)) {
             return R.parse(BaseRes.PARAM);
         }
@@ -118,8 +123,9 @@
      */
     @PostMapping("/order/pakout/complete/default/v1")
     public synchronized R orderPakoutComplete(@RequestHeader(required = false) String appkey,
-                                        @RequestBody(required = false) OpenOrderCompleteParam param) {
-        auth(appkey, param);
+                                              @RequestBody(required = false) OpenOrderCompleteParam param,
+                                              HttpServletRequest request) {
+        auth(appkey, param, request);
         return R.ok().add(openService.pakoutOrderComplete(param));
     }
 
@@ -128,14 +134,16 @@
      * 搴撳瓨缁熻
      */
     @RequestMapping("/stock/default/v1")
-    public R queryStock(@RequestHeader(required = false) String appkey) {
-        auth(appkey, null);
+    public R queryStock(@RequestHeader(required = false) String appkey,
+                        HttpServletRequest request) {
+        auth(appkey, null, request);
         return R.ok().add(openService.queryStock());
     }
 
 
-    private void auth(String appkey, Object obj) {
+    private void auth(String appkey, Object obj, HttpServletRequest request) {
         log.info("{}鎺ュ彛琚闂紱appkey:{}锛涜姹傛暟鎹細{}", "open/sensorType/list/auth/v1", appkey, JSON.toJSONString(obj));
+        request.setAttribute("cache", obj);
         if (!auth) {
             return;
         }
diff --git a/src/main/java/com/zy/common/config/ControllerResAdvice.java b/src/main/java/com/zy/common/config/ControllerResAdvice.java
index 3cd1fa6..a4cdc97 100644
--- a/src/main/java/com/zy/common/config/ControllerResAdvice.java
+++ b/src/main/java/com/zy/common/config/ControllerResAdvice.java
@@ -17,6 +17,9 @@
 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;
 
 /**
  * 杩斿洖鍊煎鐞�
@@ -42,6 +45,7 @@
             if (appAuth != null) {
                 if (o instanceof R) {
                     String appkey = request.getHeader("appkey");
+                    Object reqCache = request.getAttribute("cache");
                     if (!Cools.isEmpty(appkey)) {
                         // 淇濆瓨鎺ュ彛鏃ュ織
                         apiLogService.save(
@@ -49,7 +53,7 @@
                                 request.getRequestURI(),
                                 appkey,
                                 IpTools.gainRealIp(request),
-                                json(request),
+                                reqCache==null?"": JSON.toJSONString(reqCache),
                                 JSON.toJSONString(o),
                                 String.valueOf(((R) o).get("code")).equalsIgnoreCase("200")
                         );
@@ -61,19 +65,18 @@
     }
 
     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 "";
-//        }
-        return "";
+        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