From 8e262ea4cd665da8400d8f5806ebc51ae3d34cf9 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 24 Sep 2020 11:45:13 +0800
Subject: [PATCH] add member device info

---
 src/main/java/com/xcong/excoin/common/system/controller/LoginController.java |   21 +++++++++++++++++++++
 1 files changed, 21 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 4754bb5..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
@@ -18,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;
@@ -62,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) {
@@ -103,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);
     }
 

--
Gitblit v1.9.1