批量导入文件

public Rest importlabelData(MultipartFile file){
        return songService.importlabelData(file);
    }



@Override
    public Rest<String> importData(MultipartFile file) {
        Assert.notNull(file, "请选择文件");
        InputStream inputStream = null;
        StringBuilder builder = new StringBuilder();
        ArrayList<SongExcelModel> addList = new ArrayList<>();
        List<Object> dataList = new ArrayList<>();
        Rest<String> rest = new Rest<>();
        try{
            inputStream = new BufferedInputStream(file.getInputStream());
            dataList = EasyExcelFactory.read(inputStream, new Sheet(1, 1, SongExcelModel.class));
            List<OmsSong> omsSongs = new ArrayList<OmsSong>();
            for(int i = 0; i < dataList.size(); i++){
                Object obj = dataList.get(i);
                if( obj instanceof SongExcelModel){
                    SongExcelModel song = (SongExcelModel) obj;
                    if(StringUtils.isAnyBlank(song.getSongName(), song.getCode(), song.getLabel())){
                        builder.append("第").append(i + 1).append("行必填项存在空值, ");
                        continue;
                    }
                    //验证歌曲code是否存在与系统中,不存在则不导入
                    OmsSong omsSong = songMapper.getByCode(song.getCode());
                    if(Objects.isNull(omsSong.getId())){
                        builder.append("第").append(i + 2).append("行的歌曲code系统中不存在");
                        continue;
                    }
                    String[] split = song.getLabel().split(",");
                    Set<String> labelSet = new HashSet<>();
                    int labelNumber = 0;
                    //验证歌曲标签在标签库中是否存在,最多可设置3个标签
                    for(String str : split){
                        if(labelNumber > 3){
                            break;
                        }
                        SongLabelEnum labelEnum = this.getSongLabelEnum(str.trim());
                        if(Objects.nonNull(labelEnum) && labelSet.add(labelEnum.name())){
                            labelNumber++;
                        }
                    }
                    if(labelNumber > 0){
                        song.setLabel(String.join(",", labelSet));
                    }else {
                        song.setLabel(null);
                    }
                    omsSongs.add(omsSong);
                    addList.add(song);
                }else {
                    builder.append("第").append(i + 2).append("行格式错误, ");
                }
            }
            if(StringUtils.isNotEmpty(builder.toString())) {
                logger.error("数据导入失败,{}条数据有误: {}", dataList.size() - addList.size(), builder.toString());
                rest.setCode(ResultEnum.ERROR.getCode());
                rest.setMessage("数据导入失败: " + builder.substring(0, builder.lastIndexOf(",")));
            }else {
                this.saveImportData(addList);
                this.bulkAddSong(omsSongs);
                rest.setCode(ResultEnum.SUCCESS.getCode());
                rest.setMessage("数据导入成功");
            }
        }catch (Exception e){
            logger.error("歌曲导入异常: {}", e.getMessage());
            rest.setCode(ResultEnum.ERROR.getCode());
            rest.setMessage("导入任务异常");
        }finally {
            if(Objects.nonNull(inputStream)){
                try {
                    inputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return rest;
    }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值