java后台读取存储在HDFS中的视频文件获取视频流后使用h5进行播放/下载

系统管理员 2024-04-03 13:16 14阅读 0赞

### 一、后台代码： ###

@GetMapping("/downloadFile")
    	public void downLoad(@RequestParam(name = "fileName", required = true) final String fileName,
    			final HttpServletRequest request, final HttpServletResponse response) throws Throwable {
    
    		FileSystem fileSystem = hdfsUtils.getFileSystem();
    		
    		long fileLength = 0;
    		
    		//获取文件大小
    		FileStatus[] status = fileSystem.listStatus(new Path("/hk-video/"));
    		for (FileStatus file : status) {
    			if(file.getPath().getName().equals(fileName)) {
    				fileLength = file.getLen();
    			}
    		}
    		
    		Path path = new Path("/hk-video/" + fileName);
    		FSDataInputStream fis = fileSystem.open(path);
    		
    		response.reset();
    		// 获取从那个字节开始读取文件
    		String rangeString = request.getHeader("Range");
    		System.out.println("Range="+rangeString);
    		try {
    			// 获取响应的输出流
    			OutputStream outputStream = response.getOutputStream();
    			// 播放
    			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);
    				// 设定文件读取开始位置（以字节为单位）
    			    fis.seek(range);
    			} else {// 下载
    				// 设置响应头，把文件名字设置好
    				response.setHeader("Content-Disposition", "attachment; filename=Time20221008170000to20221008172000.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 = fis.read(cache)) != -1) {
    				outputStream.write(cache, 0, flag);
    			}
    
    			outputStream.flush();
    			outputStream.close();
    
    		} catch (FileNotFoundException e) {
    
    		} catch (IOException e) {
    
    		}
    	}

### 二、前端页面： ###

<video id="my-video" class="video-js vjs-big-play-centered" controls="controls" poster="">
         <source src="10.168.2.21:13456/downloadFile?fileName=111.mp4" type='video/mp4'>
              <p class="vjs-no-js">
                   <a href="http://videojs.com/html5-video-support/" target="_blank">supports HTML5 video</a>
              </p>
    </video>

### 三、java读取hdfs文件夹下的文件列表信息 ###

/**
    	 * 得到一个文件夹(不包含子文件夹)下的全部名字匹配上pattern的文件名称
    	 * 
    	 * @param fs
    	 * @param folderPath
    	 * @param pattern
    	 * @return
    	 * @throws IOException
    	 */
    	public static List getFilesUnderFolder(final FileSystem fs, final Path folderPath, final String pattern)
    			throws IOException {
    		List paths = new ArrayList();
    		if (fs.exists(folderPath)) {
    			FileStatus[] fileStatus = fs.listStatus(folderPath);
    			for (int i = 0; i < fileStatus.length; i++) {
    				FileStatus fileStatu = fileStatus[i];
    				if (!fileStatu.isDir()) {// 仅仅要文件
    					Path oneFilePath = fileStatu.getPath();
    					if (pattern == null) {
    						paths.add(oneFilePath);
    					} else {
    						if (oneFilePath.getName().contains(pattern)) {
    							paths.add(oneFilePath);
    						}
    					}
    				}
    			}
    		}
    		return paths;
    	}