From 87b391b8a81ee2abdaa4131d245784ecc7a54e9a Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期三, 17 六月 2020 19:19:40 +0800
Subject: [PATCH] 用户系统与卡密系统完善

---
 src/main/java/com/ks/tool/bkz/controller/ParseController.java |   49 +++++++++++++++++++++++++++++--------------------
 1 files changed, 29 insertions(+), 20 deletions(-)

diff --git a/src/main/java/com/ks/tool/bkz/controller/ParseController.java b/src/main/java/com/ks/tool/bkz/controller/ParseController.java
index bcecf8e..5713c29 100644
--- a/src/main/java/com/ks/tool/bkz/controller/ParseController.java
+++ b/src/main/java/com/ks/tool/bkz/controller/ParseController.java
@@ -3,11 +3,10 @@
 import com.alibaba.fastjson.JSONObject;
 import com.ks.tool.bkz.dto.FirstOrderSubParseResult;
 import com.ks.tool.bkz.entity.FirstOrderSubInfo;
+import com.ks.tool.bkz.exception.TBCookieException;
 import com.ks.tool.bkz.service.FirstOrderSubInfoService;
 import com.ks.tool.bkz.service.manager.RedisManager;
-import com.ks.tool.bkz.util.FirstOrderSubDataUtil;
-import com.ks.tool.bkz.util.JsonUtil;
-import com.ks.tool.bkz.util.UrlUtil;
+import com.ks.tool.bkz.util.*;
 import org.omg.CORBA.PUBLIC_MEMBER;
 import org.springframework.stereotype.Controller;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -24,20 +23,24 @@
 @Controller
 @RequestMapping("parse")
 public class ParseController {
-    String cookie = null;
     @Resource
     private RedisManager redisManager;
 
     @Resource
     private FirstOrderSubInfoService firstOrderSubInfoService;
 
+    private String getCookieKey(String tbUid) {
+        return "tbcookie-" + tbUid;
+    }
+
     @RequestMapping("getRequestUrl")
     @ResponseBody
-    public String getRequestUrl(String baseUrl) {
+    public String getRequestUrl(String baseUrl, String tbUid) {
         String host = UrlUtil.getHost(baseUrl);
         Map<String, String> params = UrlUtil.getParamsMap(baseUrl);
         params.put("t", System.currentTimeMillis() + "");
-        cookie = redisManager.get("cookie");
+
+        String cookie = redisManager.get(getCookieKey(tbUid));
         String token = FirstOrderSubDataUtil.getTokenFromCookie(cookie);
         String data = params.get("data");
         try {
@@ -78,9 +81,11 @@
     @RequestMapping("uploadCookies")
     @ResponseBody
     public String uploadCookie(String cookies) {
-        this.cookie = cookies;
-        redisManager.save("cookie", cookies, 60 * 60);
-        return cookies;
+        String tbUid = TBCookieUtil.getTBUid(cookies);
+        if (!StringUtil.isNullOrEmpty(tbUid)) {
+            redisManager.save(getCookieKey(tbUid), cookies, 60 * 60);
+        }
+        return "";
     }
 
     @RequestMapping("getLoginUrl")
@@ -100,17 +105,21 @@
     @RequestMapping("uploadContent")
     @ResponseBody
     public String uploadContent(String content) {
-        FirstOrderSubParseResult result = FirstOrderSubDataUtil.parseOrderSubData(content);
-        if (result != null && result.isHasNextPage()) {
-            if (result.getGoodsList() != null)
-                for (FirstOrderSubInfo info : result.getGoodsList()) {
-                    firstOrderSubInfoService.add(info);
-                    System.out.println(info.getId() + "-" + info.getTitle());
-                }
-            System.out.println("----------------------------");
-            return JsonUtil.loadTrueResult(null);
-        } else {
-            return JsonUtil.loadFalseResult(1, "鏃犳洿澶氭暟鎹�");
+        try {
+            FirstOrderSubParseResult result = FirstOrderSubDataUtil.parseOrderSubData(content);
+            if (result != null && result.isHasNextPage()) {
+                if (result.getGoodsList() != null)
+                    for (FirstOrderSubInfo info : result.getGoodsList()) {
+                        firstOrderSubInfoService.add(info);
+                        System.out.println(info.getId() + "-" + info.getTitle());
+                    }
+                System.out.println("----------------------------");
+                return JsonUtil.loadTrueResult(null);
+            } else {
+                return JsonUtil.loadFalseResult(1, "鏃犳洿澶氭暟鎹�");
+            }
+        } catch (TBCookieException e) {
+            return JsonUtil.loadFalseResult(1001, "娣樺疂鎺堟潈鍑洪敊");
         }
     }
 

--
Gitblit v1.8.0