From 50d3d5e3e3282f757ea639f9ca1939d429c6fd5d Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Mon, 30 Nov 2020 11:02:07 +0800 Subject: [PATCH] modify --- src/main/java/com/xcong/excoin/common/system/controller/LoginController.java | 62 ++++++++++++++++++++++++++---- 1 files changed, 53 insertions(+), 9 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 c90ce97..27bcd67 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,23 +1,33 @@ package com.xcong.excoin.common.system.controller; +import cn.hutool.core.codec.Base64; import cn.hutool.core.util.IdUtil; +import cn.hutool.core.util.StrUtil; import cn.hutool.crypto.SecureUtil; import cn.hutool.crypto.asymmetric.KeyType; import cn.hutool.crypto.asymmetric.RSA; import cn.hutool.crypto.asymmetric.Sign; import cn.hutool.crypto.asymmetric.SignAlgorithm; import com.alibaba.fastjson.JSONObject; +import com.xcong.excoin.common.LoginUserUtils; +import com.xcong.excoin.common.annotations.SubmitRepeat; 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.common.system.dto.OutCenterRegisterDto; import com.xcong.excoin.common.system.dto.RegisterDto; +import com.xcong.excoin.common.system.dto.WtWalletDto; import com.xcong.excoin.configurations.properties.ApplicationProperties; import com.xcong.excoin.configurations.properties.SecurityProperties; +import com.xcong.excoin.modules.login.vo.LoginVo; import com.xcong.excoin.modules.member.service.MemberService; import com.xcong.excoin.utils.RedisUtils; + import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; +import io.swagger.annotations.ApiResponse; +import io.swagger.annotations.ApiResponses; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Value; import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; @@ -27,6 +37,7 @@ import org.springframework.web.bind.annotation.*; import javax.annotation.Resource; +import javax.servlet.http.HttpServletRequest; import java.util.HashMap; import java.util.Map; @@ -59,13 +70,9 @@ @ApiOperation(value = "登陆接口", notes = "登陆接口") @PostMapping("/login") - public Result login(@RequestBody @Validated LoginDto loginDto) { - // 使用md5加密前端传来的密码 - Sign sign = SecureUtil.sign(SignAlgorithm.MD5withRSA); - byte[] pwdByte = sign.sign(loginDto.getPassword().getBytes()); - + public Result login(@RequestBody @Validated LoginDto loginDto, HttpServletRequest request) { // 将账号密码交给spring security验证,并调用userServiceDetails - UsernamePasswordAuthenticationToken authToken = new UsernamePasswordAuthenticationToken(loginDto.getUsername(), loginDto.getPassword()); + UsernamePasswordAuthenticationToken authToken = new UsernamePasswordAuthenticationToken(loginDto.getUsername(), SecureUtil.md5(loginDto.getPassword())); Authentication authentication = authenticationManagerBuilder.getObject().authenticate(authToken); // 获取当前验证过后的用户 @@ -73,12 +80,30 @@ // 生成UUID作为token String token = IdUtil.simpleUUID(); - redisUtils.set(AppContants.APP_LOGIN_PREFIX + token, JSONObject.toJSONString(loginUserBean), applicationProperties.getRedisExpire()); + String redisToken = ""; + String redisMember = ""; + if (LoginUserUtils.isBrowser(request)) { + redisToken = AppContants.PC_LOGIN_PREFIX + token; + redisMember = AppContants.PC_LOGIN_PREFIX + loginUserBean.getMemberEntity().getId(); + } else { + redisToken = AppContants.APP_LOGIN_PREFIX + token; + redisMember = AppContants.APP_LOGIN_PREFIX + loginUserBean.getMemberEntity().getId(); + } + + if (StrUtil.isNotBlank(redisUtils.getString(redisMember))) { + if (redisMember.contains(AppContants.APP_LOGIN_PREFIX)) { + redisUtils.del(AppContants.APP_LOGIN_PREFIX + redisUtils.getString(redisMember)); + } else { + redisUtils.del(AppContants.PC_LOGIN_PREFIX + redisUtils.getString(redisMember)); + } + } + redisUtils.set(redisToken, JSONObject.toJSONString(loginUserBean.getMemberEntity()), applicationProperties.getRedisExpire()); + redisUtils.set(redisMember, token); Map<String, Object> authInfo = new HashMap<>(); // 开启debug模式,则将加密后的token返回 if (applicationProperties.isDebug()) { authInfo.put("token", token); - authInfo.put("rsaToken", generateAsaToken(token)); + authInfo.put("rsaToken", AppContants.TOKEN_START_WITH + generateAsaToken(token)); authInfo.put("user", loginUserBean); } else { authInfo.put("token", token); @@ -89,13 +114,32 @@ public String generateAsaToken(String token) { RSA rsa = new RSA(null, securityProperties.getPublicKey()); - return rsa.encryptBase64(token + "_" +System.currentTimeMillis(), KeyType.PublicKey); + return rsa.encryptBase64(token + "_" + System.currentTimeMillis(), KeyType.PublicKey); } + //@SubmitRepeat @ApiOperation(value = "app注册接口", notes = "app注册接口,验证码必须输入可默认为123456") @PostMapping(value = "/register") public Result register(@RequestBody @Validated RegisterDto registerDto) { return memberservice.register(registerDto); } + + @ApiOperation(value = "创建接口", notes = "创建接口") + @PostMapping(value = "/registerOutCenter") + @ApiResponses({ + @ApiResponse(code = 200,message = "OK",response = LoginVo.class), + }) + public Result registerOutCenter(@RequestBody @Validated OutCenterRegisterDto outCenterRegisterDto) { + return memberservice.registerOutCenter(outCenterRegisterDto); + } + + @ApiOperation(value = "导入接口", notes = "导入接口") + @PostMapping(value = "/loginOutCenter") + @ApiResponses({ + @ApiResponse(code = 200,message = "OK",response = LoginVo.class), + }) + public Result loginOutCenter(@RequestBody WtWalletDto wtWalletDto) { + return memberservice.recovery(wtWalletDto); + } } -- Gitblit v1.9.1