关于重用(复制)输入流inputStream

最新推荐文章于 2021-08-06 12:00:24 发布

Ai阴天

最新推荐文章于 2021-08-06 12:00:24 发布

阅读量1k

点赞数 1

文章标签： stream inputStream 复制流阿里云Oss url下载

本文链接：https://blog.csdn.net/weixin_44138309/article/details/102777452

版权

业务场景:根据url拿到输入流(url其实就是个下载链接),拿到输入流后要做两个操作:1.上传阿里云Oss 2.解析输入流读取每一行,转换成实体类存数据库

问题解析:输入流在第一次被读取使用之后,流的数据就消耗完了,所以想要重用的话,需要将流数据存入容器,再从容器取出数据构造流.(这里好多同学容易误解成:流使用完了,会被关闭,其实这里打断点发现的是流使用完不会被关闭,而是里面的数据都被清空了,可以理解成水流,流走了就没有了), 下面上代码:

复制流(传入原始流,返回数据容器):

 /**
     * 复制流
     * @param inputStream
     * @return
     */
    private ByteArrayOutputStream copyStream(InputStream inputStream) {
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        try {
            byte[] buffer = new byte[4096];
            int len;
            while ((len = inputStream.read(buffer)) > -1) {
                baos.write(buffer, 0, len);
            }
            baos.flush();
        } catch (IOException e) {
            log.error("复制流异常：" + e.getMessage());
        }
        return baos;
    }

构造流对象,使用重复的流数据(构造了两个流对象做不同的操作):

try {
                entity.setPlatform_trace(platform_trace);
                inputStream = getInputStreamByUrl(entity.getUrl());
                //先复制流对象并返回,防止程序后面使用到流数据空了的流对象
                ByteArrayOutputStream baos = copyStream(inputStream);
                //构造同样数据的不同流对象
                InputStream inputStreamCopyOss = new ByteArrayInputStream(baos.toByteArray());
                InputStream inputStreamCopyRead = new ByteArrayInputStream(baos.toByteArray());
                //1.开启新线程,保存阿里云
                executor.submit(() ->
                        uploadOss(entity, inputStreamCopyOss));
                //2.保存数据库
                reader = new BufferedReader(new InputStreamReader(inputStreamCopyRead, "UTF-8"));
                String str = null;
                while (true) {
                    //每次读一行
                    str = reader.readLine();
                    ...

保存阿里云:

 /**
     * 保存阿里云
     * @param entity
     * @param inputStream
     */
    private void uploadOss(BwDataListEntity entity, InputStream inputStream) {
        String endpoint = OssConfig.getOssEndpoint();
        String accessKeyId = OssConfig.getOssAccessKeyId();
        String accessKeySecret = OssConfig.getOssAccessKeySecret();
        String bucketName = OssConfig.getBucketName();
        String objectName = "";
        if (BaiWangConfig.getJxfp().equals(entity.getType())) {
            //进项
            objectName = getObjectName("input-invoice-file", entity.getPlatform_trace(), entity.getUrl());
        } else {
            //销项
            objectName = getObjectName("output-invoice-file", entity.getPlatform_trace(), entity.getUrl());
        }
        // 创建OSSClient实例
        OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
        try {
            // 上传文件流
            ossClient.putObject(bucketName, objectName, inputStream);
        } catch (Exception e) {
            log.error("阿里云上传文件失败：" + e.getMessage());
        } finally {
            // 关闭OSSClient。
            ossClient.shutdown();
        }
    }