package com.iplatform.recvideo.controller;
|
|
import com.iplatform.model.po.Rc_video_batch;
|
import com.iplatform.model.po.Rc_video_t2;
|
import com.iplatform.recvideo.config.VideoSimilarProperties;
|
import com.iplatform.recvideo.service.VideoShowServiceImpl;
|
import com.walker.db.page.GenericPager;
|
import com.walker.infrastructure.utils.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.ui.Model;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.ResponseBody;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.File;
|
import java.io.OutputStream;
|
import java.io.RandomAccessFile;
|
import java.nio.charset.StandardCharsets;
|
import java.util.List;
|
|
@Controller
|
@RequestMapping("/video_ui")
|
public class VideoResultController {
|
|
private VideoShowServiceImpl videoShowService;
|
private VideoSimilarProperties videoSimilarProperties;
|
|
@Autowired
|
public VideoResultController(VideoShowServiceImpl videoShowService, VideoSimilarProperties videoSimilarProperties){
|
this.videoShowService = videoShowService;
|
this.videoSimilarProperties = videoSimilarProperties;
|
}
|
|
@GetMapping("/index")
|
public String index(Model model, String id) {
|
|
GenericPager<Rc_video_batch> pager = this.videoShowService.queryPageVideoBatchList();
|
List<Rc_video_batch> data = pager.getDatas();
|
model.addAttribute("src_list", data);
|
model.addAttribute("video_folder", this.videoSimilarProperties.getDataFolder());
|
|
if(StringUtils.isNotEmpty(id)){
|
List<Rc_video_t2> list = this.videoShowService.querySimilarVideoList(id);
|
model.addAttribute("similar_list", list);
|
model.addAttribute("select_id", id);
|
} else {
|
model.addAttribute("select_id", null);
|
}
|
return "video/index";
|
}
|
|
/**
|
* 本地视频转视频流流
|
* @param response
|
* @return
|
*/
|
@RequestMapping("/get_video")
|
@ResponseBody
|
public void getVideo(
|
// HttpServletRequest request,
|
HttpServletResponse response, String id) {
|
if(StringUtils.isEmpty(id)){
|
System.out.println("视频ID未提供");
|
return;
|
}
|
//视频资源存储信息
|
response.reset();
|
//获取从那个字节开始读取文件
|
// String rangeString = request.getHeader("Range");
|
String rangeString = null;
|
|
Rc_video_batch videoInfo = this.videoShowService.queryOneVideoInfo(id);
|
if(videoInfo == null){
|
System.out.println("Rc_video_batch未查找到对象: " + id);
|
return;
|
}
|
String videoPath = videoInfo.getSrc_video_path();
|
if(this.videoSimilarProperties.isTestMode()){
|
// 测试模式，需要替换文件根路径为本机windows路径
|
StringBuilder windowPath = new StringBuilder(this.videoSimilarProperties.getDataFolder());
|
windowPath.append(videoInfo.getBatch_id());
|
windowPath.append("/").append(id).append(".mp4");
|
videoPath = windowPath.toString();
|
}
|
|
try {
|
//获取响应的输出流
|
OutputStream outputStream = response.getOutputStream();
|
// File file = new File("c:\\video.mp4");
|
File file = new File(videoPath);
|
if(file.exists()){
|
RandomAccessFile targetFile = new RandomAccessFile(file, "r");
|
long fileLength = targetFile.length();
|
//播放
|
if(rangeString != null){
|
long range = Long.valueOf(rangeString.substring(rangeString.indexOf("=") + 1, rangeString.indexOf("-")));
|
//设置内容类型
|
response.setHeader("Content-Type", "video/mp4");
|
//设置此次相应返回的数据长度
|
response.setHeader("Content-Length", String.valueOf(fileLength - range));
|
//设置此次相应返回的数据范围
|
response.setHeader("Content-Range", "bytes "+range+"-"+(fileLength-1)+"/"+fileLength);
|
//返回码需要为206，而不是200
|
response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);
|
//设定文件读取开始位置（以字节为单位）
|
// targetFile.seek(range);
|
}else {//下载
|
//设置响应头，把文件名字设置好
|
response.setHeader("Content-Disposition", "attachment; filename=video.mp4" );
|
//设置文件长度
|
response.setHeader("Content-Length", String.valueOf(fileLength));
|
//解决编码问题
|
response.setHeader("Content-Type","application/octet-stream");
|
}
|
|
byte[] cache = new byte[1024 * 300];
|
int flag;
|
while ((flag = targetFile.read(cache))!=-1){
|
outputStream.write(cache, 0, flag);
|
}
|
}else {
|
String message = "file: not exists";
|
//解决编码问题
|
response.setHeader("Content-Type","application/json");
|
outputStream.write(message.getBytes(StandardCharsets.UTF_8));
|
}
|
|
outputStream.flush();
|
outputStream.close();
|
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
