package com.ycl.service.impl;
|
|
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;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ycl.domain.form.PlanForm;
|
import com.ycl.domain.vo.PlanVO;
|
import com.ycl.domain.query.PlanQuery;
|
import org.springframework.stereotype.Service;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.util.Assert;
|
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* 项目计划表 服务实现类
|
*
|
* @author lhr
|
* @since 2024-11-22
|
*/
|
@Service
|
@RequiredArgsConstructor
|
public class PlanServiceImpl extends ServiceImpl<PlanMapper, Plan> implements PlanService {
|
|
private final PlanMapper planMapper;
|
|
/**
|
* 添加
|
* @param form
|
* @return
|
*/
|
@Override
|
public Result add(PlanForm form) {
|
Plan entity = PlanForm.getEntityByForm(form, null);
|
baseMapper.insert(entity);
|
return Result.ok("添加成功");
|
}
|
|
/**
|
* 修改
|
* @param form
|
* @return
|
*/
|
@Override
|
public Result update(PlanForm form) {
|
Plan entity = baseMapper.selectById(form.getId());
|
|
// 为空抛IllegalArgumentException，做全局异常处理
|
Assert.notNull(entity, "记录不存在");
|
BeanUtils.copyProperties(form, entity);
|
baseMapper.updateById(entity);
|
return Result.ok("修改成功");
|
}
|
|
/**
|
* 批量删除
|
* @param ids
|
* @return
|
*/
|
@Override
|
public Result remove(List<String> ids) {
|
baseMapper.deleteBatchIds(ids);
|
return Result.ok("删除成功");
|
}
|
|
/**
|
* id删除
|
* @param id
|
* @return
|
*/
|
@Override
|
public Result removeById(String id) {
|
baseMapper.deleteById(id);
|
return Result.ok("删除成功");
|
}
|
|
/**
|
* 分页查询
|
* @param query
|
* @return
|
*/
|
@Override
|
public Result page(PlanQuery query) {
|
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());
|
}
|
|
/**
|
* 根据id查找
|
* @param id
|
* @return
|
*/
|
@Override
|
public Result detail(Integer id) {
|
PlanVO vo = baseMapper.getById(id);
|
Assert.notNull(vo, "记录不存在");
|
return Result.ok().data(vo);
|
}
|
|
/**
|
* 列表
|
* @return
|
*/
|
@Override
|
public Result all() {
|
List<Plan> entities = baseMapper.selectList(null);
|
List<PlanVO> vos = entities.stream()
|
.map(entity -> PlanVO.getVoByEntity(entity, null))
|
.collect(Collectors.toList());
|
return Result.ok().data(vos);
|
}
|
}
|
