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