SpringBoot+Vue实现简单的文件上传(策略模式)

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 能够处理更大、更复杂的工作簿,并且在处理一些新特性和功能上更具优势。

相关推荐
灰海1 小时前
vue中通过heatmap.js实现热力图(多个热力点)热区展示(带鼠标移入弹窗)
前端·javascript·vue.js·heatmap·heatmapjs
哈喽姥爷1 小时前
Spring Boot---自动配置原理和自定义Starter
java·spring boot·后端·自定义starter·自动配置原理
小蒜学长2 小时前
基于springboot 校园餐厅预约点餐微信小程序的设计与实现(代码+数据库+LW)
数据库·spring boot·微信小程序
码上暴富3 小时前
vue2迁移到vite[保姆级教程]
前端·javascript·vue.js
老华带你飞3 小时前
考研论坛平台|考研论坛小程序系统|基于java和微信小程序的考研论坛平台小程序设计与实现(源码+数据库+文档)
java·vue.js·spring boot·考研·小程序·毕设·考研论坛平台小程序
YAY_tyy3 小时前
基于 Vue3 + VueOffice 的多格式文档预览组件实现(支持 PDF/Word/Excel/PPT)
前端·javascript·vue.js·pdf·word·excel
dazhong20124 小时前
Spring Boot 项目新增 Module 完整指南
java·spring boot·后端
xrkhy4 小时前
SpringBoot之日志处理(logback和AOP记录操作日志)
java·spring boot·logback
上官浩仁6 小时前
springboot synchronized 本地锁入门与实战
java·spring boot·spring
m0_748461396 小时前
Spring Boot + Vue 项目中使用 Redis 分布式锁案例
vue.js·spring boot·redis