射洪市项目管理系统
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
fuliqi
2024-11-28 a1fc7e3e337393954a95b5f85c1a3f59d61208c4 
 business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java


@@ -1,18 +1,25 @@


package com.ycl.service.impl;




import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;


import com.baomidou.mybatisplus.core.metadata.IPage;


import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;


import com.ycl.common.base.Result;


import com.ycl.common.enums.business.FileTypeEnum;


import com.ycl.common.utils.SecurityUtils;


import com.ycl.domain.entity.File;


import com.ycl.domain.entity.ProjectInfo;


import com.ycl.domain.form.ProjectInfoForm;


import com.ycl.domain.query.ProjectInfoQuery;


import com.ycl.domain.vo.*;


import com.ycl.framework.utils.PageUtil;


import com.ycl.mapper.FileMapper;


import com.ycl.mapper.ProjectInfoMapper;


import com.ycl.service.FileService;


import com.ycl.service.ProjectInfoService;


import lombok.RequiredArgsConstructor;


import org.springframework.beans.BeanUtils;


import org.springframework.stereotype.Service;


import org.springframework.transaction.annotation.Transactional;


import org.springframework.util.Assert;




import java.util.ArrayList;


@@ -32,7 +39,8 @@


public class ProjectInfoServiceImpl extends ServiceImpl<ProjectInfoMapper, ProjectInfo> implements ProjectInfoService {




    private final ProjectInfoMapper projectInfoMapper;




    private final FileService fileService;


    private final FileMapper fileMapper;


    /**


     * 添加


     *


@@ -40,10 +48,22 @@


     * @return


     */


    @Override


    @Transactional(rollbackFor = Exception.class)


    public Result add(ProjectInfoForm form) {


        //添加基本信息


        ProjectInfo entity = ProjectInfoForm.getEntityByForm(form, null);


        Long userId = SecurityUtils.getUserId();


        entity.setCreateBy(userId);


        entity.setUpdateBy(userId);


        baseMapper.insert(entity);


        return Result.ok("添加成功");


        //添加文件


        List<File> fileList = form.getFileList();


        fileList.forEach(item->{


            item.setBusId(entity.getId());


            item.setType(FileTypeEnum.PROJECT_INFO);


        });


        fileService.saveBatch(fileList);


        return Result.ok("添加成功").data(entity.getId());


    }




    /**


@@ -53,12 +73,28 @@


     * @return


     */


    @Override


    @Transactional(rollbackFor = Exception.class)


    public Result update(ProjectInfoForm form) {


        ProjectInfo entity = baseMapper.selectById(form.getId());




        // 为空抛IllegalArgumentException，做全局异常处理


        Assert.notNull(entity, "记录不存在");


        BeanUtils.copyProperties(form, entity);


        ProjectInfoForm.getEntityByForm(form,entity);


        Long userId = SecurityUtils.getUserId();


        entity.setUpdateBy(userId);




        List<File> fileList = form.getFileList();


        fileList.forEach(item->{


            item.setBusId(entity.getId());


            item.setType(FileTypeEnum.PROJECT_INFO);


        });


        //删除原有文件


        QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>();


        fileQueryWrapper.eq("type",FileTypeEnum.PROJECT_INFO.getType());


        fileQueryWrapper.eq("bus_id",entity.getId());


        fileMapper.delete(fileQueryWrapper);


        //替换成现有


        fileService.saveBatch(fileList);


        //更新项目信息


        baseMapper.updateById(entity);


        return Result.ok("修改成功");


    }


@@ -70,6 +106,7 @@


     * @return


     */


    @Override


    //TODO:待完善


    public Result remove(List<String> ids) {


        baseMapper.deleteBatchIds(ids);


        return Result.ok("删除成功");


@@ -82,6 +119,7 @@


     * @return


     */


    @Override


    //TODO:待完善


    public Result removeById(String id) {


        baseMapper.deleteById(id);


        return Result.ok("删除成功");


@@ -95,9 +133,17 @@


     */


    @Override


    public Result page(ProjectInfoQuery query) {


        IPage<ProjectInfoVO> page = PageUtil.getPage(query, ProjectInfoVO.class);


        IPage<ProjectInfo> page = PageUtil.getPage(query, ProjectInfo.class);


        baseMapper.getPage(page, query);


        return Result.ok().data(page.getRecords()).total(page.getTotal());


        List<ProjectInfo> records = page.getRecords();


        List<ProjectInfoVO> list = records.stream()


                .map(entity -> {


                    ProjectInfoVO vo = ProjectInfoVO.getVoByEntity(entity, null);


                    vo.setProjectColorCode("green");


                    return vo;


                })


                .collect(Collectors.toList());


        return Result.ok().data(list).total(page.getTotal());


    }




    /**


@@ -108,8 +154,14 @@


     */


    @Override


    public Result detail(Integer id) {


        ProjectInfoVO vo = baseMapper.getById(id);


        Assert.notNull(vo, "记录不存在");


        ProjectInfo entity = baseMapper.getById(id);


        Assert.notNull(entity, "记录不存在");


        ProjectInfoVO vo = ProjectInfoVO.getVoByEntity(entity, null);


        QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>();


        fileQueryWrapper.eq("type",FileTypeEnum.PROJECT_INFO.getType());


        fileQueryWrapper.eq("bus_id",vo.getId());


        List<File> files = fileMapper.selectList(fileQueryWrapper);


        vo.setFileList(files);


        return Result.ok().data(vo);


    }