多表增删改查模板（暂存）

秒速五厘米 2024-03-16 22:17 8阅读 0赞

/**
     * 质量检查记录
     *
     * @author wx
     * @date 2023-04-14 19:25:51
     */
    @RestController
    @RequiredArgsConstructor
    @RequestMapping("/inspectrecord" )
    @Api(value = "inspectrecord", tags = "质量检查记录管理")
    public class InspectRecordController {
    
        private final  InspectRecordService inspectRecordService;
    
        /**
         * 分页查询
         * @param dto 质量检查记录
         * @return
         */
        @ApiOperation(value = "分页查询", notes = "分页查询")
        @PostMapping("/getPage" )
    //    @PreAuthorize("@pms.hasPermission('quality_inspectrecord_view')" )
        public R<Page<InspectRecordListVO>> getPage(@RequestBody InspectRecordListDTO dto){
            return R.ok(inspectRecordService.getPage(dto));
        }
    
        /**
         * 详情查询
         * @param dto 质量检查记录
         * @return
         */
        @ApiOperation(value = "详情查询", notes = "详情查询")
        @PostMapping("/getInfo" )
    //    @PreAuthorize("@pms.hasPermission('quality_inspectrecord_view')" )
        public R<InspectRecordDetailVO> getInfo(@RequestBody InspectRecordDetailDTO dto){
            return R.ok(inspectRecordService.getInfo(dto));
        }
    
        /**
         * 任务清单内巡检单详情查询
         * @param dto 质量检查记录
         * @return
         */
        @ApiOperation(value = "任务清单内巡检单详情查询", notes = "详情查询")
        @PostMapping("/getPlanInspectRecordInfo" )
    //    @PreAuthorize("@pms.hasPermission('quality_inspectrecord_view')" )
        public R<InspectRecordDetailVO> getPlanInspectRecordInfo(@RequestBody InspectRecordDetailDTO dto){
            return R.ok(inspectRecordService.getPlanInspectRecordInfo(dto));
        }
    
        /**
         * 多表新增
         * @param dto 质量检查记录
         * @return
         */
        @ApiOperation(value = "新增质量检查记录", notes = "新增质量检查记录")
        @PostMapping("/save")
    //    @PreAuthorize("@pms.hasPermission('quality_inspectrecord_add')" )
        public R save(@Validated(SaveGroup.class) @RequestBody InspectRecordDetailDTO dto) {
            return R.ok(inspectRecordService.save(dto));
        }
        /**
         * 多表更新
         * @param dto 质量检查记录
         * @return
         */
        @ApiOperation(value = "修改质量检查记录", notes = "修改质量检查记录")
        @PostMapping("/update")
    //    @PreAuthorize("@pms.hasPermission('quality_inspectrecord_edit')" )
        public R update(@Validated(SaveGroup.class) @RequestBody InspectRecordDetailDTO dto) {
            inspectRecordService.update(dto);
            return R.ok("success");
        }
        @ApiOperation(value = "删除质量检查记录", notes = "删除质量检查记录")
        @PostMapping("/delete")
    //    @PreAuthorize("@pms.hasPermission('quality_inspectrecord_del')" )
        public R delete( @RequestBody InspectRecordDetailDTO dto) {
            inspectRecordService.deleteMultiTables(dto);
            return R.ok("success");
        }
        @ApiOperation(value = "通过ids批量删除质量检查记录", notes = "通过ids批量删除质量检查记录")
        @PostMapping("/deleteBatch")
    //    @PreAuthorize("@pms.hasPermission('quality_inspectrecord_deleteBatch')" )
        public R deleteBatch(@RequestParam(value = "ids") Long[] ids) {
            inspectRecordService.deleteBatch(ids);
            return R.ok("success");
        }
    
        /**
         * 暂存-新增
         * @param dto 质量检查记录
         * @return
         */
        @ApiOperation(value = "暂存新增质量检查记录", notes = "暂存新增质量检查记录")
        @PostMapping("/saveHold")
    //    @PreAuthorize("@pms.hasPermission('quality_inspectrecord_saveHold')" )
        public R saveHold(@Validated(SaveGroup.class) @RequestBody InspectRecordDetailDTO dto) {
            inspectRecordService.saveHold(dto);
    
            return R.ok("success");
        }
        /**
         * 暂存-更新
         * @param dto 质量检查记录
         * @return
         */
        @ApiOperation(value = "暂存修改质量检查记录", notes = "暂存修改质量检查记录")
        @PostMapping("/updateHold")
    //    @PreAuthorize("@pms.hasPermission('quality_inspectrecord_updateHold')" )
        public R updateHold(@Validated(SaveGroup.class) @RequestBody InspectRecordDetailDTO dto) {
            inspectRecordService.updateHold(dto);
            return R.ok("success");
        }
    }

