血缘查询--工号校验

src/main/java/com/ydtech/modules/admin/controller/SysUserController.java

@@ -729,7 +729,7 @@ public class SysUserController extends BaseController {
     @ApiOperation(value = "获取推荐人分页信息")
     @PostMapping(value = "/referrerFindPage")
     public HttpResult<PageVo<SysReferrerVo>> referrerFindPage(@RequestBody PageDto<SysReferrerVo> pageDto) {
-        return HttpResult.ok(esmUserReferrerService.gainReferrerPage(pageDto));
+        return esmUserReferrerService.gainReferrerPage(pageDto);
     }
 
 

src/main/java/com/ydtech/modules/esm/service/EsmUserReferrerService.java

@@ -2,6 +2,7 @@ package com.ydtech.modules.esm.service;
 
 
 import com.baomidou.mybatisplus.extension.service.IService;
+import com.ydtech.core.page.HttpResult;
 import com.ydtech.modules.admin.model.SysUser;
 import com.ydtech.modules.admin.model.vo.SysReferrerVo;
 import com.ydtech.modules.base.model.dto.PageDto;
@@ -39,7 +40,7 @@ public interface EsmUserReferrerService extends IService<EsmUserReferrer> {
      * @author: lig
      * @date: 2023年08月08日 0008
      */
-    public PageVo<SysReferrerVo> gainReferrerPage(PageDto<SysReferrerVo> pageDto);
+    public HttpResult<PageVo<SysReferrerVo>> gainReferrerPage(PageDto<SysReferrerVo> pageDto);
 
     public List<EsmUserPolicyVO> countUserPolicy(String wCase, Object[] paramValues);
 

src/main/java/com/ydtech/modules/esm/service/impl/EsmUserReferrerServiceImpl.java

@@ -3,6 +3,7 @@ package com.ydtech.modules.esm.service.impl;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ydtech.core.page.HttpResult;
 import com.ydtech.exception.SystemException;
 import com.ydtech.modules.admin.model.SysUser;
 import com.ydtech.modules.admin.model.vo.SysReferrerVo;
@@ -117,7 +118,7 @@ public class EsmUserReferrerServiceImpl extends ServiceImpl<EsmUserReferrerMappe
         return remove(queryWrapper);
     }
 
- 
+
     List<String>  getUpUser(String userId,List<String> list){
 
         EsmUserReferrer esmUserReferrer=this.getById(userId);
@@ -130,9 +131,13 @@ public class EsmUserReferrerServiceImpl extends ServiceImpl<EsmUserReferrerMappe
     }
 
     @Override
-    public PageVo gainReferrerPage(PageDto<SysReferrerVo> pageDto) {
+    public HttpResult<PageVo<SysReferrerVo>> gainReferrerPage(PageDto<SysReferrerVo> pageDto) {
 
         if(StringUtils.isNotEmpty(pageDto.getDto().getAffiliation())){
+            EsmUserReferrer esmUserReferrer=this.getById(pageDto.getDto().getAffiliation());
+            if(esmUserReferrer==null){
+                return HttpResult.error("工号有误，请重新输入");
+            }
             List<String> up3list=new ArrayList<>();
             getUpUser(pageDto.getDto().getAffiliation(),up3list);
             List<String> down3list=this.baseMapper.getIdDown3(pageDto.getDto().getAffiliation());
@@ -167,7 +172,7 @@ public class EsmUserReferrerServiceImpl extends ServiceImpl<EsmUserReferrerMappe
         pageDto.setPageNum(firstIndex);
         pageVo.setContent(super.baseMapper.selectReferrerByAffiliation(pageDto));
 
-        return pageVo;
+        return HttpResult.ok(pageVo);
     }