diff --git a/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionStaffServiceImpl.java b/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionStaffServiceImpl.java index c6a1ce90..4f911528 100644 --- a/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionStaffServiceImpl.java +++ b/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionStaffServiceImpl.java @@ -230,21 +230,26 @@ public class InspectionStaffServiceImpl extends ServiceImpllambdaQuery().eq(InspectionStaff::getUserId, userId)); - if (ObjectUtil.isNull(staff)) { - staff = new InspectionStaff(); - staff.setUserId(userId); - staff.setFolderId(folderId); - //新增 - this.save(staff); - } else { - //修改文件夹id - this.update(Wrappers.lambdaUpdate().eq(InspectionStaff::getUserId, userId).set(InspectionStaff::getFolderId, folderId)); + //查询员工子表是否存在数据 + InspectionStaff staff = this.getOne(Wrappers.lambdaQuery().eq(InspectionStaff::getUserId, userId)); + if (ObjectUtil.isNull(staff)) { + staff = new InspectionStaff(); + staff.setUserId(userId); + staff.setFolderId(folderId); + //新增 + this.save(staff); + } else { + //修改文件夹id + this.update(Wrappers.lambdaUpdate().eq(InspectionStaff::getUserId, userId).set(InspectionStaff::getFolderId, folderId)); + } + return folderId; + }else { + log.error("用户不存在"); + return null; } - return folderId; } /** @@ -265,6 +270,10 @@ public class InspectionStaffServiceImpl extends ServiceImpllambdaUpdate().eq(InspectionStaff::getUserId, userId).set(InspectionStaff::getFolderId, folderId)); }