/**
     * 质量检查记录
     *
     * @author wx
     * @date 2023-04-14 19:25:51
     */
    @Slf4j
    @Service
    public class InspectRecordServiceImpl implements InspectRecordService {
    
        @Resource
        private InspectRecordMapper inspectRecordMapper;
    
        @Resource
        private QualityInspectPlanInfoMapper qualityInspectPlanInfoMapper;
    
        @Resource
        private QualityInspectRecordMapper qualityInspectRecordMapper;
    
        @Resource
        private QualityDutyAreaMapper qualityDutyAreaMapper;
    
        @Resource
        private QualityRectifyOrderMapper qualityRectifyOrderMapper;
    
        @Resource
        private QualityRectifyItemMapper qualityRectifyItemMapper;
    
        @Resource
        private RemoteFileService remoteFileService;
    
        @Override
        public Page<InspectRecordListVO> getPage(InspectRecordListDTO dto){
            if (ObjectUtil.isNotEmpty(dto.getWriteTimeEnd())) {
                dto.setWriteTimeEnd(dto.getWriteTimeEnd().plusDays(1L));
            }
            Page<InspectRecordListVO> page = inspectRecordMapper.getPage(dto);
            List<InspectRecordListVO> records = page.getRecords();
            if (CollUtil.isNotEmpty(records)) {
                records.stream().forEach(r -> {
                    if (ObjectUtil.isNotEmpty(r.getPhotoUrl())) {
                        r.setPhotoUrlJSONArray(JSONUtil.parseArray(r.getPhotoUrl()));
                    }
                });
                // 过滤任务清单内临时保存的巡检单数据
                records = records.stream().filter(p ->   !(p.getInspectPlanId() != null &&  p.getHoldStatus().equals(HoldStatusEnum.HOLDING.getStyle()))).collect(Collectors.toList());
                page.setRecords(records);
                page.setTotal(records.size());
            }
            return page;
        }
        @Override
        public List<InspectRecordListVO> getList(InspectRecordListDTO dto){
            List<InspectRecordListVO> list = inspectRecordMapper.getList(dto);
            return list;
        }
    
    
        /**
        * 详情
        * @param dto 入参
        * @return
        */
        @SneakyThrows
        @Override
        public InspectRecordDetailVO getInfo(InspectRecordDetailDTO dto){
            if (Objects.isNull(dto)) {
                new IllegalStateException("参数异常");
            }
            Long mainColumn = dto.getId();
            if(Objects.isNull(mainColumn)){
                new IllegalStateException("参数异常");
            }
            InspectRecordDetailVO mainTableVO = new InspectRecordDetailVO();
            QualityInspectRecord mainTableEntity = qualityInspectRecordMapper.selectOne(Wrappers.<QualityInspectRecord>lambdaQuery().eq(QualityInspectRecord::getId,mainColumn));
            if(Objects.isNull(mainTableEntity)) return null;
            BeanUtil.copyProperties(mainTableEntity,mainTableVO);
            if (ObjectUtil.isNotEmpty(mainTableVO.getPhotoUrl())) {
                mainTableVO.setPhotoUrlJSONArray(JSONUtil.parseArray(mainTableVO.getPhotoUrl()));
            }
            QualityDutyArea qualityDutyArea = qualityDutyAreaMapper.selectOne(Wrappers.<QualityDutyArea>lambdaQuery().eq(QualityDutyArea::getDutyAreaNum,mainTableEntity.getDutyAreaNum() ));
            if (Objects.nonNull(qualityDutyArea)){
                //设置责任区域名称
                mainTableVO.setDutyAreaName(qualityDutyArea.getName());
            }
            return mainTableVO;
        }
    
    
        /**
         * 任务清单内巡检单详情查询
         * @param dto 入参
         * @return
         */
        @SneakyThrows
        @Override
        public InspectRecordDetailVO getPlanInspectRecordInfo(InspectRecordDetailDTO dto){
            if (Objects.isNull(dto)) {
                new IllegalStateException("参数异常");
            }
            if(Objects.isNull(dto.getInspectPlanInfoId()) || Objects.isNull(dto.getInspectPlanNum()) || Objects.isNull(dto.getInspectPlanId())){
                new IllegalStateException("参数异常");
            }
            InspectRecordDetailVO mainTableVO = new InspectRecordDetailVO();
            QualityInspectRecord mainTableEntity = qualityInspectRecordMapper.selectOne(Wrappers.<QualityInspectRecord>lambdaQuery()
                    .eq(QualityInspectRecord::getInspectPlanId,dto.getInspectPlanId())
                    .eq(QualityInspectRecord::getInspectPlanNum,dto.getInspectPlanNum())
                    .eq(QualityInspectRecord::getInspectPlanInfoId,dto.getInspectPlanInfoId()));
            if(Objects.isNull(mainTableEntity)) return null;
            BeanUtil.copyProperties(mainTableEntity,mainTableVO);
            if (ObjectUtil.isNotEmpty(mainTableVO.getPhotoUrl())) {
                mainTableVO.setPhotoUrlJSONArray(JSONUtil.parseArray(mainTableVO.getPhotoUrl()));
            }
            QualityDutyArea qualityDutyArea = qualityDutyAreaMapper.selectOne(Wrappers.<QualityDutyArea>lambdaQuery().eq(QualityDutyArea::getDutyAreaNum,mainTableEntity.getDutyAreaNum() ));
            if (!Objects.isNull(qualityDutyArea)){
                //设置责任区域名称
                mainTableVO.setDutyAreaName(qualityDutyArea.getName());
            }
            return mainTableVO;
        }
    
        /**
        * 保存
        * @param dto
        */
        @SneakyThrows
        @Transactional(rollbackFor = Exception.class)
        @Override
        public InspectRecordDetailVO save(InspectRecordDetailDTO dto) {
            if (Objects.isNull(dto)) {
                throw new IllegalStateException("参数异常");
            }
            if(Objects.nonNull(dto.getInspectPlanInfoId()) && Objects.nonNull(dto.getInspectPlanNum()) && Objects.nonNull(dto.getInspectPlanId())){
                // 判断当前任务是否存在整改单
                this.checkIfRectifyOrderExist(dto);
            }
            // 创建主表
            QualityInspectRecord mainEntity = new QualityInspectRecord();
            BeanUtil.copyProperties(dto,mainEntity);
            // 制单人
            BuildUser user = SecurityUtils.getUser();
            mainEntity.setWriteUser(StrUtil.toString(user.getId()));
            // 制单时间
            mainEntity.setWriteTime(LocalDateTime.now());
            // 主表暂存状态为已提交
            mainEntity.setHoldStatus(HoldStatusEnum.SUBMIT.getStyle());
            // 检查照片url
            mainEntity.setPhotoUrl(JSONUtil.toJsonStr(dto.getPhotoUrl()));
            // 主表保存
            qualityInspectRecordMapper.insert(mainEntity);
            // 附件表绑定业务id
            if (CollectionUtil.isNotEmpty(dto.getFileIds())){
                SysFileBindDTO sysFileBindDTO = new SysFileBindDTO();
                sysFileBindDTO.setFileIds(dto.getFileIds());
                sysFileBindDTO.setRelationId(String.valueOf(mainEntity.getId()));
                remoteFileService.bindFile(sysFileBindDTO);
            }
            // 更新计划任务为已检查
            if (ObjectUtil.isNotEmpty(dto.getInspectPlanInfoId())) {
              QualityInspectPlanInfo info = qualityInspectPlanInfoMapper.selectById(dto.getInspectPlanInfoId());
                if (ObjectUtil.isNotEmpty(info)) {
                    info.setInspectionFlag(true);
                    qualityInspectPlanInfoMapper.updateById(info);
                }
            }
            InspectRecordDetailDTO query = new InspectRecordDetailDTO();
            query.setId(mainEntity.getId());
            return this.getInfo(query);
        }
        /**
        * 更新
        * @param dto
        */
        @Transactional(rollbackFor = Exception.class)
        @SneakyThrows
        @Override
        public void update(InspectRecordDetailDTO dto) {
            if (Objects.isNull(dto)) {
                throw new IllegalStateException("参数异常");
            }
            if(Objects.nonNull(dto.getInspectPlanInfoId()) && Objects.nonNull(dto.getInspectPlanNum()) && Objects.nonNull(dto.getInspectPlanId())){
                // 判断当前任务是否存在整改单
                this.checkIfRectifyOrderExist(dto);
            }
            QualityInspectRecord mainEntity = new QualityInspectRecord();
            BeanUtil.copyProperties(dto,mainEntity);
            // 检查照片url
            mainEntity.setPhotoUrl(JSONUtil.toJsonStr(dto.getPhotoUrl()));
            // 制单人
            BuildUser user = SecurityUtils.getUser();
            mainEntity.setWriteUser(StrUtil.toString(user.getId()));
            // 制单时间
            mainEntity.setWriteTime(LocalDateTime.now());
            // 主表暂存状态为已提交
            mainEntity.setHoldStatus(HoldStatusEnum.SUBMIT.getStyle());
            if (Objects.nonNull(mainEntity.getId())) {
                qualityInspectRecordMapper.updateById(mainEntity);
            } else {
                qualityInspectRecordMapper.insert(mainEntity);
            }
            // 附件表绑定业务id
            if (CollectionUtil.isNotEmpty(dto.getFileIds())){
                SysFileBindDTO sysFileBindDTO = new SysFileBindDTO();
                sysFileBindDTO.setFileIds(dto.getFileIds());
                sysFileBindDTO.setRelationId(String.valueOf(mainEntity.getId()));
                remoteFileService.bindFile(sysFileBindDTO);
            }
            // 更新计划任务为已检查
            if (ObjectUtil.isNotEmpty(dto.getInspectPlanInfoId())) {
                QualityInspectPlanInfo info = qualityInspectPlanInfoMapper.selectById(dto.getInspectPlanInfoId());
                if (ObjectUtil.isNotEmpty(info)) {
                    info.setInspectionFlag(true);
                    qualityInspectPlanInfoMapper.updateById(info);
                }
            }
        }
        /**
         * 删除
         * @param dto
         */
        @Transactional(rollbackFor = Exception.class)
        @SneakyThrows
        @Override
        public void deleteMultiTables(InspectRecordDetailDTO dto) {
            if (Objects.isNull(dto.getId()))
                throw new IllegalStateException("参数异常");
            QualityInspectRecord mainTableEntity = qualityInspectRecordMapper.selectOne(Wrappers.<QualityInspectRecord>lambdaQuery().eq(QualityInspectRecord::getId,dto.getId()));
            if (Objects.isNull(mainTableEntity))
                throw new IllegalStateException("参数异常");
            //删除主表数据
            qualityInspectRecordMapper.deleteById(mainTableEntity.getId());
        }
        /**
         * 批量删除
         * @param ids 主键数组
         */
        @Transactional(rollbackFor = Exception.class)
        @SneakyThrows
        @Override
        public void deleteBatch(Long[] ids){
            if (ArrayUtils.isEmpty(ids))
                throw new IllegalStateException("参数异常");
            for (Long id : ids) {
                //查询主表数据
                QualityInspectRecord mainTableEntity = qualityInspectRecordMapper.selectById(id);
                if (Objects.isNull(mainTableEntity))
                    throw new IllegalStateException("参数异常");
                //删除主表数据
                if (Objects.nonNull(mainTableEntity.getId())) {
                    qualityInspectRecordMapper.deleteById(mainTableEntity.getId());
                }
            }
        }
    
        /**
         * 暂存-更新
         */
        @Transactional(rollbackFor = Exception.class)
        @SneakyThrows
        @Override
        public void updateHold(InspectRecordDetailDTO dto) {
            if (Objects.isNull(dto)) {
                throw new IllegalStateException("参数异常");
            }
            if(Objects.nonNull(dto.getInspectPlanInfoId()) && Objects.nonNull(dto.getInspectPlanNum()) && Objects.nonNull(dto.getInspectPlanId())){
                // 判断当前任务是否存在整改单
                this.checkIfRectifyOrderExist(dto);
            }
            QualityInspectRecord mainEntity = new QualityInspectRecord();
            BeanUtil.copyProperties(dto,mainEntity);
            mainEntity.setHoldStatus(HoldStatusEnum.HOLDING.getStyle());
            mainEntity.setPhotoUrl(JSONUtil.toJsonStr(dto.getPhotoUrl()));
            if (Objects.nonNull(mainEntity.getId())) {
                qualityInspectRecordMapper.updateById(mainEntity);
            } else {
                BuildUser user = SecurityUtils.getUser();
                mainEntity.setWriteUser(StrUtil.toString(user.getId()));
                mainEntity.setWriteTime(LocalDateTime.now());
                qualityInspectRecordMapper.insert(mainEntity);
            }
            // 附件表绑定业务id
            if (CollectionUtil.isNotEmpty(dto.getFileIds())){
                SysFileBindDTO sysFileBindDTO = new SysFileBindDTO();
                sysFileBindDTO.setFileIds(dto.getFileIds());
                sysFileBindDTO.setRelationId(String.valueOf(mainEntity.getId()));
                remoteFileService.bindFile(sysFileBindDTO);
            }
        }
    
        /**
         * 暂存-保存
         */
        @SneakyThrows
        @Transactional(rollbackFor = Exception.class)
        @Override
        public void saveHold(InspectRecordDetailDTO dto) {
            if (Objects.nonNull(dto.getInspectPlanInfoId()) && Objects.nonNull(dto.getInspectPlanNum()) && Objects.nonNull(dto.getInspectPlanId())) {
                // 判断当前任务是否存在整改单
                this.checkIfRectifyOrderExist(dto);
            }
            QualityInspectRecord mainEntity = new QualityInspectRecord();
            //主表暂存状态
            mainEntity.setHoldStatus(HoldStatusEnum.HOLDING.getStyle());
            BeanUtil.copyProperties(dto, mainEntity);
            BuildUser user = SecurityUtils.getUser();
            mainEntity.setWriteUser(StrUtil.toString(user.getId()));
            mainEntity.setWriteTime(LocalDateTime.now());
            mainEntity.setPhotoUrl(JSONUtil.toJsonStr(dto.getPhotoUrl()));
            qualityInspectRecordMapper.insert(mainEntity);
            // 附件表绑定业务id
            if (CollectionUtil.isNotEmpty(dto.getFileIds())){
                SysFileBindDTO sysFileBindDTO = new SysFileBindDTO();
                sysFileBindDTO.setFileIds(dto.getFileIds());
                sysFileBindDTO.setRelationId(String.valueOf(mainEntity.getId()));
                remoteFileService.bindFile(sysFileBindDTO);
            }
        }
    
        /**
         * 判断当前任务是否存在整改单
         *
         * @param dto
         */
        public void checkIfRectifyOrderExist(InspectRecordDetailDTO dto) {
            // 根据计划查询整改单
            QualityRectifyOrder qalityRectifyOrder = qualityRectifyOrderMapper.selectOne(Wrappers.<QualityRectifyOrder>lambdaQuery().eq(QualityRectifyOrder::getInspectPlanId, dto.getInspectPlanId()).eq(QualityRectifyOrder::getInspectPlanNum, dto.getInspectPlanNum()));
            if (Objects.nonNull(qalityRectifyOrder)) {
                // 查询整改项
                QualityRectifyItem qualityRectifyItem = qualityRectifyItemMapper.selectOne(Wrappers.<QualityRectifyItem>lambdaQuery().eq(QualityRectifyItem::getInspectPlanInfoId, dto.getInspectPlanInfoId()));
                if (Objects.nonNull(qualityRectifyItem)) {
                    // 若为暂存状态直接删除，反之抛出异常
                    if (qalityRectifyOrder.getRectifyStatus().equals(HoldStatusEnum.HOLDING.getStyle())) {
                        // 删除整改单
                        qualityRectifyOrderMapper.deleteById(qalityRectifyOrder.getId());
                        // 删除整改项
                        qualityRectifyItemMapper.deleteById(qualityRectifyItem.getId());
                    } else {
                        throw new IllegalStateException("当前任务已存在提交的整改单");
                    }
                }
            }
        }
    
    }