luohairen
2024-12-25 488ac4d44ee6a2dfb841cf047083cba80dfb5432
business/src/main/java/com/ycl/service/impl/PlanServiceImpl.java
@@ -2,7 +2,10 @@
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.ycl.common.base.Result;
import com.ycl.common.enums.business.ProjectStatusEnum;
import com.ycl.common.enums.business.ProjectTypeEnum;
import com.ycl.domain.entity.Plan;
import com.ycl.domain.vo.ProjectPlanResponseVO;
import com.ycl.framework.utils.PageUtil;
import com.ycl.mapper.PlanMapper;
import com.ycl.service.PlanService;
@@ -87,8 +90,16 @@
     */
    @Override
    public Result page(PlanQuery query) {
        IPage<PlanVO> page = PageUtil.getPage(query, PlanVO.class);
        IPage<ProjectPlanResponseVO> page = PageUtil.getPage(query, ProjectPlanResponseVO.class);
        baseMapper.getPage(page, query);
        // 对分页后的属性进行处理
        List<ProjectPlanResponseVO> records = page.getRecords();
        for (ProjectPlanResponseVO record : records) {
            record.setProjectType(ProjectTypeEnum.getDescByType(record.getProjectType()));
            record.setProjectStatus(ProjectStatusEnum.getDescByType(record.getProjectStatus()));
            record.setProjectColorCode("green");
        }
        return Result.ok().data(page.getRecords()).total(page.getTotal());
    }