springboot服务器文件读取工具类

devtools/2024/10/21 6:02:11/

本地文件和网络文件读取

一. SpringBoot的RestTemplate配置

  • RestTemplate

二. 文件读取

  • RangeEntity 分片下载的封装对象
java">package com.zzc.component.download;import com.zzc.common.utils.StrUtils;
import lombok.Data;import javax.servlet.http.HttpServletResponse;public class RangeEntity {private String range;private long contentLength;private String contentRange;private Long firstPos;private Long lastPos;public RangeEntity(String range, long contentLength) {parseRange(range, contentLength);}private void parseRange(String range, long contentLength) {this.range = range;this.contentLength = contentLength;if (StrUtils.isBlank(range) || "null".equalsIgnoreCase(range)) {this.firstPos = 0L;this.contentLength = contentLength;this.contentRange = "bytes " + 0 + "-" + (contentLength - 1) + "/" + contentLength;return;}range = range.trim();String rangeBytes = range.replaceAll("bytes=", "").replaceAll(" ", "");if (rangeBytes.endsWith("-")) {//bytes=99999-rangeBytes = rangeBytes.substring(0, rangeBytes.indexOf("-"));this.firstPos = Long.valueOf(rangeBytes.trim());this.contentLength = contentLength - this.firstPos;this.contentRange = "bytes " + this.firstPos + "-" + (contentLength - 1) + "/" + contentLength;} else {this.firstPos = Long.valueOf(rangeBytes.substring(0, rangeBytes.indexOf("-")));this.lastPos = Long.valueOf(rangeBytes.substring(rangeBytes.indexOf("-") + 1));this.contentLength = this.lastPos - this.firstPos + 1;this.contentRange = "bytes " + this.firstPos + "-" + this.lastPos + "/" + contentLength;}}public int getStatusCode(boolean isIeOrPlayer) {if (((StrUtils.isBlank(range) || "null".equalsIgnoreCase(range)) && isIeOrPlayer)|| (this.firstPos == null && this.lastPos == null)) {return HttpServletResponse.SC_OK;} else {return HttpServletResponse.SC_PARTIAL_CONTENT;}}public String getRange() {return range;}public long getContentLength() {return contentLength;}public String getContentRange() {return contentRange;}public boolean isNotRange() {return lastPos == null || firstPos == 0;}public Long getFirstPos() {return firstPos;}
}
  • HeaderUtils 请求header判断类
java">package com.zzc.component.download;import com.zzc.common.utils.StrUtils;import javax.servlet.http.HttpServletRequest;public class HeaderUtils {public static boolean isIeBrowser(HttpServletRequest request) {String[] isSignals = {"MSIE", "Trident", "Edge", "Safari"};String userAgent = request.getHeader("User-Agent");if (StrUtils.isNotBlank(userAgent)) {for (String signal : isSignals) {if (userAgent.contains(signal)) {return true;}}}return false;}public static boolean isWinPlayer(HttpServletRequest request) {String userAgent = request.getHeader("User-Agent");if (StrUtils.isNotBlank(userAgent)) {return userAgent.contains("Windows-Media-Player");}return false;}
}
  • HeaderConstant请求header对应字段
java">package com.zzc.component.download;public class HeaderConstant {public static final String KEY_ACCEPT_RANGE = "Accept-Ranges";public static final String KEY_CONTENT_LENGTH = "Content-Length";public static final String KEY_CONTENT_RANGE = "Content-Range";public static final String KEY_CONTENT_TYPE = "Content-Type";public static final String KEY_CONTENT_DISPOSITION = "Content-Disposition";
}
java">package com.zzc.component.download;import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;public interface FileComponent {void play(String path, HttpServletRequest request, HttpServletResponse response);void download(String path, HttpServletRequest request, HttpServletResponse response);}
java">package com.zzc.component.download;import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.RandomAccessFile;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.nio.channels.Channels;
import java.nio.channels.FileChannel;
import java.nio.channels.WritableByteChannel;@Slf4j
@Component
public class FileComponentImpl implements FileComponent {@Overridepublic void play(String path, HttpServletRequest request, HttpServletResponse response) {RandomAccessFile raf = null;try {File file = new File(path);if (!file.exists()) {throw new RuntimeException();}long length = file.length();String range = request.getHeader("Range");String filename = file.getName();String contentType = "video/mp4";//文件类型RangeEntity rangeEntity = new RangeEntity(range, length);addHeader(request, response, rangeEntity, contentType, filename);raf = new RandomAccessFile(file, "r");OutputStream outputStream = response.getOutputStream();if (rangeEntity.isNotRange()) {channelCopy(raf.getChannel(), Channels.newChannel(outputStream), rangeEntity.getFirstPos(), length);outputStream.flush();return;}long contentLength = rangeEntity.getContentLength();channelCopy(raf.getChannel(), Channels.newChannel(outputStream), rangeEntity.getFirstPos(), contentLength);outputStream.flush();//outputStream.close();} catch (Exception e) {} finally {if (raf != null) {try {raf.close();} catch (IOException e) {}}}}private static void addHeader(HttpServletRequest request, HttpServletResponse response, RangeEntity rangeEntity, String contentType, String filename) {boolean isIeOrPlayer = HeaderUtils.isWinPlayer(request) || HeaderUtils.isIeBrowser(request);response.setStatus(rangeEntity.getStatusCode(isIeOrPlayer));response.setHeader(HeaderConstant.KEY_ACCEPT_RANGE, "bytes");response.setHeader(HeaderConstant.KEY_CONTENT_LENGTH, rangeEntity.getContentLength() + "");if (rangeEntity.getContentRange() != null) {response.setHeader(HeaderConstant.KEY_CONTENT_RANGE, rangeEntity.getContentRange());}response.setHeader(HeaderConstant.KEY_CONTENT_TYPE, contentType);//response.setHeader(HeaderConstant.KEY_CONTENT_DISPOSITION, "inline; filename=\"" + filename + "\"; filename*=utf-8''" + filename);if (isIeOrPlayer) {try {response.setHeader(HeaderConstant.KEY_CONTENT_DISPOSITION, "inline; filename=\"" + URLEncoder.encode(filename, "UTF-8"));} catch (UnsupportedEncodingException e) {throw new RuntimeException(e);}}}@Overridepublic void download(String path, HttpServletRequest request, HttpServletResponse response) {FileInputStream is = null;try {File file = new File(path);if (!file.exists()) {throw new RuntimeException();}String name = file.getName();long length = file.length();String contentType = "video/mp4";is = new FileInputStream(file);response.setHeader(HeaderConstant.KEY_CONTENT_LENGTH,  length + "");response.setHeader(HeaderConstant.KEY_CONTENT_TYPE, contentType);response.setHeader(HeaderConstant.KEY_CONTENT_DISPOSITION, "attachment;filename=" + name);ServletOutputStream outputStream = response.getOutputStream();channelCopy(is.getChannel(), Channels.newChannel(outputStream), 0, length);outputStream.flush();outputStream.close();} catch (Exception e) {} finally {if (is != null) {try {is.close();} catch (IOException e) {}}}}private static void channelCopy(FileChannel readableChannel, WritableByteChannel writableByteChannel, long position, long size) {try {if (writableByteChannel.isOpen()) {readableChannel.transferTo(position, size, writableByteChannel);}} catch (Exception e) {} finally {try {readableChannel.close();writableByteChannel.close();} catch (IOException e) {}}}
}
java">package com.zzc.component.download;import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;public interface OBSComponent {void download(String url, HttpServletRequest request, HttpServletResponse response);}
java">package com.zzc.component.download;import com.zzc.component.http.RestTemplateComponent;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.client.ClientHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.client.ResponseExtractor;import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.ReadableByteChannel;
import java.nio.channels.WritableByteChannel;
import java.util.HashMap;
import java.util.Map;@Slf4j
@Component
public class OBSComponentImpl implements OBSComponent {@Autowiredprivate RestTemplateComponent restTemplateComponent;@Overridepublic void download(String url, HttpServletRequest request, HttpServletResponse response) {Map<String, String> headers = new HashMap<>();ResponseExtractor<Boolean> extractor = new ResponseExtractor<Boolean>() {@Overridepublic Boolean extractData(ClientHttpResponse clientHttpResponse) throws IOException {HttpStatus statusCode = clientHttpResponse.getStatusCode();if (statusCode.isError()) {return false;}//TODO headMap<String, String> header = new HashMap<>();return true;}};restTemplateComponent.transfer(url, headers, extractor);}private static void channelCopy(ReadableByteChannel readableByteChannel, WritableByteChannel writableByteChannel) {try {ByteBuffer buffer = ByteBuffer.allocateDirect(32 * 1024);while (writableByteChannel.isOpen() && readableByteChannel.read(buffer) != -1) {buffer.flip();writableByteChannel.write(buffer);buffer.compact();}buffer.flip();while (writableByteChannel.isOpen() && buffer.hasRemaining()) {writableByteChannel.write(buffer);}} catch (Exception e) {} finally {try {readableByteChannel.close();writableByteChannel.close();} catch (IOException e) {}}}
}

http://www.ppmy.cn/devtools/111136.html

相关文章

使用 MongoDB 构建 AI：Patronus 如何自动进行大语言模型评估来增强对生成式 AI 的信心

使用 MongoDB 构建 AI：Patronus 如何自动进行大语言模型评估来增强对生成式 AI 的信心

大语言模型可能不可靠&#xff0c;这几乎算不上头条新闻。对于某些用例&#xff0c;这可能会带来不便。而对于其他行业&#xff0c;尤其是受监管行业&#xff0c;后果则要严重得多。于是&#xff0c;业内首个大语言模型自动评估平台 Patronus AI 应运而生。 Patronus AI 由 Met…
阅读更多...
浏览器和nodejs事件循环（Event Loop）有什么区别？

浏览器和nodejs事件循环（Event Loop）有什么区别？

单线程和异步 JS是单线程的&#xff0c;无论在浏览器还是在nodejs浏览器中JS执行和DOM渲染共用一个线程&#xff0c;是互斥的异步是单线程的解决方案 1. 浏览器中的事件循环 异步里面分宏任务和微任务 宏任务&#xff1a;setTimeout&#xff0c;setInterval&#xff0c;set…
阅读更多...
pytorch对不同的可调参数，分配不同的学习率

pytorch对不同的可调参数，分配不同的学习率

在 PyTorch 中&#xff0c;你可以通过为优化器传递不同的学习率来针对不同的可调参数分配不同的学习率。这通常通过向优化器传递一个字典列表来实现&#xff0c;其中每个字典指定特定参数组的学习率。下面是一个示例代码&#xff0c;展示了如何实现这一点&#xff1a; import …
阅读更多...
Python 多线程

Python 多线程

开始学习Python线程 线程模块 使用Threading模块创建线程 线程同步 线程优先级队列&#xff08; Queue&#xff09; 多线程类似于同时执行多个不同程序&#xff0c;多线程运行有如下优点&#xff1a; 使用线程可以把占据长时间的程序中的任务放到后台去处理。用户界面可以更…
阅读更多...
Pyspark下操作dataframe方法(1)

Pyspark下操作dataframe方法(1)

文章目录 Pyspark dataframe创建DataFrame使用Row对象使用元组与scheam使用字典与scheam注意 agg 聚合操作alias 设置别名字段设置别名设置dataframe别名 cache 缓存checkpoint RDD持久化到外部存储coalesce 设置dataframe分区数量collect 拉取数据columns 获取dataframe列 Pys…
阅读更多...
苹果宣布iOS 18正式版9月17日推送：支持27款iPhone升级

苹果宣布iOS 18正式版9月17日推送：支持27款iPhone升级

9月10日消息&#xff0c;在苹果秋季发布会结束后&#xff0c; 苹果宣布将于9月17日(下周二)推送iOS 18正式版系统。 苹果官网显示&#xff0c;iOS 18正式版将兼容第二代iPhone SE及之后的所有机型&#xff0c;加上刚发布的iPhone 16系列&#xff0c;共兼容27款iPhone。 iOS 18升…
阅读更多...
【数据获取与读取】JSON CSV

【数据获取与读取】JSON CSV

数据分析流程 获取数据-读取数据-评估数据-清洗数据-整理数据-分析数据-可视化数据 公开数据集 飞桨&#xff08;百度旗下深度学习平台&#xff09;数据集&#xff1a;https:/aistudio.baidu.com/aistudio/datasetoverview 天池&#xff08;阿里云旗下开发者竞赛平台&#xf…
阅读更多...
​了解MySQL 的二进制日志文件​Binlog

​了解MySQL 的二进制日志文件​Binlog

1. SQL 语句的几种类型 首先介绍一下&#xff0c;对于一个 SQL 语句&#xff0c;它常常被分为以下几种类型&#xff1a; DDL&#xff08;Data Definition Language&#xff0c;数据定义语言&#xff09;&#xff1a;用来操作数据库、表、列等&#xff0c;比如 CREATE、ALTER…
阅读更多...
最新文章

Copyright @ 2022~2023