|
@@ -4,6 +4,8 @@ package com.management.platform.controller;
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
import com.management.platform.entity.Project;
|
|
|
import com.management.platform.entity.ProjectLevel;
|
|
|
+import com.management.platform.entity.TimeType;
|
|
|
+import com.management.platform.mapper.TimeTypeMapper;
|
|
|
import com.management.platform.mapper.UserMapper;
|
|
|
import com.management.platform.service.ProjectLevelService;
|
|
|
import com.management.platform.service.ProjectService;
|
|
@@ -33,6 +35,8 @@ public class ProjectLevelController {
|
|
|
private UserMapper userMapper;
|
|
|
@Resource
|
|
|
private ProjectService projectService;
|
|
|
+ @Resource
|
|
|
+ private TimeTypeMapper timeTypeMapper;
|
|
|
@RequestMapping("/addOrMod")
|
|
|
public HttpRespMsg addOrMod(HttpServletRequest request, ProjectLevel projectLevel){
|
|
|
return projectLevelService.addOrMod(request,projectLevel);
|
|
@@ -46,14 +50,18 @@ public class ProjectLevelController {
|
|
|
return httpRespMsg;
|
|
|
}
|
|
|
@RequestMapping("/delete")
|
|
|
- public HttpRespMsg delete(Integer id){
|
|
|
+ public HttpRespMsg delete(Integer id,HttpServletRequest request){
|
|
|
HttpRespMsg httpRespMsg=new HttpRespMsg();
|
|
|
- int cut = projectService.count(new QueryWrapper<Project>().eq("project_level_id", id));
|
|
|
- if(cut>0){
|
|
|
- httpRespMsg.setError("已绑定项目");
|
|
|
- return httpRespMsg;
|
|
|
+ Integer companyId = userMapper.selectById(request.getHeader("token")).getCompanyId();
|
|
|
+ TimeType timeType = timeTypeMapper.selectById(companyId);
|
|
|
+ if(timeType.getProjectLevelState()==1){
|
|
|
+ int cut = projectService.count(new QueryWrapper<Project>().eq("level", id).eq("company_id",companyId));
|
|
|
+ if(cut>0){
|
|
|
+ httpRespMsg.setError("已绑定项目");
|
|
|
+ return httpRespMsg;
|
|
|
+ }
|
|
|
+ projectLevelService.removeById(id);
|
|
|
}
|
|
|
- projectLevelService.removeById(id);
|
|
|
return httpRespMsg;
|
|
|
}
|
|
|
}
|