zero hai 11 meses
pai
achega
a82ac7d6b2

+ 2 - 0
PCS/src/main/java/cn/cslg/permission/mapper/qiaobi/AssoPersonVipTypeMapper.java

@@ -2,6 +2,7 @@ package cn.cslg.permission.mapper.qiaobi;
 
 import cn.cslg.permission.domain.qiaobi.AssoPersonVipType;
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Mapper;
 
 /**
 * @author admin
@@ -9,6 +10,7 @@ import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 * @createDate 2024-11-01 11:06:57
 * @Entity generator.domain.AssoPersonVipType
 */
+@Mapper
 public interface AssoPersonVipTypeMapper extends BaseMapper<AssoPersonVipType> {
 
 }

+ 9 - 0
PCS/src/main/java/cn/cslg/permission/service/LoginService.java

@@ -10,6 +10,7 @@ import cn.cslg.permission.common.utils.message.MessageUtils;
 import cn.cslg.permission.domain.*;
 import cn.cslg.permission.domain.associate.AssoFunctionModule;
 import cn.cslg.permission.domain.associate.AssoPersonnelMachine;
+import cn.cslg.permission.domain.qiaobi.AssoPersonVipType;
 import cn.cslg.permission.exception.ExceptionEnum;
 import cn.cslg.permission.exception.XiaoShiException;
 import cn.cslg.permission.mapper.FunctionMapper;
@@ -17,6 +18,7 @@ import cn.cslg.permission.mapper.PersonInvitationCodeMapper;
 import cn.cslg.permission.mapper.PersonnelMapper;
 import cn.cslg.permission.mapper.associate.AssoFunctionModuleMapper;
 import cn.cslg.permission.mapper.associate.AssoPersonnelMachineMapper;
+import cn.cslg.permission.mapper.qiaobi.AssoPersonVipTypeMapper;
 import cn.cslg.permission.service.associate.PerDpService;
 import cn.cslg.permission.service.associate.PersonRoleService;
 import cn.cslg.permission.service.associate.RoleFunctionDataService;
@@ -84,6 +86,8 @@ public class LoginService extends ServiceImpl<PersonnelMapper, Personnel> {
     private AssoFunctionModuleMapper assoFunctionModuleMapper;
     @Autowired
     private PersonInvitationCodeMapper personInvitationCodeMapper;
+    @Autowired
+    private AssoPersonVipTypeMapper assoPersonVipTypeMapper;
 
     /**
      * @param loginVO 登录参数类
@@ -349,6 +353,11 @@ public class LoginService extends ServiceImpl<PersonnelMapper, Personnel> {
     //校验登录时参数
     public EncryptionLoginVO loginCommonMethod(LoginCommonDTO vo,Personnel personnel) throws Exception {
         Integer personnelId = personnel.getId();
+        List<AssoPersonVipType> personVipTypes = assoPersonVipTypeMapper.selectList(new LambdaQueryWrapper<AssoPersonVipType>()
+                .eq(AssoPersonVipType::getPersonId, personnelId));
+        if (org.springframework.util.CollectionUtils.isEmpty(personVipTypes)) {
+            throw new XiaoShiException(ExceptionEnum.PERMISSION_ERROR);
+        }
         Long timeMillis = vo.getCurrentTimeMillis();
         String appKey = vo.getAppKey();
         String sign = vo.getSign();