简小朤
1 month ago
14 changed files with 225 additions and 17 deletions
@ -0,0 +1,22 @@ |
|||
package com.coin.system.config; |
|||
|
|||
import org.springframework.beans.factory.annotation.Value; |
|||
import org.springframework.context.annotation.Configuration; |
|||
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry; |
|||
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; |
|||
|
|||
@Configuration |
|||
public class FileUploadConfig implements WebMvcConfigurer { |
|||
|
|||
@Value("${file.upload.path}") |
|||
private String uploadPath; |
|||
|
|||
@Value("${file.upload.sourcePath}") |
|||
private String sourcePath; |
|||
|
|||
@Override |
|||
public void addResourceHandlers(ResourceHandlerRegistry registry) { |
|||
registry.addResourceHandler(sourcePath+"/**") |
|||
.addResourceLocations("file:" + uploadPath + "/"); |
|||
} |
|||
} |
|||
@ -0,0 +1,29 @@ |
|||
package com.coin.system.config; |
|||
|
|||
import org.springframework.boot.web.servlet.MultipartConfigFactory; |
|||
import org.springframework.context.annotation.Bean; |
|||
import org.springframework.context.annotation.Configuration; |
|||
import org.springframework.util.unit.DataSize; |
|||
import org.springframework.web.multipart.MultipartResolver; |
|||
import org.springframework.web.multipart.support.StandardServletMultipartResolver; |
|||
|
|||
import javax.servlet.MultipartConfigElement; |
|||
|
|||
@Configuration |
|||
public class MultipartConfig { |
|||
|
|||
@Bean |
|||
public MultipartResolver multipartResolver() { |
|||
return new StandardServletMultipartResolver(); |
|||
} |
|||
|
|||
@Bean |
|||
public MultipartConfigElement multipartConfigElement() { |
|||
MultipartConfigFactory factory = new MultipartConfigFactory(); |
|||
// 设置单个文件最大大小为10MB
|
|||
factory.setMaxFileSize(DataSize.ofMegabytes(10)); |
|||
// 设置总上传数据最大大小为20MB
|
|||
factory.setMaxRequestSize(DataSize.ofMegabytes(20)); |
|||
return factory.createMultipartConfig(); |
|||
} |
|||
} |
|||
@ -0,0 +1,30 @@ |
|||
package com.coin.system.controller; |
|||
|
|||
import com.coin.system.service.FileService; |
|||
import com.coin.common.core.domain.R; |
|||
import lombok.RequiredArgsConstructor; |
|||
import org.springframework.web.bind.annotation.*; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
|
|||
import javax.servlet.http.HttpServletResponse; |
|||
|
|||
/** |
|||
* 文件通用接口 |
|||
*/ |
|||
@RestController |
|||
@RequestMapping("/file") |
|||
@RequiredArgsConstructor |
|||
public class FileController { |
|||
|
|||
private final FileService fileService; |
|||
|
|||
@PostMapping("/upload") |
|||
public R<String> upload(@RequestParam("file") MultipartFile file) { |
|||
return R.ok("上传成功！",fileService.uploadFile(file)); |
|||
} |
|||
|
|||
@GetMapping("/preview/{filePath}") |
|||
public void preview(@PathVariable String filePath, HttpServletResponse response) { |
|||
fileService.previewFile(filePath, response); |
|||
} |
|||
} |
|||
@ -0,0 +1,21 @@ |
|||
package com.coin.system.service; |
|||
|
|||
import org.springframework.web.multipart.MultipartFile; |
|||
import javax.servlet.http.HttpServletResponse; |
|||
|
|||
public interface FileService { |
|||
|
|||
/** |
|||
* 上传文件 |
|||
* @param file 文件 |
|||
* @return 文件访问路径 |
|||
*/ |
|||
String uploadFile(MultipartFile file); |
|||
|
|||
/** |
|||
* 预览文件 |
|||
* @param filePath 文件路径 |
|||
* @param response HTTP响应 |
|||
*/ |
|||
void previewFile(String filePath, HttpServletResponse response); |
|||
} |
|||
@ -0,0 +1,95 @@ |
|||
package com.coin.system.service.impl; |
|||
|
|||
import com.coin.system.service.FileService; |
|||
import com.coin.common.exception.ServiceException; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.apache.commons.io.FilenameUtils; |
|||
import org.apache.commons.lang3.StringUtils; |
|||
import org.springframework.beans.factory.annotation.Value; |
|||
import org.springframework.stereotype.Service; |
|||
import org.springframework.web.multipart.MultipartFile; |
|||
|
|||
import javax.servlet.http.HttpServletResponse; |
|||
import java.io.*; |
|||
import java.nio.file.Files; |
|||
import java.nio.file.Path; |
|||
import java.nio.file.Paths; |
|||
import java.time.LocalDate; |
|||
import java.time.format.DateTimeFormatter; |
|||
import java.util.UUID; |
|||
|
|||
@Slf4j |
|||
@Service |
|||
public class FileServiceImpl implements FileService { |
|||
|
|||
@Value("${file.upload.path}") |
|||
private String uploadPath; |
|||
|
|||
@Value("${file.upload.maxSize}") |
|||
private long maxFileSize; // 默认最大10MB
|
|||
|
|||
@Override |
|||
public String uploadFile(MultipartFile file) { |
|||
if (file.isEmpty()) { |
|||
throw new ServiceException("上传文件不能为空"); |
|||
} |
|||
|
|||
// 检查文件大小
|
|||
if (file.getSize() > maxFileSize * 1024 * 1024) { |
|||
throw new ServiceException("文件大小超过限制"); |
|||
} |
|||
|
|||
try { |
|||
// 生成文件存储路径（按日期分类）
|
|||
String datePath = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyy/MM/dd")); |
|||
Path uploadDir = Paths.get(uploadPath, datePath); |
|||
Files.createDirectories(uploadDir); |
|||
|
|||
// 生成唯一文件名
|
|||
String originalFilename = file.getOriginalFilename(); |
|||
String extension = FilenameUtils.getExtension(originalFilename); |
|||
String newFileName = UUID.randomUUID().toString() + "." + extension; |
|||
Path targetPath = uploadDir.resolve(newFileName); |
|||
|
|||
// 使用NIO进行文件写入
|
|||
Files.copy(file.getInputStream(), targetPath); |
|||
|
|||
// 返回相对路径
|
|||
return datePath + "/" + newFileName; |
|||
} catch (IOException e) { |
|||
log.error("文件上传失败", e); |
|||
throw new ServiceException("文件上传失败"); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void previewFile(String filePath, HttpServletResponse response) { |
|||
if (StringUtils.isEmpty(filePath)) { |
|||
throw new ServiceException("文件路径不能为空"); |
|||
} |
|||
|
|||
Path path = Paths.get(uploadPath, filePath); |
|||
if (!Files.exists(path)) { |
|||
throw new ServiceException("文件不存在"); |
|||
} |
|||
|
|||
try { |
|||
String contentType = Files.probeContentType(path); |
|||
response.setContentType(contentType); |
|||
|
|||
// 使用缓冲流提高性能
|
|||
try (InputStream in = new BufferedInputStream(Files.newInputStream(path)); |
|||
OutputStream out = new BufferedOutputStream(response.getOutputStream())) { |
|||
byte[] buffer = new byte[4096]; |
|||
int bytesRead; |
|||
while ((bytesRead = in.read(buffer)) != -1) { |
|||
out.write(buffer, 0, bytesRead); |
|||
} |
|||
out.flush(); |
|||
} |
|||
} catch (IOException e) { |
|||
log.error("文件预览失败", e); |
|||
throw new ServiceException("文件预览失败"); |
|||
} |
|||
} |
|||
} |
|||
@ -1,4 +1,4 @@ |
|||
package com.coin.app.domain.vo; |
|||
package com.coin.system.vo; |
|||
|
|||
import lombok.AllArgsConstructor; |
|||
import lombok.Data; |
|||
Loading…
Reference in new issue