Microservice/pqs
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

代码调整

Browse Source
This commit is contained in:
hongawen
2023-05-29 08:47:32 +08:00
parent a74500afe8
commit 8028a058de
2 changed files with 151 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pqs-common/common-oss/src/main/java/com/njcn/oss/utils/FileStorageUtil.java
View File

@@ -143,7 +143,7 @@ public class FileStorageUtil {
* @date 2023/3/7 23:04
* @param filePath 文件在服务器的路径
*/
public void downloadStream(HttpServletResponse response, String filePath) throws IOException {
public void downloadStream(HttpServletResponse response, String filePath) {
InputStream inputStream;
OutputStream toClient = null;
try {

166
pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/controller/ResourceController.java
View File

@@ -1,30 +1,36 @@
package com.njcn.device.pq.controller;
import cn.hutool.core.io.FileUtil;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.njcn.common.config.GeneralInfo;
import com.njcn.common.pojo.annotation.OperateInfo;
import com.njcn.common.pojo.enums.common.LogEnum;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.response.HttpResult;
import com.njcn.common.utils.HttpResultUtil;
import com.njcn.device.pq.pojo.param.LargeScreenParam;
import com.njcn.device.pq.mapper.ResourceMapper;
import com.njcn.device.pq.pojo.po.ResourceData;
import com.njcn.device.pq.pojo.vo.MonitoringPointScaleVO;
import com.njcn.device.pq.service.LargeScreenService;
import com.njcn.device.pq.service.ResourceService;
import com.njcn.system.pojo.po.Resinformation;
import com.njcn.oss.utils.FileStorageUtil;
import com.njcn.web.controller.BaseController;
import io.minio.GetObjectArgs;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.annotation.Validated;
import org.apache.commons.io.FileUtils;
import org.apache.tomcat.util.http.fileupload.IOUtils;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.*;
import java.util.List;
/**
@@ -41,6 +47,12 @@ public class ResourceController extends BaseController {
private final ResourceService iResourceAdministrationService;
private final ResourceMapper resourceMapper;
private final FileStorageUtil fileStorageUtil;
private final GeneralInfo generalInfo;
/**
* 上传资源
*/
@@ -70,18 +82,140 @@ public class ResourceController extends BaseController {
return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, result, methodDescribe);
}
/**
* 预览资源
*/
//todo 未完成
@OperateInfo(info = LogEnum.BUSINESS_COMMON)
@GetMapping("/previewFile")
@ApiOperation("预览资源")
@GetMapping("/play1")
@ApiOperation("播放视频资源")
@ApiImplicitParam(name = "id", value = "id", required = true)
public HttpResult<Object> previewFile(@RequestParam("id") String id, HttpServletRequest request, HttpServletResponse response) throws IOException {
String methodDescribe = getMethodDescribe("downloadFile");
iResourceAdministrationService.previewFile(id, request, response);
return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, "成功", methodDescribe);
public void play1(@RequestParam("id") String id, HttpServletRequest request, HttpServletResponse response) {
ResourceData resourceData = resourceMapper.selectById(id);
fileStorageUtil.downloadStream(response,resourceData.getResUrl());
}
@OperateInfo(info = LogEnum.BUSINESS_COMMON)
@GetMapping("/play2")
@ApiOperation("播放视频资源")
@ApiImplicitParam(name = "id", value = "id", required = true)
public void play2(@RequestParam("id") String id, HttpServletRequest request, HttpServletResponse response) {
RandomAccessFile targetFile = null;
OutputStream outputStream = null;
try {
outputStream = response.getOutputStream();
response.reset();
//获取请求头中Range的值
String rangeString = request.getHeader(HttpHeaders.RANGE);
//打开文件
File file = File.createTempFile(IdWorker.getIdStr(),".mp4");
ResourceData resourceData = resourceMapper.selectById(id);
InputStream ins = fileStorageUtil.getFileStream(resourceData.getResUrl());
long l = System.currentTimeMillis();
FileUtil.writeFromStream(ins,file);
long ll = System.currentTimeMillis();
System.out.println("流转文件耗时："+(ll-l)+"=============================");
if (file.exists()) {
//使用RandomAccessFile读取文件
targetFile = new RandomAccessFile(file, "r");
long fileLength = targetFile.length();
long requestSize = (int) fileLength;
//分段下载视频
if (StringUtils.hasText(rangeString)) {
//从Range中提取需要获取数据的开始和结束位置
long requestStart = 0, requestEnd = 0;
String[] ranges = rangeString.split("=");
if (ranges.length > 1) {
String[] rangeDatas = ranges[1].split("-");
requestStart = Integer.parseInt(rangeDatas[0]);
if (rangeDatas.length > 1) {
requestEnd = Integer.parseInt(rangeDatas[1]);
}
}
if (requestEnd != 0 && requestEnd > requestStart) {
requestSize = requestEnd - requestStart + 1;
}
//根据协议设置请求头
response.setHeader(HttpHeaders.ACCEPT_RANGES, "bytes");
response.setHeader(HttpHeaders.CONTENT_TYPE, "video/mp4");
if (!StringUtils.hasText(rangeString)) {
response.setHeader(HttpHeaders.CONTENT_LENGTH, fileLength + "");
} else {
long length;
if (requestEnd > 0) {
length = requestEnd - requestStart + 1;
response.setHeader(HttpHeaders.CONTENT_LENGTH, "" + length);
response.setHeader(HttpHeaders.CONTENT_RANGE, "bytes " + requestStart + "-" + requestEnd + "/" + fileLength);
} else {
length = fileLength - requestStart;
response.setHeader(HttpHeaders.CONTENT_LENGTH, "" + length);
response.setHeader(HttpHeaders.CONTENT_RANGE, "bytes " + requestStart + "-" + (fileLength - 1) + "/"
+ fileLength);
}
}
//断点传输下载视频返回206
response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);
//设置targetFile，从自定义位置开始读取数据
targetFile.seek(requestStart);
} else {
//如果Range为空则下载整个视频
response.setHeader(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=demo.mp4");
//设置文件长度
response.setHeader(HttpHeaders.CONTENT_LENGTH, String.valueOf(fileLength));
}
//从磁盘读取数据流返回
byte[] cache = new byte[4096];
try {
while (requestSize > 0) {
int len = targetFile.read(cache);
if (requestSize < cache.length) {
outputStream.write(cache, 0, (int) requestSize);
} else {
outputStream.write(cache, 0, len);
if (len < cache.length) {
break;
}
}
requestSize -= cache.length;
}
} catch (IOException e) {
// tomcat原话。写操作IO异常几乎总是由于客户端主动关闭连接导致，所以直接吃掉异常打日志
//比如使用video播放视频时经常会发送Range为0- 的范围只是为了获取视频大小，之后就中断连接了
log.info(e.getMessage());
}
} else {
throw new RuntimeException("文件路劲有误");
}
outputStream.flush();
} catch (Exception e) {
log.error("文件传输错误", e);
throw new RuntimeException("文件传输错误");
}finally {
if(outputStream != null){
try {
outputStream.close();
} catch (IOException e) {
log.error("流释放错误", e);
}
}
if(targetFile != null){
try {
targetFile.close();
} catch (IOException e) {
log.error("文件流释放错误", e);
}
}
}
}
public static long getFileSize(InputStream inputStream) throws IOException {
byte[] buffer = new byte[1024];
int len;
long size = 0;
while ((len = inputStream.read(buffer)) != -1) {
size += len;
}
inputStream.close();
return size;
}
/**