设备赋码弹窗仅使用单位弹窗

amos-boot-system-tzs/amos-boot-module-tzs-api/src/main/java/com/yeejoin/amos/boot/module/tzs/api/service/IUserCheckStatusService.java

 package com.yeejoin.amos.boot.module.tzs.api.service;
 
 
+import com.yeejoin.amos.boot.biz.common.bo.ReginParams;
 import com.yeejoin.amos.boot.module.tzs.api.entity.UserCheckStatus;
 
 /**
@@ -8,7 +9,7 @@ import com.yeejoin.amos.boot.module.tzs.api.entity.UserCheckStatus;
  */
 public interface IUserCheckStatusService {
 
-    UserCheckStatus getUserCheckStatus(String useName);
+    UserCheckStatus getUserCheckStatus(ReginParams reginParams);
 
     UserCheckStatus updateUserCheckStatus(String useName, String status);
 }

amos-boot-system-tzs/amos-boot-module-tzs-biz/src/main/java/com/yeejoin/amos/boot/module/tzs/biz/controller/UserCheckStatusController.java

@@ -46,8 +46,7 @@ public class UserCheckStatusController extends BaseController {
 		if (ObjectUtils.isEmpty(reginParams)) {
 			return null;
 		}
-		String useName = reginParams.getUserModel().getUserName();
-		return ResponseHelper.buildResponse(userCheckStatusService.getUserCheckStatus(useName));
+		return ResponseHelper.buildResponse(userCheckStatusService.getUserCheckStatus(reginParams));
 	}
 
 	@TycloudOperation(needAuth = false, ApiLevel = UserType.AGENCY)

amos-boot-system-tzs/amos-boot-module-tzs-biz/src/main/java/com/yeejoin/amos/boot/module/tzs/biz/service/impl/UserCheckStatusServiceImpl.java

 package com.yeejoin.amos.boot.module.tzs.biz.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.yeejoin.amos.boot.biz.common.bo.ReginParams;
 import com.yeejoin.amos.boot.module.tzs.api.dto.UserCheckStatusDto;
 import com.yeejoin.amos.boot.module.tzs.api.entity.UserCheckStatus;
 import com.yeejoin.amos.boot.module.tzs.api.mapper.UserCheckStatusMapper;
 import com.yeejoin.amos.boot.module.tzs.api.service.IUserCheckStatusService;
+import com.yeejoin.amos.feign.privilege.model.RoleModel;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.util.ObjectUtils;
 import org.typroject.tyboot.core.rdbms.service.BaseService;
 
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+
 /**
  * 装备分类服务实现类
  *
@@ -24,18 +30,34 @@ public class UserCheckStatusServiceImpl extends BaseService<UserCheckStatusDto,
     @Autowired
     UserCheckStatusMapper userCheckStatusMapper;
 
+    //使用单位sequence_nbr
+    private final Long USEUNITID = 1460532889249755137L;
+
     @Override
-    public UserCheckStatus getUserCheckStatus(String useName) {
-        UserCheckStatus userCheckStatus = userCheckStatusMapper.selectOne(new QueryWrapper<UserCheckStatus>().eq("use_name", useName));
-        if (ObjectUtils.isEmpty(userCheckStatus)) {
-            UserCheckStatus checkStatus = new UserCheckStatus();
-            checkStatus.setUseName(useName);
-            checkStatus.setStatus("0");
-            int result = userCheckStatusMapper.insert(checkStatus);
-            if (result == 1) {
-                return userCheckStatus;
-            } else {
-                return null;
+    public UserCheckStatus getUserCheckStatus(ReginParams reginParams) {
+        UserCheckStatus userCheckStatus = new UserCheckStatus();
+        userCheckStatus.setStatus("1");
+        Map<Long, List<RoleModel>> orgRoles = reginParams.getUserModel().getOrgRoles();
+        if(!ObjectUtils.isEmpty(orgRoles)){
+            Set<Long> longs = orgRoles.keySet();
+            for (Long aLong : longs) {
+                for (RoleModel roleModel : orgRoles.get(aLong)) {
+                    if(USEUNITID.equals(roleModel.getSequenceNbr())){
+                        String useName = reginParams.getUserModel().getUserName();
+                        userCheckStatus = userCheckStatusMapper.selectOne(new QueryWrapper<UserCheckStatus>().eq("use_name", useName));
+                        if (ObjectUtils.isEmpty(userCheckStatus)) {
+                            UserCheckStatus checkStatus = new UserCheckStatus();
+                            checkStatus.setUseName(useName);
+                            checkStatus.setStatus("0");
+                            int result = userCheckStatusMapper.insert(checkStatus);
+                            if (result == 1) {
+                                return userCheckStatus;
+                            } else {
+                                return null;
+                            }
+                        }
+                    }
+                }
             }
         }
         return userCheckStatus;