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代表无限制