业务场景:根据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();
}
}