package com.atguigu.controller;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.util.MultiValueMap;
import org.springframework.web.bind.annotation.RequestMapping;
import javax.servlet.ServletContext;
import javax.servlet.http.HttpSession;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
/**
* ResponseEntity:可以作为控制器方法的返回值,表示响应到浏览器的完整的响应报文
*/
@Controller
public class FileUpAndDownController {
@RequestMapping("test/down")
public ResponseEntity<byte[]> testResponseEntity(HttpSession session) throws
IOException {
//获取ServletContext对象
ServletContext servletContext = session.getServletContext();
//获取服务器中文件的真实路径
String realPath = servletContext.getRealPath("/img/1.jpg");
//创建输入流
InputStream is = new FileInputStream(realPath);
//创建字节数组,is.available()获取输入流所对应文件的字节数
byte[] bytes = new byte[is.available()];
//将流读到字节数组中
is.read(bytes);
//创建HttpHeaders对象设置响应头信息
MultiValueMap<String, String> headers = new HttpHeaders();
//设置要下载方式以及下载文件的名字
headers.add("Content-Disposition", "attachment;filename=1.jpg");
//设置响应状态码
HttpStatus statusCode = HttpStatus.OK;
//创建ResponseEntity对象
ResponseEntity<byte[]> responseEntity = new ResponseEntity<>(bytes, headers,
statusCode);
//关闭输入流
is.close();
return responseEntity;
}
}
文件的上传 :
文件上传的要求: * 1.form表单请求方式必须为post * 2.form表单必须设置属性enctype="multipart/form-data"
@RequestMapping("/test/up")
public String testup(HttpSession session, MultipartFile photo) throws IOException {
// 获取上传文件的文件名
String originalFilename = photo.getOriginalFilename();
System.out.println(originalFilename);
// 获取上传的文件的后缀名
String substring = originalFilename.substring(originalFilename.lastIndexOf("."));
// 获取uuid
String s = UUID.randomUUID().toString();
// 拼接一个新的文件名
originalFilename=s+substring;
// 获取servletContext
ServletContext servletContext = session.getServletContext();
// 获取当前工程下photo的真实路径
String photoPath = servletContext.getRealPath("photo");
// 创建photoPath所对应的File对象
File file = new File(photoPath);
// 判断file所对应的目录是否存在
if(!file.exists()){
file.mkdir();
}
String finalPath = photoPath+File.separator+originalFilename;
// 上传文件
photo.transferTo(new File(finalPath));
return "success";
}
还需要配置依赖和在springmvc中配置文件上传解析器
下面是依赖:
<!-- 设置-->
<!-- https://mvnrepository.com/artifact/commons-fileupload/commons-fileupload -->
<dependency>
<groupId>commons-fileupload</groupId>
<artifactId>commons-fileupload</artifactId>
<version>1.3.1</version>
</dependency>
下面是文件上传解析器:
<!-- 配置文件上传解析器,此bean是根据id获取-->
<bean id="multipartResolver" class="org.springframework.web.multipart.commons.CommonsMultipartResolver">
</bean>