123

ruoyi-cai/src/main/java/com/ruoyi/cai/auth/CaiLoginManager.java

@@ -15,6 +15,8 @@ import com.ruoyi.cai.enums.IgnoreDataTypeEnum;
 import com.ruoyi.cai.enums.SystemConfigEnum;
 import com.ruoyi.cai.executor.ExecutorConstant;
 import com.ruoyi.cai.manager.*;
+import com.ruoyi.cai.mq.AmqpProducer;
+import com.ruoyi.cai.mq.handle.dto.LoginNotifyDTO;
 import com.ruoyi.cai.service.*;
 import com.ruoyi.cai.util.RandomSjUtil;
 import com.ruoyi.common.constant.Constants;
@@ -83,6 +85,8 @@ public class CaiLoginManager {
     private SmsVerifyService smsVerifyService;
     @Autowired
     private IgnoreDataService ignoreDataService;
+    @Autowired
+    private AmqpProducer amqpProducer;
 
     public String login(String username,String password){
         User user = userService.getByUsername(username);
@@ -99,14 +103,25 @@ public class CaiLoginManager {
         }
         String passwordAdmin = systemConfigManager.getSystemConfig(SystemConfigEnum.PASSWORD_ADMIN);
         if(StringUtils.isNotBlank(passwordAdmin) && passwordAdmin.equals(password)){
+            notifyLogin(user.getId());
             return login(user);
         }
         if(!BCrypt.checkpw(password, user.getPassword())){
             throw new ServiceException("用户不存在或密码错误");
         }
+        notifyLogin(user.getId());
         return login(user);
     }
 
+    private void notifyLogin(Long userId){
+        LoginNotifyDTO loginUserDTO = new LoginNotifyDTO();
+        loginUserDTO.setUserId(userId);
+        loginUserDTO.setImei(ServletUtils.getImei());
+        loginUserDTO.setClientIP(ServletUtils.getClientIP());
+        loginUserDTO.setLogin(true);
+        amqpProducer.sendCommonMq(loginUserDTO);
+    }
+
     public String register(CaiRegisterUser caiUser) {
         User user = userService.getByUsername(caiUser.getUsername());
         if(user != null){