package com.ycl.service.impl; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.ycl.common.utils.SecurityUtils; import com.ycl.domain.entity.ProjectInvestmentInfo; import com.ycl.mapper.ProjectInvestmentInfoMapper; import com.ycl.service.ProjectInvestmentInfoService; import com.ycl.common.base.Result; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.ycl.domain.form.ProjectInvestmentInfoForm; import com.ycl.domain.vo.ProjectInvestmentInfoVO; import com.ycl.domain.query.ProjectInvestmentInfoQuery; import org.springframework.stereotype.Service; import lombok.RequiredArgsConstructor; import com.ycl.framework.utils.PageUtil; import org.springframework.beans.BeanUtils; import org.springframework.util.Assert; import java.util.List; import java.util.stream.Collectors; /** * 投资项目基础信息表 服务实现类 * * @author flq * @since 2024-11-27 */ @Service @RequiredArgsConstructor public class ProjectInvestmentInfoServiceImpl extends ServiceImpl implements ProjectInvestmentInfoService { private final ProjectInvestmentInfoMapper projectInvestmentInfoMapper; /** * 添加 * @param form * @return */ @Override public Result add(ProjectInvestmentInfoForm form) { ProjectInvestmentInfo entity = ProjectInvestmentInfoForm.getEntityByForm(form, null); Long userId = SecurityUtils.getUserId(); entity.setCreateBy(userId); entity.setUpdateBy(userId); baseMapper.insert(entity); //避免新增一个后重复新增返回id给前端判断 return Result.ok("添加成功").data(entity.getId()); } /** * 修改 * @param form * @return */ @Override public Result update(ProjectInvestmentInfoForm form) { ProjectInvestmentInfo entity = baseMapper.selectById(form.getId()); // 为空抛IllegalArgumentException,做全局异常处理 Assert.notNull(entity, "记录不存在"); BeanUtils.copyProperties(form, entity); Long userId = SecurityUtils.getUserId(); entity.setUpdateBy(userId); baseMapper.updateById(entity); return Result.ok("修改成功"); } /** * 批量删除 * @param ids * @return */ @Override public Result remove(List 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(ProjectInvestmentInfoQuery query) { IPage page = PageUtil.getPage(query, ProjectInvestmentInfoVO.class); baseMapper.getPage(page, query); return Result.ok().data(page.getRecords()).total(page.getTotal()); } /** * 根据id查找 * @param id * @return */ @Override public Result detail(Integer id) { QueryWrapper queryWrapper = new QueryWrapper<>(); queryWrapper.eq("project_id",id); ProjectInvestmentInfo entity = baseMapper.selectOne(queryWrapper); ProjectInvestmentInfoVO vo = new ProjectInvestmentInfoVO(); if(entity!=null) { vo = ProjectInvestmentInfoVO.getVoByEntity(entity, null); } return Result.ok().data(vo); } /** * 列表 * @return */ @Override public Result all() { List entities = baseMapper.selectList(null); List vos = entities.stream() .map(entity -> ProjectInvestmentInfoVO.getVoByEntity(entity, null)) .collect(Collectors.toList()); return Result.ok().data(vos); } }