fuliqi
2024-12-12 23e187554324e1bdff896d7d18634711a7372755
business/src/main/java/com/ycl/service/impl/ProjectPlanRecordServiceImpl.java
@@ -6,12 +6,14 @@
import com.ycl.common.base.Result;
import com.ycl.common.utils.DateUtils;
import com.ycl.domain.entity.Plan;
import com.ycl.domain.entity.ProjectPlanInfo;
import com.ycl.domain.entity.ProjectPlanRecord;
import com.ycl.domain.vo.ProjectPlanRecordAddRequest;
import com.ycl.domain.vo.ProjectPlanRecordResponseVO;
import com.ycl.domain.vo.ProjetPlanRecordItem;
import com.ycl.framework.utils.PageUtil;
import com.ycl.mapper.PlanMapper;
import com.ycl.mapper.ProjectPlanInfoMapper;
import com.ycl.mapper.ProjectPlanRecordMapper;
import com.ycl.service.ProjectPlanRecordService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
@@ -43,6 +45,7 @@
    private final ProjectPlanRecordMapper projectPlanRecordMapper;
    private final PlanMapper planMapper;
    private final ProjectPlanInfoMapper projectPlanInfoMapper;
    /**
     * 添加
@@ -52,8 +55,9 @@
    @Override
    public Result add(ProjectPlanRecordAddRequest request) {
        ProjectPlanRecord item = new ProjectPlanRecord();
        Plan one = new LambdaQueryChainWrapper<>(planMapper).eq(Plan::getProjectInfoId, request.getProjectInfoId()).one();
        // 查询项目计划id
        item.setPlanId(new LambdaQueryChainWrapper<>(planMapper).eq(Plan::getProjectInfoId, request.getProjectInfoId()).one().getId().longValue());
        item.setPlanId(new LambdaQueryChainWrapper<>(planMapper).eq(Plan::getProjectInfoId, request.getProjectInfoId()).one().getId());
        // 判断标志位是否为0,如果为0,则为月度计划,1为季度计划,2为年度计划
        if (request.getPlanTimeFlag() == MONTH_FLAG) {
            // 判断id是否为0
@@ -149,7 +153,12 @@
     */
    @Override
    public Result removeById(String id) {
        // 删除计划记录
        baseMapper.deleteById(id);
        // 删除计划项
        new LambdaUpdateChainWrapper<>(projectPlanInfoMapper)
                .eq(ProjectPlanInfo::getProjectPlanRecordId, id)
                .remove();
        return Result.ok("删除成功");
    }