From 7c31b75ddb67608b0a86ce2317608f275d44a872 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 14 May 2020 10:56:02 +0800
Subject: [PATCH] modify swagger some config

---
 src/main/java/com/xcong/excoin/common/system/controller/LoginController.java |   33 +++++++++++++++++++++++++--------
 1 files changed, 25 insertions(+), 8 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/common/system/controller/LoginController.java b/src/main/java/com/xcong/excoin/common/system/controller/LoginController.java
index 28feccb..e10dc75 100644
--- a/src/main/java/com/xcong/excoin/common/system/controller/LoginController.java
+++ b/src/main/java/com/xcong/excoin/common/system/controller/LoginController.java
@@ -1,12 +1,17 @@
 package com.xcong.excoin.common.system.controller;
 
 import cn.hutool.core.util.IdUtil;
+import cn.hutool.crypto.asymmetric.KeyType;
+import cn.hutool.crypto.asymmetric.RSA;
 import com.alibaba.fastjson.JSONObject;
 import com.xcong.excoin.common.contants.AppContants;
 import com.xcong.excoin.common.response.Result;
 import com.xcong.excoin.common.system.bean.LoginUserBean;
 import com.xcong.excoin.common.system.dto.LoginDto;
+import com.xcong.excoin.configurations.properties.ApplicationProperties;
+import com.xcong.excoin.configurations.properties.SecurityProperties;
 import com.xcong.excoin.utils.RedisUtils;
+import io.swagger.annotations.Api;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
@@ -26,12 +31,16 @@
  * @Version V1.0
  **/
 @Slf4j
+@Api(value = "登陆类", tags = "登陆类")
 @RestController
 @RequestMapping(value = "/")
 public class LoginController {
 
-    @Value("${rsa.private_key}")
-    private String privateKey;
+    @Resource
+    private ApplicationProperties applicationProperties;
+
+    @Resource
+    private SecurityProperties securityProperties;
 
     @Resource
     private AuthenticationManagerBuilder authenticationManagerBuilder;
@@ -46,12 +55,20 @@
         String token = IdUtil.simpleUUID();
         LoginUserBean loginUserBean = (LoginUserBean) authentication.getPrincipal();
         redisUtils.set(AppContants.APP_LOGIN_PREFIX + token, JSONObject.toJSONString(loginUserBean), 300000);
-        Map<String, Object> authInfo = new HashMap<String, Object>(2){
-            {
-                put("token", token);
-                put("user", loginUserBean);
-            }
-        };
+        Map<String, Object> authInfo = new HashMap<>();
+        if (applicationProperties.isDebug()) {
+            authInfo.put("token", token);
+            authInfo.put("rsaToken", generateAsaToken(token));
+            authInfo.put("user", loginUserBean);
+        } else {
+            authInfo.put("token", token);
+            authInfo.put("user", loginUserBean);
+        }
         return Result.ok("success", authInfo);
     }
+
+    public String generateAsaToken(String token) {
+        RSA rsa = new RSA(null, securityProperties.getPublicKey());
+        return rsa.encryptBase64(token + "_" +System.currentTimeMillis(), KeyType.PublicKey);
+    }
 }

--
Gitblit v1.9.1