主表与多个子表的保存与更新

妖狐艹你老母 2024-03-16 12:50 11阅读 0赞

/**
     * 质量检查计划
     *
     * @author build code generator
     * @date 2023-04-14 14:56:41
     */
    @Slf4j
    @Service
    public class InspectPlanServiceImpl implements InspectPlanService {
    
        @Resource
        private InspectPlanMapper inspectPlanMapper;
    
        @Resource
        private QualityInspectPlanMapper qualityInspectPlanMapper;
        @Resource
        private QualityInspectPlanInfoMapper qualityInspectPlanInfoMapper;
        @Override
        public Page<InspectPlanListVO> getPage(InspectPlanListDTO dto){
            //return inspectPlanMapper.getPage(dto);
            if (ObjectUtil.isNotEmpty(dto.getWriteTimeEnd())) {
                dto.setWriteTimeEnd(dto.getWriteTimeEnd().plusDays(1L));
            }
            if (ObjectUtil.isNotEmpty(dto.getPlanDateEnd())) {
                dto.setPlanDateEnd(dto.getPlanDateEnd().plusDays(1L));
            }
            Page<InspectPlanListVO> page = inspectPlanMapper.getPage(dto);
            return page;
        }
        @Override
        public List<InspectPlanListVO> getList(InspectPlanListDTO dto){
            List<InspectPlanListVO> list = inspectPlanMapper.getList(dto);
            return list;
        }
    
    
        /**
        * 详情
        * @param dto 入参
        * @return
        */
        @SneakyThrows
        @Override
        public InspectPlanDetailVO getInfo(InspectPlanDetailDTO dto){
            if (Objects.isNull(dto)) {
                new IllegalStateException("参数异常");
            }
            Long mainColumn = dto.getId();
            if(Objects.isNull(mainColumn)){
                new IllegalStateException("参数异常");
            }
            InspectPlanDetailVO mainTableVO = new InspectPlanDetailVO();
            QualityInspectPlan mainTableEntity = qualityInspectPlanMapper.selectOne(Wrappers.<QualityInspectPlan>lambdaQuery().eq(QualityInspectPlan::getId,mainColumn));
            if(Objects.isNull(mainTableEntity)) return null;
            BeanUtil.copyProperties(mainTableEntity,mainTableVO);
            List<QualityInspectPlanInfo> qualityInspectPlanInfo = qualityInspectPlanInfoMapper.selectList(Wrappers.<QualityInspectPlanInfo>lambdaQuery().eq(QualityInspectPlanInfo::getInspectPlanId,mainTableEntity.getId() ).eq(QualityInspectPlanInfo::getInspectPlanNum,mainTableEntity.getInspectPlanNum() ));
            List<QualityInspectPlanInfoVO> qualityInspectPlanInfoList = new ArrayList<>();
            for (QualityInspectPlanInfo iterm : qualityInspectPlanInfo) {
                QualityInspectPlanInfoVO tempVO = new QualityInspectPlanInfoVO();
                BeanUtil.copyProperties(iterm,tempVO);
                qualityInspectPlanInfoList.add(tempVO);
            }
            mainTableVO.setQualityInspectPlanInfo(qualityInspectPlanInfoList);
            return mainTableVO;
        }
        /**
        * 保存
        * @param dto
        */
        @SneakyThrows
        @Transactional(rollbackFor = Exception.class)
        @Override
        public InspectPlanDetailVO save(InspectPlanDetailDTO dto) {
            if (Objects.isNull(dto)) {
                throw new IllegalStateException("参数异常");
            }
            LambdaQueryWrapper<QualityInspectPlan> queryWrapper = Wrappers.<QualityInspectPlan>lambdaQuery();
            queryWrapper.eq(QualityInspectPlan::getPlanDate,dto.getPlanDate());
            if (qualityInspectPlanMapper.exists(queryWrapper)) {
                Assert.isTrue(false, dto.getPlanDate() + "已有计划，请重新编辑");
            }
            //创建主表
            QualityInspectPlan mainEntity = new QualityInspectPlan();
            BeanUtil.copyProperties(dto,mainEntity);
            // 制单人
            BuildUser user = SecurityUtils.getUser();
            mainEntity.setWriteUser(StrUtil.toString(user.getId()));
            // 制单时间
            mainEntity.setWriteTime(LocalDateTime.now());
            //主表保存
            qualityInspectPlanMapper.insert(mainEntity);
            List<QualityInspectPlanInfo> qualityInspectPlanInfoList = new ArrayList<>();
    
            List<QualityInspectPlanInfoDTO> qualityInspectPlanInfoDTOList = dto.getQualityInspectPlanInfo();
            if (CollectionUtil.isNotEmpty(qualityInspectPlanInfoDTOList)) {
                for (QualityInspectPlanInfoDTO tempDTO : qualityInspectPlanInfoDTOList) {
                    QualityInspectPlanInfo temp = new QualityInspectPlanInfo();
                    BeanUtil.copyProperties( tempDTO,temp);
                    //关联字段
                    temp.setInspectPlanId(mainEntity.getId());
                    //关联字段
                    temp.setInspectPlanNum(mainEntity.getInspectPlanNum());
                    qualityInspectPlanInfoList.add(temp);
                }
            }
            if (CollectionUtil.isNotEmpty(qualityInspectPlanInfoList)&&qualityInspectPlanInfoList.size()>0) {
                for (QualityInspectPlanInfo temp : qualityInspectPlanInfoList) {
                     qualityInspectPlanInfoMapper.insert(temp);
                }
            }
            InspectPlanDetailDTO query = new InspectPlanDetailDTO();
            query.setId(mainEntity.getId());
            return this.getInfo(query);
        }
        /**
        * 更新
        * @param dto
        */
        @Transactional(rollbackFor = Exception.class)
        @SneakyThrows
        @Override
        public void update(InspectPlanDetailDTO dto) {
            if (Objects.isNull(dto)) {
                throw new IllegalStateException("参数异常");
            }
            QualityInspectPlan mainEntity = new QualityInspectPlan();
            BeanUtil.copyProperties(dto,mainEntity);
            if (Objects.nonNull(mainEntity.getId())) {
                qualityInspectPlanMapper.updateById(mainEntity);
            } else {
                qualityInspectPlanMapper.insert(mainEntity);
            }
            List<QualityInspectPlanInfoDTO> qualityInspectPlanInfoDTO = dto.getQualityInspectPlanInfo();
            qualityInspectPlanInfoMapper.delete(Wrappers.<QualityInspectPlanInfo>lambdaQuery().eq(QualityInspectPlanInfo::getInspectPlanId,mainEntity.getId()).eq(QualityInspectPlanInfo::getInspectPlanNum,mainEntity.getInspectPlanNum()));
            if(CollectionUtil.isNotEmpty(qualityInspectPlanInfoDTO)){
                for (QualityInspectPlanInfoDTO tempDTO : qualityInspectPlanInfoDTO) {
                    QualityInspectPlanInfo entity = new QualityInspectPlanInfo();
                    BeanUtil.copyProperties(tempDTO,entity);
                    entity.setInspectPlanId(mainEntity.getId());
                    entity.setInspectPlanNum(mainEntity.getInspectPlanNum());
                    qualityInspectPlanInfoMapper.insert(entity);
                }
            }
        }
        /**
         * 删除
         * @param dto
         */
        @Transactional(rollbackFor = Exception.class)
        @SneakyThrows
        @Override
        public void deleteMultiTables(InspectPlanDetailDTO dto) {
            if (Objects.isNull(dto.getId()))
                throw new IllegalStateException("参数异常");
            QualityInspectPlan mainTableEntity = qualityInspectPlanMapper.selectOne(Wrappers.<QualityInspectPlan>lambdaQuery().eq(QualityInspectPlan::getId,dto.getId()));
            if (Objects.isNull(mainTableEntity))
                throw new IllegalStateException("参数异常");
            List<QualityInspectPlanInfo> qualityInspectPlanInfo = qualityInspectPlanInfoMapper.selectList(Wrappers.<QualityInspectPlanInfo>lambdaQuery().eq(QualityInspectPlanInfo::getInspectPlanId,mainTableEntity.getId() ).eq(QualityInspectPlanInfo::getInspectPlanNum,mainTableEntity.getInspectPlanNum() ));
            Assert.isFalse(CollectionUtil.isNotEmpty(qualityInspectPlanInfo),"该日期有计划任务，不可删除。");
            //删除主表数据
            qualityInspectPlanMapper.deleteById(mainTableEntity.getId());
            if(StringUtils.isNotBlank( mainTableEntity.getInspectPlanNum())){
                qualityInspectPlanInfoMapper.delete(Wrappers.<QualityInspectPlanInfo>lambdaQuery().eq(QualityInspectPlanInfo::getInspectPlanId,mainTableEntity.getId() ).eq(QualityInspectPlanInfo::getInspectPlanNum,mainTableEntity.getInspectPlanNum() ));
            }
        }
        /**
         * 批量删除
         * @param ids 主键数组
         */
        @Transactional(rollbackFor = Exception.class)
        @SneakyThrows
        @Override
        public void deleteBatch(Long[] ids){
            if (ArrayUtils.isEmpty(ids))
                throw new IllegalStateException("参数异常");
            for (Long id : ids) {
                //查询主表数据
                QualityInspectPlan mainTableEntity = qualityInspectPlanMapper.selectById(id);
                if (Objects.isNull(mainTableEntity))
                    throw new IllegalStateException("参数异常");
                //删除主表数据
                if (Objects.nonNull(mainTableEntity.getId())) {
                    qualityInspectPlanMapper.deleteById(mainTableEntity.getId());
                }
                if( StringUtils.isNotBlank(mainTableEntity.getInspectPlanNum())){
                    qualityInspectPlanInfoMapper.delete(Wrappers.<QualityInspectPlanInfo>lambdaQuery().eq(QualityInspectPlanInfo::getInspectPlanId,mainTableEntity.getId() ).eq(QualityInspectPlanInfo::getInspectPlanNum,mainTableEntity.getInspectPlanNum() ));
                }
            }
        }
    
    }

