返回文件流/文件预览/文件上传下载

698 阅读1分钟
package com.example.replace.controller;

import org.apache.tomcat.util.http.fileupload.IOUtils;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.nio.charset.StandardCharsets;

/**
 * @description:
 * @author: Sandul
 * @create: 2021-05-07 09:47
 **/
@Controller
@RequestMapping("/test")
public class TestController {

	@GetMapping("/pdf")
	public void Outpust(HttpServletResponse response) throws IOException {
		//使用这个返回pdf文件类型的文件流(浏览器中直接预览)
		//response.setContentType(MediaType.APPLICATION_PDF_VALUE); 
		//使用这个直接返回流,在浏览器中直接下载
		response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);
		//
		response.setCharacterEncoding(StandardCharsets.UTF_8.name());
		OutputStream outputStream = null;
		InputStream inputStream = null;
		long start = System.currentTimeMillis();
		try {
			outputStream = response.getOutputStream();
			inputStream = new BufferedInputStream(new FileInputStream("D:\\JAVA\\javaCode\\replace\\src\\main\\resources\\123.pdf"));
//			inputStream = new ByteArrayInputStream("hello world".getBytes());
			int len=0;
			byte[] bytes = new byte[4096];
			while ((len=inputStream.read(bytes))!=-1){
				outputStream.write(bytes,0,len);
				outputStream.flush();
				System.out.println("-----------------"+len);
			}
//			IOUtils.copy(inputStream, outputStream);
			long end = System.currentTimeMillis();
			System.out.println(end-start);
		}catch (Exception e){
			e.printStackTrace();
		} finally {
			IOUtils.closeQuietly(outputStream);
			IOUtils.closeQuietly(inputStream);
		}
	}
	//文件上传
	@ResponseBody
	@PostMapping("/excel/upload")
	public boolean uploadExcel(HttpServletRequest request,MultipartFile excel){
		OutputStream outputStream=null;
		try (InputStream inputStream = excel.getInputStream();){
			System.out.println("文件上传开始。。。。。");
			String name = excel.getOriginalFilename();
			System.out.println("文件名:"+name);
			File file = new File("D:\\upload\\"+name);
			if (!file.exists()){
				file.createNewFile();
			}
			System.out.println(file.getAbsolutePath());
			outputStream = new BufferedOutputStream(new FileOutputStream(file));
			int len = 0;
			byte[] bytes = new byte[2048];
			while ((len=inputStream.read(bytes))!=-1){
				outputStream.write(bytes,0,len);
				outputStream.flush();
				System.out.println(len);
			}
			System.out.println("文件上传完成!");
			return true;
		} catch (IOException e) {
			e.printStackTrace();
			return false;
		}finally {
			IOUtils.closeQuietly(outputStream);
		}

	}

}

如果报上传文件过大的异常,可以在yml配置

spring:
	servlet:
	    multipart:
	      # 上传的文件大小
	      max-file-size: -1 #-1代表无限制
	      # 发送的请求大小
	      max-request-size: -1 #-1代表无限制