From 6ba8a320809159d4e1fbcf97ee578b63163f287d Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@63.com>
Date: 星期六, 09 四月 2022 14:10:26 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/asrs/controller/OpenController.java |   29 +++++++++++++++++++++++------
 1 files changed, 23 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/zy/asrs/controller/OpenController.java b/src/main/java/com/zy/asrs/controller/OpenController.java
index 925fa41..99e4332 100644
--- a/src/main/java/com/zy/asrs/controller/OpenController.java
+++ b/src/main/java/com/zy/asrs/controller/OpenController.java
@@ -1,5 +1,6 @@
 package com.zy.asrs.controller;
 
+import com.alibaba.fastjson.JSON;
 import com.core.common.*;
 import com.core.exception.CoolException;
 import com.zy.asrs.entity.DocType;
@@ -9,12 +10,10 @@
 import com.zy.asrs.entity.param.OpenOrderPakinParam;
 import com.zy.asrs.service.*;
 import com.zy.common.model.DetlDto;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.transaction.annotation.Transactional;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
 import java.util.ArrayList;
 import java.util.Date;
@@ -23,9 +22,14 @@
 /**
  * Created by vincent on 2022/4/8
  */
+@Slf4j
 @RestController
 @RequestMapping("open/asrs")
 public class OpenController {
+
+    public static final ArrayList<String> APP_KEY_LIST = new ArrayList<String>() {{
+        add("ea1f0459efc02a79f046f982767939ae");
+    }};
 
     @Autowired
     private OrderService orderService;
@@ -40,12 +44,24 @@
     @Autowired
     private LocDetlService locDetlService;
 
+    private void auth(String appkey, Object obj) {
+        log.info("{}鎺ュ彛琚闂紒appkey:{},璇锋眰鏁版嵁锛歿}", "open/sensorType/list/auth/v1", appkey, JSON.toJSONString(obj));
+        if (Cools.isEmpty(appkey)) {
+            throw new CoolException("璁よ瘉澶辫触锛岃纭appkey鏃犺锛�");
+        }
+        if (!APP_KEY_LIST.contains(appkey)) {
+            throw new CoolException("璁よ瘉澶辫触锛岃纭appkey鏃犺锛�");
+        }
+    }
+
     /**
      * 娣诲姞鍏ュ簱鍗�
      */
     @PostMapping("/order/pakin/default/v1")
     @Transactional
-    public synchronized R orderCreate(@RequestBody OpenOrderPakinParam param) {
+    public synchronized R orderCreate(@RequestHeader(required = false) String appkey,
+                                      @RequestBody OpenOrderPakinParam param) {
+        auth(appkey, param);
         if (Cools.isEmpty(param)) {
             return R.parse(BaseRes.PARAM);
         }
@@ -144,7 +160,8 @@
      * 搴撳瓨缁熻
      */
     @RequestMapping("/stock/default/v1")
-    public R queryStock() {
+    public R queryStock(@RequestHeader(required = false) String appkey) {
+        auth(appkey, null);
         return R.ok(locDetlService.queryStockTotal());
     }
 

--
Gitblit v1.9.1