培训计划

cement-business/src/main/java/cn/wise/sc/cement/business/controller/PlanTrainingController.java

@@ -2,11 +2,14 @@ package cn.wise.sc.cement.business.controller;
 
 import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.util.StrUtil;
+import cn.wise.sc.cement.business.entity.PlanTraining;
+import cn.wise.sc.cement.business.mapper.PlanTrainingMapper;
 import cn.wise.sc.cement.business.model.BaseResponse;
 import cn.wise.sc.cement.business.model.PageQuery;
 import cn.wise.sc.cement.business.model.query.PlanTrainingQuery;
 import cn.wise.sc.cement.business.model.vo.PlanTrainingVo;
 import cn.wise.sc.cement.business.service.IPlanTrainingService;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.slf4j.Logger;
@@ -14,6 +17,7 @@ import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import javax.annotation.Resource;
 import javax.servlet.http.HttpServletResponse;
 import java.util.Date;
 import java.util.List;
@@ -35,6 +39,9 @@ public class PlanTrainingController {
 	@Autowired
 	private IPlanTrainingService iplanTrainingService;
 
+	@Resource
+	private PlanTrainingMapper planTrainingMapper;
+
 	@ApiOperation(value = "培训计划分页")
 	@GetMapping("/getPage")
 	public BaseResponse getPage(PageQuery pageQuery) {
@@ -106,6 +113,12 @@ public class PlanTrainingController {
 		if (planTrainingQuery.getRemind() == 1 && planTrainingQuery.getRemindTime() == null) {
 			return BaseResponse.errorMsg("请选择提醒时间");
 		}
+		QueryWrapper<PlanTraining> qw=new QueryWrapper<>();
+		qw.eq("id",planTrainingQuery.getId());
+		int count=planTrainingMapper.selectCount(qw);
+		if (count<=0){
+			return BaseResponse.errorMsg("数据错误");
+		}
 		try {
 			return iplanTrainingService.update(planTrainingQuery);
 		} catch (Exception e) {