SpringBoot+Vue实现简单的文件上传
1 环境 SpringBoot 3.2.1，Vue 2，ElementUI
2 问题 前两篇文章，我们上传了txt、Excel文件，其实文件类型有很多种，如果我们的upload组件没有上传文件类型的限制，那么同一个接口会上传多种文件，而类型不一样，其处理的方式也不一样。一种方式是写if判断，我们判断其类型，然后做特定的处理，那样代码会有大量if判断；另外一个可以使用策略模式来处理，简单来说就是定义一个接口，多个实现类，不同的文件类型使用不同的实现类。这里对策略模式不做详细的介绍，有兴趣的可以自行上网查阅。直接上代码。
接口
java
public interface FileService {
Result upload(MultipartFile file);
}关于txt文件的处理类
java
@Service("txt")
public class TextServiceImpl implements FileService {
@Override
public Result upload(@RequestParam("file") MultipartFile file) {
if (file.isEmpty()) {
return Result.builder().code(500).msg("上传失败！").build();
}
try (BufferedReader reader = new BufferedReader(new InputStreamReader(file.getInputStream()))) {
String line;
while ((line = reader.readLine()) != null) {
// 在这里处理读取到的每一行内容
System.out.println(line);
}
} catch (IOException e) {
e.printStackTrace();
}
return Result.builder().code(200).msg("上传成功").build();
}
}关于xls文件的处理
java
@Service("xls")
public class XLSServiceImpl implements FileService {
@Override
public Result upload(MultipartFile file) {
if (file.isEmpty()) {
return Result.builder().code(500).msg("上传失败！").build();
}
try (Workbook workbook = new HSSFWorkbook(file.getInputStream())) {
//获取第一个sheet页
Sheet sheet = workbook.getSheetAt(0);
//遍历每行
for (Row row : sheet) {
//遍历每个单元格
for (Cell cell : row) {
System.out.print(cell.getStringCellValue() + " ");
}
System.out.println();
}
} catch (IOException e) {
e.printStackTrace();
}
return Result.builder().code(200).msg("上传成功").build();
}
}关于xlsx文件的处理
java
@Service("xlsx")
public class XLSXServiceImpl implements FileService {
@Override
public Result upload(MultipartFile file) {
if (file.isEmpty()) {
return Result.builder().code(500).msg("上传失败！").build();
}
try (Workbook workbook = new XSSFWorkbook(file.getInputStream())) {
//获取第一个sheet页
Sheet sheet = workbook.getSheetAt(0);
//遍历每行
for (Row row : sheet) {
//遍历每个单元格
for (Cell cell : row) {
System.out.print(cell.getStringCellValue() + " ");
}
System.out.println();
}
} catch (IOException e) {
e.printStackTrace();
}
return Result.builder().code(200).msg("上传成功").build();
}
}入口
java
@RestController
public class FileController {
//处理实现类的集合，spring中，对于同一类bean不同的别名是可以注入到一个map中或者list中的
//key为实现类的别名，value就是bean
@Autowired
private Map<String, FileService> fileServiceMap;
@RequestMapping("/fileUpload")
private Result fileUpload(@RequestParam("file") MultipartFile file) throws IOException {
if (file.isEmpty()) {
return Result.builder().code(500).msg("上传失败！").build();
}
String fileName = file.getOriginalFilename();
//这里我们获取到后缀名，然后就可以在map中找到对应的实现类
FileService fileService = fileServiceMap.get(getSuffix(fileName));
if(fileService == null){
return Result.builder().code(500).msg("没有合适的处理器处理上传的文件！").build();
}
fileService.upload(file);
return Result.builder().code(200).msg("上传成功").build();
}
//获取文件的后缀名
private static String getSuffix(String filePath) {
int dotIndex = filePath.lastIndexOf('.');
if (dotIndex > 0 && dotIndex < filePath.length() - 1) {
return filePath.substring(dotIndex + 1);
}
return "";
}
}附：
1 @Service("txt")就是给实现类bean取个别名，然后方便我们后续通过文件的后缀名找到对应的实现类。
2 HSSF 和 XSSF 主要有以下区别：
HSSF 是用于操作 Excel 2003 及以前版本（.xls 文件）的类库。它基于二进制格式。
XSSF 则是用于操作 Excel 2007 及以后版本（.xlsx 文件）的类库，这些版本采用的是 Open XML 格式。XSSF 能够处理更大、更复杂的工作簿，并且在处理一些新特性和功能上更具优势。