学习笔记（十）：上传文件、下载视频

一、上传文件

    String uploadPath ="D:/opt/upFiles";
    @ApiOperation(value="上传录音录像文件", notes="上传录音录像文件（video目录）")
    @PostMapping("/uploadVideo")
    public R<?> uploadVideo(@RequestParam(value="file", required=false) MultipartFile file,
                            @RequestParam(value = "matterCode")String matterCode) {
        try {
            //获取附件拓展名
            String name = file.getOriginalFilename();
            System.out.println("文件名:" + name);
            if (org.springframework.util.StringUtils.isEmpty(name) || !name.contains(".")) {
                return R.error("文件名无效！");
            }
            String[] split = name.split("\\.");
            String prefx = split[split.length - 1];

            List<String> invalidTypeList = Arrays.asList("sh", "exe", "bat");
            if (invalidTypeList.contains(prefx)) {
                return R.error("类型不符！");
            }

            //线上
            String path = uploadPath + "/video/";
            String fileName = UUID.randomUUID().toString() + "." + prefx;

            @Cleanup BufferedInputStream bis = new BufferedInputStream(file.getInputStream());
            @Cleanup FileOutputStream os = new FileOutputStream(new File(path, fileName));

            int b;
            byte[] buffer = new byte[1024];
            while ((b = bis.read(buffer)) != -1) {
                os.write(buffer, 0, b);
                os.flush();
            }
            MatterLog matterLog = new MatterLog();
            String oldName = "/video/"+name;
            matterLog.setVideoName(oldName);
            matterLog.setVideoRecord(fileName);
            System.out.println("videoOldName:"+name);
            System.out.println("videoNewName:"+fileName);

            matterLogmapper.updateMatterVideo(oldName,fileName,matterCode);
        } catch (IOException e) {
            log.error("uploadVideo exception. ", e);
            return R.error("录音录像入库失败，请重试！");
        }
        return R.OK("录音录像入库成功","/video/"+file.getOriginalFilename());
    }

二、下载视频（转流）

@ApiOperation(value = "文件下载-录音录像文件", notes = "文件下载-录音录像文件")
    @GetMapping("/downloadVideo")
    public void downloadVideo(@RequestParam(value = "name",required = true) String name,
                                       HttpServletRequest request, HttpServletResponse response) {
        try {
            //1，参数处理
            String filePath = uploadPath + "/video/" + name;
            filePath = filePath.replace("\\", File.separator).replace("//", File.separator)
                    .replace("/", File.separator);
            filePath = filePath.replace("\\\\", File.separator).replace("\\", File.separator)
                    .replace("//", File.separator).replace("/", File.separator);

            File file = new File(filePath);
            @Cleanup RandomAccessFile randomFile = new RandomAccessFile(file, "r");
            long contentLength = randomFile.length();
            String range = request.getHeader("Range");
            int start = 0, end = 0;
            if(range != null && range.startsWith("bytes=")){
                String[] values = range.split("=")[1].split("-");
                start = Integer.parseInt(values[0]);
                if(values.length > 1){
                    end = Integer.parseInt(values[1]);
                }
            }
            int requestSize = 0;
            if(end != 0 && end > start){
                requestSize = end - start + 1;
            } else {
                requestSize = Integer.MAX_VALUE;
            }

            response.setContentType("video/mp4");
            response.setHeader("Accept-Ranges", "bytes");
            response.setHeader("ETag", name);
            response.setHeader("Last-Modified", new Date().toString());
            //第一次请求只返回content length来让客户端请求多次实际数据
            if(range == null){
                response.setHeader("Content-length", contentLength + "");
            }else{
                //以后的多次以断点续传的方式来返回视频数据
                response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);
                long requestStart = 0, requestEnd = 0;
                String[] ranges = range.split("=");
                if(ranges.length > 1){
                    String[] rangeDatas = ranges[1].split("-");
                    requestStart = Integer.parseInt(rangeDatas[0]);
                    if(rangeDatas.length > 1){
                        requestEnd = Integer.parseInt(rangeDatas[1]);
                    }
                }
                long length = 0;
                if(requestEnd > 0){
                    length = requestEnd - requestStart + 1;
                    response.setHeader("Content-length", "" + length);
                    response.setHeader("Content-Range", "bytes " + requestStart + "-" + requestEnd + "/" + contentLength);
                }else{
                    length = contentLength - requestStart;
                    response.setHeader("Content-length", "" + length);
                    response.setHeader("Content-Range", "bytes "+ requestStart + "-" + (contentLength - 1) + "/" + contentLength);
                }
            }
            @Cleanup ServletOutputStream out = response.getOutputStream();
            int needSize = requestSize;
            randomFile.seek(start);
            while(needSize > 0){
                byte[] buffer = new byte[4096];
                int len = randomFile.read(buffer);
                if(needSize < buffer.length){
                    out.write(buffer, 0, needSize);
                } else {
                    out.write(buffer, 0, len);
                    if(len < buffer.length){
                        break;
                    }
                }
                needSize -= buffer.length;
            }
        } catch (IOException e) {
            log.error("downloadVideo exception :", e);
        }
    }

三、下载文件

    @ApiOperation(value = "文件下载", notes = "文件下载")
    @GetMapping("/download")
    public void handleCommonFile(@RequestParam("name") String name, HttpServletResponse response) {
        try {
            //1，参数处理
            String filePath = uploadPath + "/video/" + name;
            filePath = filePath.replace("\\", File.separator).replace("//", File.separator).replace("/", File.separator);

            String type = "";
            if (name.contains(".")) {
                type = name.substring(name.lastIndexOf(".") + 1);
            }

            //2，文件下载
            List<String> validTypeList = Arrays.asList("jpeg", "jpg", "png", "JPEG", "JPG", "PNG", "mp4");
            if (validTypeList.contains(type)) {
                response.setContentType("image/jpeg");
            } else {
                response.setContentType("application/octet-stream");
                response.setHeader("Content-Disposition", "attachment; filename=" + new String(name.getBytes(), StandardCharsets.ISO_8859_1));
            }
            @Cleanup BufferedInputStream bis = new BufferedInputStream(new FileInputStream(filePath));
            @Cleanup OutputStream os = response.getOutputStream();

            int b;
            byte[] buffer = new byte[1024];
            while ((b = bis.read(buffer)) != -1) {
                os.write(buffer, 0, b);
            }
            os.flush();
        } catch (IOException e) {
            log.error("handleCommonFile exception :", e);
        }
    }