From 0eaaac435d91a2240a63b5b36758b9044c54c32f Mon Sep 17 00:00:00 2001
From: Junjie <540245094@qq.com>
Date: 星期三, 20 十二月 2023 12:34:41 +0800
Subject: [PATCH] 登录鉴别

---
 zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java |   18 +++++++++++++-----
 1 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java b/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java
index 85d7974..df4a984 100644
--- a/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java
+++ b/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java
@@ -6,6 +6,7 @@
 import com.zy.asrs.common.domain.dto.PowerDto;
 import com.zy.asrs.common.domain.entity.Parameter;
 import com.zy.asrs.common.domain.enums.HtmlNavIconType;
+import com.zy.asrs.common.domain.enums.LoginSystemType;
 import com.zy.asrs.common.properties.OSinfo;
 import com.zy.asrs.common.properties.SystemProperties;
 import com.zy.asrs.common.sys.entity.*;
@@ -46,7 +47,7 @@
 
     @RequestMapping("/login.action")
     @ManagerAuth(value = ManagerAuth.Auth.NONE, memo = "鐧诲綍")
-    public R loginAction(String username, String password){
+    public R loginAction(String username, String password, Boolean wms) {
         if (username.equals("super") && password.equals(Cools.md5(superPwd))) {
             Map<String, Object> res = new HashMap<>();
             res.put("username", username);
@@ -56,21 +57,28 @@
         LambdaQueryWrapper<User> userWrapper = new LambdaQueryWrapper<>();
         userWrapper.eq(User::getUsername, username);
         User user = userService.getOne(userWrapper);
-        if (Cools.isEmpty(user)){
+        if (Cools.isEmpty(user)) {
             return R.parse(CodeRes.USER_10001);
         }
-        if (user.getStatus()!=1){
+        if (user.getStatus() != 1) {
             return R.parse(CodeRes.USER_10002);
         }
-        if (!user.getPassword().equals(password)){
+        if (!user.getPassword().equals(password)) {
             return R.parse(CodeRes.USER_10003);
         }
+        String system = null;//鐧婚檰绯荤粺
+        if (wms) {
+            system = String.valueOf(LoginSystemType.WMS);
+        }else {
+            system = String.valueOf(LoginSystemType.WCS);
+        }
         String token = Cools.enToken(System.currentTimeMillis() + username, user.getPassword());
-        userLoginService.remove(new LambdaQueryWrapper<UserLogin>().eq(UserLogin::getUserId, user.getId()));
+        userLoginService.remove(new LambdaQueryWrapper<UserLogin>().eq(UserLogin::getUserId, user.getId()).eq(UserLogin::getSystem, system));
         UserLogin userLogin = new UserLogin();
         userLogin.setUserId(user.getId());
         userLogin.setToken(token);
         userLogin.setCreateTime(new Date());
+        userLogin.setSystem(system);
         userLoginService.save(userLogin);
         Map<String, Object> res = new HashMap<>();
         res.put("username", user.getUsername());

--
Gitblit v1.9.1