From 300059e394d10191768ec5e2dae03b0344464aea Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 24 Sep 2020 11:51:06 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/common/system/controller/LoginController.java |   23 +++++++++++++++++++++++
 1 files changed, 23 insertions(+), 0 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 d8bfd99..97ee995 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
@@ -10,6 +10,7 @@
 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;
@@ -17,7 +18,10 @@
 import com.xcong.excoin.common.system.dto.RegisterDto;
 import com.xcong.excoin.configurations.properties.ApplicationProperties;
 import com.xcong.excoin.configurations.properties.SecurityProperties;
+import com.xcong.excoin.modules.member.entity.MemberDeviceInfo;
+import com.xcong.excoin.modules.member.service.MemberDeviceInfoService;
 import com.xcong.excoin.modules.member.service.MemberService;
+import com.xcong.excoin.utils.HttpUtils;
 import com.xcong.excoin.utils.RedisUtils;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -61,6 +65,9 @@
     @Resource
     private RedisUtils redisUtils;
 
+    @Resource
+    private MemberDeviceInfoService memberDeviceInfoService;
+
     @ApiOperation(value = "登陆接口", notes = "登陆接口")
     @PostMapping("/login")
     public Result login(@RequestBody @Validated LoginDto loginDto, HttpServletRequest request) {
@@ -102,6 +109,21 @@
             authInfo.put("token", token);
             authInfo.put("user", loginUserBean);
         }
+
+        MemberDeviceInfo deviceInfo = memberDeviceInfoService.getByMemberId(loginUserBean.getMemberEntity().getId());
+        String ip = HttpUtils.getRealIp(request);
+        String userAgent = HttpUtils.getUserAgent(request);
+        if (deviceInfo == null) {
+            deviceInfo = new MemberDeviceInfo();
+            deviceInfo.setIp(ip);
+            deviceInfo.setUserAgent(userAgent);
+            deviceInfo.setMemberId(loginUserBean.getMemberEntity().getId());
+            memberDeviceInfoService.save(deviceInfo);
+        } else {
+            deviceInfo.setIp(ip);
+            deviceInfo.setUserAgent(userAgent);
+            memberDeviceInfoService.updateById(deviceInfo);
+        }
         return Result.ok("success", authInfo);
     }
 
@@ -110,6 +132,7 @@
         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) {

--
Gitblit v1.9.1