下面是DTO层：

@Data
    @ApiModel(value = "quality_inspect_plan传输对象")
    public class InspectPlanDetailDTO extends BasePage {
    
        /**
         * 主键
         */
        @NotNull(groups = {UpdateGroup.class, HoldUpdateGroup.class}, message = "主键不能为空")
        @ApiModelProperty(value = "主键")
        private Long id;
        /**
         * 编码
         */
        @ApiModelProperty(value = "编码")
        private String inspectPlanNum;
        /**
         * 项目编码
         */
        @ApiModelProperty(value = "项目编码")
        private String projectNum;
        /**
         * 所属部门编号
         */
        @ApiModelProperty(value = "所属部门编号")
        private String deptNum;
        /**
         * 计划日期
         */
        @JsonFormat(pattern = "yyyy-MM-dd", timezone="GMT+8")
    //    @NotNull(groups ={UpdateGroup.class, SaveGroup.class, HoldGroup.class, HoldUpdateGroup.class}, message = "计划日期不能为空")
        @ApiModelProperty(value = "计划日期")
        private LocalDate planDate;
        /**
         * 制单人
         */
        @ApiModelProperty(value = "制单人")
        private String writeUser;
        /**
         * 制单部门编号
         */
        @ApiModelProperty(value = "制单部门编号")
        private String writeDeptNum;
        /**
         * 制单时间
         */
        @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone="GMT+8")
        @ApiModelProperty(value = "制单时间")
        private LocalDateTime writeTime;
        /**
         * 审批状态;01-未提交,02-运行中,03-流程正常结束,05-流程异常
         */
        @ApiModelProperty(value = "审批状态;01-未提交,02-运行中,03-流程正常结束,05-流程异常")
        private String applicationStatus;
        /**
         * 暂存状态;0-暂存,1-已保存
         */
        @ApiModelProperty(value = "暂存状态;0-暂存,1-已保存")
        private String holdStatus;
        /**
         * 是否启用;1-启用,0-未启用
         */
        @ApiModelProperty(value = "是否启用;1-启用,0-未启用")
        private Boolean enable;
        /**
         * 删除标识;0-未删除,1-已删除
         */
        @ApiModelProperty(value = "删除标识;0-未删除,1-已删除")
        private String delFlag;
        /**
         * 版本号
         */
        @ApiModelProperty(value = "版本号")
        private Integer version;
        /**
         * 创建人编码
         */
        @ApiModelProperty(value = "创建人编码")
        private String createBy;
        @ApiModelProperty(value = "创建人编码Name")
        private String createByName;
    
        /**
         * 创建时间
         */
        @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone="GMT+8")
        @ApiModelProperty(value = "创建时间")
        private LocalDateTime createTime;
        /**
         * 更新人编码
         */
        @ApiModelProperty(value = "更新人编码")
        private String updateBy;
        @ApiModelProperty(value = "更新人编码Name")
        private String updateByName;
    
        /**
         * 更新时间
         */
        @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone="GMT+8")
        @ApiModelProperty(value = "更新时间")
        private LocalDateTime updateTime;
        /**
         * 租户
         */
        @ApiModelProperty(value = "租户")
        private Long tenantId;
        /**
         * quality_inspect_plan_info
         */
        @ApiModelProperty(value = "quality_inspect_plan_info")
        private List<QualityInspectPlanInfoDTO> qualityInspectPlanInfo;
    }
    
    
    
    @Data
    @ApiModel(value = "quality_inspect_plan_info传输对象")
    public class QualityInspectPlanInfoDTO extends BasePage {
    
        /**
         * 主键
         */
        @NotNull(groups = {UpdateGroup.class, HoldUpdateGroup.class}, message = "主键不能为空")
        @ApiModelProperty(value = "主键")
        private Long id;
        /**
         * 检查计划id
         */
        @ApiModelProperty(value = "检查计划id")
        private Long inspectPlanId;
        /**
         * 检查计划编码
         */
        @ApiModelProperty(value = "检查计划编码")
        private String inspectPlanNum;
    
        /**
         * 检查内容id
         */
        @ApiModelProperty(value="检查内容id")
        private Long inspectInfoId;
    
        /**
         * 计划日期
         */
        @JsonFormat(pattern = "yyyy-MM-dd", timezone="GMT+8")
        @ApiModelProperty(value = "计划日期")
        private LocalDate planDate;
    
        /**
         * 推送标识;0-手动,1-自动
         */
        @ApiModelProperty(value = "推送标识;0-手动,1-自动")
        private Boolean pushFlag;
        /**
         * wbs分项id
         */
        @ApiModelProperty(value = "wbs分项id")
        private Long wbsId;
        /**
         * wbs编号
         */
        @ApiModelProperty(value = "wbs编号")
        private String wbsNum;
        /**
         * 工序id
         */
        @ApiModelProperty(value = "工序id")
        private Long processId;
        /**
         * 控制标准
         */
        @ApiModelProperty(value = "控制标准")
        private String controlStandard;
        /**
         * 质量通病
         */
        @ApiModelProperty(value = "质量通病")
        private String commonProblem;
        /**
         * 问题等级
         */
        @ApiModelProperty(value = "问题等级")
        private String problemLevel;
        /**
         * 整改时限(天)
         */
        @ApiModelProperty(value = "整改时限(天)")
        private Integer rectifyLimit;
        /**
         * 控制项
         */
        @ApiModelProperty(value = "控制项")
        private String controlItem;
        /**
         * 检测方法
         */
        @ApiModelProperty(value = "检测方法")
        private String checkMethod;
        /**
         * 检测频率
         */
        @ApiModelProperty(value = "检测频率")
        private String checkFrequency;
        /**
         * 频次
         */
        @ApiModelProperty(value = "频次")
        private String frequency;
        /**
         * 检验批
         */
        @ApiModelProperty(value = "检验批")
        private String inspectionLot;
        /**
         * 检查标识;0-未检查,1-已检查
         */
        @ApiModelProperty(value = "检查标识;0-未检查,1-已检查")
        private Boolean inspectionFlag;
        /**
         * 未检查原因
         */
        @ApiModelProperty(value = "未检查原因")
        private String uninspectedReason;
        /**
         * 责任人
         */
        @ApiModelProperty(value = "责任人")
        private String principal;
        /**
         * 审批状态;01-未提交,02-运行中,03-流程正常结束,05-流程异常
         */
        @ApiModelProperty(value = "审批状态;01-未提交,02-运行中,03-流程正常结束,05-流程异常")
        private String applicationStatus;
        /**
         * 暂存状态;0-暂存,1-已保存
         */
        @ApiModelProperty(value = "暂存状态;0-暂存,1-已保存")
        private String holdStatus;
        /**
         * 是否启用;1-启用,0-未启用
         */
        @ApiModelProperty(value = "是否启用;1-启用,0-未启用")
        private Boolean enable;
        /**
         * 删除标识;0-未删除,1-已删除
         */
        @ApiModelProperty(value = "删除标识;0-未删除,1-已删除")
        private String delFlag;
        /**
         * 版本号
         */
        @ApiModelProperty(value = "版本号")
        private Integer version;
        /**
         * 创建人编码
         */
        @ApiModelProperty(value = "创建人编码")
        private String createBy;
        @ApiModelProperty(value = "创建人编码Name")
        private String createByName;
    
        /**
         * 创建时间
         */
        @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone="GMT+8")
        @ApiModelProperty(value = "创建时间")
        private LocalDateTime createTime;
        /**
         * 更新人编码
         */
        @ApiModelProperty(value = "更新人编码")
        private String updateBy;
        @ApiModelProperty(value = "更新人编码Name")
        private String updateByName;
    
        /**
         * 更新时间
         */
        @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone="GMT+8")
        @ApiModelProperty(value = "更新时间")
        private LocalDateTime updateTime;
        /**
         * 租户
         */
        @ApiModelProperty(value = "租户")
        private Long tenantId;
    }