Pārlūkot izejas kodu

添加用户供应商类型的标志,校验供应商的有效性

v_HuilingDeng 4 gadi atpakaļ
vecāks
revīzija
933b6b9968

+ 13 - 1
abi-cloud-qr-platform-server/src/main/java/com/abi/qms/platform/service/impl/UserManagerServiceImpl.java

@@ -89,9 +89,11 @@ public class UserManagerServiceImpl extends ServiceImpl<UserInfoMapper, UserInfo
             throw new BusinessException("非ABI用户必须选择供应商");
         }
         //查询供应商
+        Integer factoryType = null;
         if(req.getFactoryId()!=null){
             BaseFactory factory = baseFactoryMapper.selectById(req.getFactoryId());
             AssertUtil.isNull(factory,"供应商不存在");
+            factoryType = factory.getFactoryType();
         }
 
         //操作人
@@ -105,6 +107,8 @@ public class UserManagerServiceImpl extends ServiceImpl<UserInfoMapper, UserInfo
         userInfo.setPassword(req.getPassword());
         userInfo.setIsAbi(req.getIsAbi());
         userInfo.setFactoryId(req.getFactoryId());
+        //添加用户供应商类型的标志
+        userInfo.setFactoryType(factoryType);
         userInfo.setRoleCode(req.getRoleCode());
         userInfo.setPhone(req.getPhone());
         userInfoMapper.insert(userInfo);
@@ -143,7 +147,15 @@ public class UserManagerServiceImpl extends ServiceImpl<UserInfoMapper, UserInfo
         //1-修改主表
         userInfo.setEmail(req.getEmail());
         userInfo.setPhone(req.getPhone());
-        userInfo.setFactoryId(req.getFactoryId());
+        //校验供应商
+        if(Objects.nonNull(req.getFactoryId())) {
+            if(!userInfo.getFactoryId().equals(req.getFactoryId())) {
+                BaseFactory factory = baseFactoryMapper.selectById(req.getFactoryId());
+                AssertUtil.isNull(factory,"供应商不存在");
+                userInfo.setFactoryId(req.getFactoryId());
+                userInfo.setFactoryType(factory.getFactoryType());
+            }
+        }
         userInfo.setRoleCode(req.getRoleCode());
         userInfoMapper.updateById(userInfo);