package com.estate.expert.admin.controller.admin;
|
|
import com.estate.expert.admin.domain.misc.UserFile;
|
import com.estate.expert.admin.enums.BizResultCode;
|
import com.estate.expert.admin.service.UserFileService;
|
import com.estate.expert.common.auth.IgnoreAuth;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.SneakyThrows;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.core.io.InputStreamResource;
|
import org.springframework.http.HttpHeaders;
|
import org.springframework.http.MediaType;
|
import org.springframework.http.ResponseEntity;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.annotation.Resource;
|
import java.io.IOException;
|
import java.net.URLEncoder;
|
|
@Api(tags = "管理后台/文件管理")
|
@CrossOrigin
|
@Slf4j
|
@Controller
|
@RequestMapping("/public/file")
|
public class UserFileDownloadController {
|
|
@Resource
|
UserFileService userFileService;
|
|
@SneakyThrows
|
@ApiOperation(value = "下载文件（名称）")
|
@IgnoreAuth
|
@GetMapping("/download/{name}")
|
public ResponseEntity<org.springframework.core.io.Resource> downloadFileByName(@PathVariable String name) {
|
var userFile = userFileService.getByName(name);
|
BizResultCode.FILE_NOT_FOUND.check(userFile == null);
|
|
return downloadFile(userFile);
|
}
|
|
@SneakyThrows
|
@ApiOperation(value = "下载文件（文件ID）")
|
@IgnoreAuth
|
@GetMapping("/download/file")
|
public ResponseEntity<org.springframework.core.io.Resource> downloadFileById(@RequestParam(name = "id") Integer id) {
|
var userFile = userFileService.get(id);
|
BizResultCode.FILE_NOT_FOUND.check(userFile == null);
|
|
return downloadFile(userFile);
|
}
|
|
@ApiOperation(value = "图片显示")
|
@IgnoreAuth
|
@GetMapping("/image/{name}")
|
public ResponseEntity<org.springframework.core.io.Resource> showImage(@PathVariable String name) {
|
var userFile = userFileService.getByName(name);
|
BizResultCode.FILE_NOT_FOUND.check(userFile == null);
|
|
return ResponseEntity.ok()
|
.contentLength(userFile.getSize())
|
.contentType(MediaType.parseMediaType(userFile.getContentType()))
|
.body(userFileService.resolveResource(userFile));
|
}
|
|
private ResponseEntity<org.springframework.core.io.Resource> downloadFile(UserFile userFile) throws IOException {
|
InputStreamResource resource = userFileService.resolveResource(userFile);
|
var fileSize = userFile.getSize();
|
// if(resource == null) {
|
// resource = new InputStreamResource(getClass().getClassLoader().getResourceAsStream("images/wukong.png"));
|
// fileSize = 6215360L;// resource.contentLength();
|
// }
|
BizResultCode.FILE_NOT_FOUND.check(resource==null || !resource.exists());
|
|
String encodedFileName = URLEncoder.encode(userFile.getOriginFileName(), "UTF-8");
|
String headerValue = String.format("attachment; filename=\"%s\"", encodedFileName);
|
return ResponseEntity.ok()
|
.contentType(MediaType.parseMediaType(userFile.getContentType()))
|
.contentLength(fileSize)
|
.header(HttpHeaders.CONTENT_DISPOSITION, headerValue)
|
.body(resource);
|
}
|
|
|
|
}
|
