ResponseEntity<byte[]>和ResponseEntity<InputStreamResource>有以下区别:
ResponseEntity<byte[]>:表示响应体是字节数组。适用于将字节数组作为文件内容直接返回给客户端浏览器进行下载。字节数组可以直接转换为文件内容,客户端可以根据响应的Content-Type进行下载处理。
ResponseEntity<InputStreamResource>:表示响应体是输入流资源。适用于需要在服务端进行一些处理,如读取文件、加密等,然后将处理后的输入流封装为InputStreamResource对象返回给客户端。客户端可以使用输入流进行进一步处理,例如保存到本地文件或进行其他操作。
选择使用哪种返回值类型取决于你的具体需求和场景。
如果你的文件下载逻辑在服务端已经处理完毕,且可以直接将文件内容封装为字节数组返回给客户端,那么选择ResponseEntity<byte[]>更合适。
如果你需要在服务端进行一些额外的处理,然后将处理后的输入流返回给客户端,那么选择ResponseEntity<InputStreamResource>更合适。
需要注意的是,在使用ResponseEntity<InputStreamResource>时,需要注意及时关闭输入流,以避免资源泄漏。
@GetMapping("/download")
public ResponseEntity<InputStreamResource> downloadFile(@RequestParam("fileName") String fileName) throws IOException {
MediaType mediaType = MediaTypeUtils.getMediaTypeForFileName(this.servletContext, fileName);
System.out.println("fileName: " + fileName);
System.out.println("mediaType: " + mediaType);
File file = new File(UPLOADED_FOLDER + fileName);
InputStreamResource resource = new InputStreamResource(new FileInputStream(file));
return ResponseEntity.ok()
.header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=" + file.getName())
.contentType(mediaType)
.contentLength(file.length())
.body(resource);
}
/*在返回ResponseEntity之前,使用try-with-resources语句块或显式地调用close()方法来关闭FileInputStream对象,以确保资源被正确释放。
例如,在try块结束后添加一个finally块,并在其中调用fileInputStream.close()来关闭输入流:
*/
try {
fileInputStream = new FileInputStream(file);
resource = new InputStreamResource(fileInputStream);
} finally {
if (fileInputStream != null) {
fileInputStream.close();
}
}