代码优化

ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysUserProjectServiceImpl.java

@@ -6,6 +6,7 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.google.common.base.Objects;
 import lombok.RequiredArgsConstructor;
+import org.dromara.common.core.exception.ServiceException;
 import org.dromara.common.core.utils.StreamUtils;
 import org.dromara.common.mybatis.core.mapper.BaseMapperPlus;
 import org.dromara.common.mybatis.service.AbstractBaseService;
@@ -90,6 +91,19 @@ public class SysUserProjectServiceImpl extends AbstractBaseService<SysUserProjec
         return result;
     }
 
+    @Override
+    public void validEntityBeforeSave(SysUserProject entity) {
+        super.validEntityBeforeSave(entity);
+        //校验用户是否已经存在
+        List<SysUserProject> supList = baseMapper.selectList(SysUserProject::getUserId, entity.getUserId(), SysUserProject::getProjectId, entity.getProjectId());
+        SysUserProject sysUserProject = StreamUtils.findFirst(supList, sup -> !Objects.equal(sup.getId(), entity.getId()));
+        if (sysUserProject != null) {
+            SysUser user = sysUserMapper.selectById(entity.getUserId());
+            SysProject project = sysProjectMapper.selectById(entity.getProjectId());
+            throw new ServiceException("账号：" + user.getUserName() + ",已存在项目：" + project.getProjectName() + " 中！");
+        }
+    }
+
     @Override
     public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
         ids.forEach(id -> {