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