通用的SpringBoot集成的文件上传与下载

废话不多说--直接看代码

controller

package com.webank.wedatasphere.qualitis.controller.thymeleaf;

import com.webank.wedatasphere.qualitis.handler.CommonExcelService;

import com.webank.wedatasphere.qualitis.project.dao.repository.ProjectFileRepository;

import com.webank.wedatasphere.qualitis.project.dao.repository.ProjectMenuRepository;

import com.webank.wedatasphere.qualitis.project.entity.Project;

import com.webank.wedatasphere.qualitis.project.entity.ProjectFile;

import com.webank.wedatasphere.qualitis.project.entity.ProjectMenu;

import com.webank.wedatasphere.qualitis.response.BiaoZunExcelDTO;

import com.webank.wedatasphere.qualitis.response.Resp;

import com.webank.wedatasphere.qualitis.util.CreateTreeUtils;

import com.webank.wedatasphere.qualitis.util.LocalTimeUtils;

import org.apache.commons.io.FileUtils;

import org.apache.commons.lang3.StringUtils;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.beans.factory.annotation.Value;

import org.springframework.core.env.Environment;

import org.springframework.core.io.FileSystemResource;

import org.springframework.core.io.Resource;

import org.springframework.http.HttpHeaders;

import org.springframework.http.MediaType;

import org.springframework.http.ResponseEntity;

import org.springframework.web.bind.annotation.*;

import org.springframework.web.multipart.MultipartFile;

import org.springframework.web.servlet.ModelAndView;

import javax.annotation.PostConstruct;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import java.io.*;

import java.net.URLEncoder;

import java.nio.file.Paths;

import java.util.ArrayList;

import java.util.List;

@RestController

public class CommonController {

    /**

     * 菜单实现

     */

    @Autowired

    private ProjectMenuRepository projectMenuRepository;

    /**

     * 菜单实现

     */

    @Autowired

    private ProjectFileRepository projectFileRepository;

    private String fileSavePath;

    @Autowired

    private Environment environment;

    @Autowired

    private CommonExcelService commonExcelService;

    @PostConstruct

    public void loadEvn() {

        // 切换代码执行环境

        String os = System.getProperty("os.name");

        if (os.toLowerCase().startsWith("win")) {

            fileSavePath = environment.getProperty("file.save.path_windows");

        } else {

            fileSavePath = environment.getProperty("file.save.path_linux");

        }

    }

    @RequestMapping(value = {"/getMenuList"}, method = RequestMethod.GET, name = "获取菜单列表·")

    @ResponseBody

    public Resp<?> getMenuList(HttpServletRequest request) {

        List<ProjectMenu> all = projectMenuRepository.findAll();

        List<ProjectMenu> tree = CreateTreeUtils.createTree(all);

        // 生成目录树

        return Resp.Ok(tree);

    }

    /**

     * 菜单目录

     *

     * @return

     */

    @RequestMapping(value = {"/menuIndex"}, method = RequestMethod.GET)

    public ModelAndView menuIndex() {

        ModelAndView modelAndView = new ModelAndView("layui-admin/index2");

        List<ProjectMenu> all = projectMenuRepository.findAll();

        List<ProjectMenu> tree = CreateTreeUtils.createTree(all);

        modelAndView.addObject("resourceGroups", tree);

        return modelAndView;

    }

    /**

     * 导入

     *

     * @param request  request

     * @param response response

     */

    @RequestMapping(value = {"/insertFile"}, method = {RequestMethod.GET, RequestMethod.POST}, name = "文件上传·")

    @ResponseBody

    public Resp<?> insertFile(HttpServletRequest request, HttpServletResponse response, @RequestParam("file") MultipartFile multipartFile) {

        String originalFilename = multipartFile.getOriginalFilename();

        if (StringUtils.isBlank(originalFilename)) {

            return Resp.error("文件名不能为空");

        }

        // 校验目录库

        String currentTime = LocalTimeUtils.getCurrentTimeSimple();

        File file = new File(fileSavePath + "/" + currentTime);

        if (!file.isDirectory()) {

            file.mkdirs();

        }

        String filePath = "";

        // 文件保留路径

        String fileName = fileSavePath + "/" + currentTime + "/" + originalFilename;

        File dest = new File(fileName);

        ProjectFile projectFile = new ProjectFile();

        try {

            // 保存本地文件

            FileUtils.copyInputStreamToFile(multipartFile.getInputStream(), dest);

            // 文件回调组装入库

            filePath = request.getScheme() + "://" + request.getServerName() + ":" + request.getServerPort() + "/" + fileName;

            // 文件上传到目录库

            projectFile.setFileName(originalFilename);

            projectFile.setFilePath(fileName);

            projectFile.setName(filePath);

            projectFile.setCreateTime(currentTime);

            projectFileRepository.save(projectFile);

        } catch (Exception e) {

            e.printStackTrace();

            return Resp.error(e.getMessage());

        }

        // 返回文件路径

        return Resp.Ok(projectFile);

    }

    /**

     * 导入数据库

     *

     * @param request  request

     * @param response response

     */

    @RequestMapping(value = {"/excelSaveDb"}, method = {RequestMethod.GET, RequestMethod.POST}, name = "文件上传·")

    @ResponseBody

    public Resp<?> excelSaveDb(HttpServletRequest request, HttpServletResponse response, @RequestParam("file") MultipartFile multipartFile) {

        String originalFilename = multipartFile.getOriginalFilename();

        if (StringUtils.isBlank(originalFilename)) {

            return Resp.error("文件名不能为空");

        }

        if (!originalFilename.contains(".xls")) {

            return Resp.error("文件格式错误:必须要是XLS或者XLSX格式");

        }

        // 校验目录库

        String currentTime = LocalTimeUtils.getCurrentTimeSimple();

        File file = new File(fileSavePath + "/" + currentTime);

        if (!file.isDirectory()) {

            file.mkdirs();

        }

        String filePath = "";

        // 文件保留路径

        String fileName = fileSavePath + "/" + currentTime + "/" + originalFilename;

        File dest = new File(fileName);

        ProjectFile projectFile = new ProjectFile();

        try {

            // 保存本地文件

            FileUtils.copyInputStreamToFile(multipartFile.getInputStream(), dest);

            // 文件回调组装入库

            filePath = request.getScheme() + "://" + request.getServerName() + ":" + request.getServerPort() + "/" + fileName;

            // 文件上传到目录库

            projectFile.setFileName(originalFilename);

            projectFile.setFilePath(fileName);

            projectFile.setName(filePath);

            projectFile.setCreateTime(currentTime);

            projectFileRepository.save(projectFile);

        } catch (Exception e) {

            e.printStackTrace();

            return Resp.error(e.getMessage());

        }

        // 读取成为Excel入库

        commonExcelService.handlerBiaoZunUpload(fileName);

        // 返回文件路径

        return Resp.Ok(projectFile);

    }

    /**

     * 导出

     *

     * @param request  request

     * @param response response

     */

    @RequestMapping(value = "/downloadFile", method = {RequestMethod.GET, RequestMethod.POST})

    @ResponseBody

    public ResponseEntity<Resource> export(HttpServletRequest request, HttpServletResponse response) throws UnsupportedEncodingException {

        String fileId = request.getParameter("fileId");

        if (StringUtils.isNotBlank(fileId)) {

            ProjectFile one = projectFileRepository.getOne(Long.parseLong(fileId));

            String filePath = one.getFilePath();

            String fileName = one.getFileName();

            // 创建FileSystemResource对象

            Resource resource = new FileSystemResource(Paths.get(filePath).toFile());

            // 设置响应头信息

            HttpHeaders headers = new HttpHeaders();

            headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=" + URLEncoder.encode(fileName, "UTF-8"));

            // 返回文件作为响应体

            return ResponseEntity.ok()

                    .headers(headers)

                    .contentType(MediaType.APPLICATION_OCTET_STREAM)

                    .body(resource);

        }

        return ResponseEntity.ok().body(null);

    }

    /**

     * 导出Excel

     *

     * @param projects  request

     * @param response response

     */

    @RequestMapping(value = "/exportExcel", method = {RequestMethod.GET, RequestMethod.POST})

    @ResponseBody

    public void exportExcel(HttpServletResponse response,@RequestBody List<Project> projects) throws IOException {

        List<BiaoZunExcelDTO> list = new ArrayList<>(projects.size());

        for (Project project : projects) {

            BiaoZunExcelDTO biaoZunExcelDTO = new BiaoZunExcelDTO();

            biaoZunExcelDTO.setSjzlbzml(project.getSjzlbzml());

            biaoZunExcelDTO.setCnName(project.getCnName());

            biaoZunExcelDTO.setCkwx(project.getCkwx());

            biaoZunExcelDTO.setSyfw(project.getSyfw());

            biaoZunExcelDTO.setFbzt(project.getFbzt());

            biaoZunExcelDTO.setSjzllx(project.getSjzllx());

            list.add(biaoZunExcelDTO);

        }

        // 导出数据位Excel

       commonExcelService.handlerBiaoZunDownload(response.getOutputStream(),list);

    }

}

Service

package com.webank.wedatasphere.qualitis.handler;

import com.alibaba.excel.ExcelReader;

import com.alibaba.excel.ExcelWriter;

import com.alibaba.excel.metadata.Sheet;

import com.alibaba.excel.support.ExcelTypeEnum;

import com.webank.wedatasphere.qualitis.project.constant.ExcelSheetName;

import com.webank.wedatasphere.qualitis.project.dao.repository.ProjectMuLuRepository;

import com.webank.wedatasphere.qualitis.project.dao.repository.ProjectRepository;

import com.webank.wedatasphere.qualitis.project.entity.ProjectMuLu;

import com.webank.wedatasphere.qualitis.project.excel.ExcelTemplateRuleByProject;

import com.webank.wedatasphere.qualitis.response.BiaoZunExcelDTO;

import com.webank.wedatasphere.qualitis.response.Resp;

import org.apache.commons.lang3.StringUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import java.io.FileInputStream;

import java.io.IOException;

import java.io.OutputStream;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

@Component

public class CommonExcelService {

    private Logger logger = LoggerFactory.getLogger(CommonExcelService.class);

    @Autowired

    private ProjectRepository repository;

    @Autowired

    private ProjectMuLuRepository projectMuLuRepository;

    /**

     * 标准的导入Excel

     *

     * @param fileSavePath

     * @return

     */

    public Resp<?> handlerBiaoZunUpload(String fileSavePath) {

        // 这里 需要指定读用哪个class去读，然后读取第一个sheet 文件流会自动关闭

        try {

            logger.info("附件上传excel===》" + fileSavePath);

            FileInputStream fileInputStream = new FileInputStream(fileSavePath);

            ExcelReader excelReader = new ExcelReader(fileInputStream, null, new ExcelBiaoZunListener(repository, projectMuLuRepository));

            List<Sheet> sheets = excelReader.getSheets();

            Sheet sheet = sheets.get(0);

            sheet.setClazz(BiaoZunExcelDTO.class);

            sheet.setHeadLineMun(1);

            excelReader.read(sheet);

            logger.info("读取完成excel===》" + fileSavePath);

        } catch (Exception e) {

            e.printStackTrace();

            return Resp.error(e.getMessage());

        }

        return Resp.Ok();

    }

    /**

     * 标准的导出Excel

     *

     * @param outputStream

     * @return

     */

    public void handlerBiaoZunDownload(OutputStream outputStream, List<BiaoZunExcelDTO> list) throws IOException {

        // 这里 需要指定读用哪个class去读，然后读取第一个sheet 文件流会自动关闭

        try {

            List<ProjectMuLu> muLuRepositoryAll = projectMuLuRepository.findAll();

            for (BiaoZunExcelDTO biaoZunExcelDTO : list) {

                for (ProjectMuLu muLu : muLuRepositoryAll) {

                    if (StringUtils.isNotBlank(biaoZunExcelDTO.getSjzlbzml()) && biaoZunExcelDTO.getSjzlbzml().equals(muLu.getId().toString())) {

                        biaoZunExcelDTO.setSjzlbzml(muLu.getTitle());

                    }

                }

            }

            ExcelWriter writer = new ExcelWriter(outputStream, ExcelTypeEnum.XLSX, true);

            Sheet templateRuleSheet = new Sheet(1, 0, BiaoZunExcelDTO.class);

            templateRuleSheet.setSheetName("导出标准");

            List<List<String>> heads = new ArrayList<>();

            List<String> head1 = new ArrayList<>();

            List<String> head2 = new ArrayList<>();

            List<String> head3 = new ArrayList<>();

            List<String> head4 = new ArrayList<>();

            List<String> head5 = new ArrayList<>();

            List<String> head6 = new ArrayList<>();

            head1.add("数据质量名称");

            head2.add("适用范围");

            head3.add("数据质量类型");

            head4.add("参考文献");

            head5.add("所属质量标准目录");

            head6.add("发布状态");

            heads.add(head1);

            heads.add(head2);

            heads.add(head3);

            heads.add(head4);

            heads.add(head5);

            heads.add(head6);

            templateRuleSheet.setHead(heads);

            writer.write(list, templateRuleSheet);

            writer.finish();

            logger.info("写入完成excel===》");

        } catch (Exception e) {

            e.printStackTrace();

        } finally {

            outputStream.close();

        }

    }

}

很老很老的easyexcel的API

package com.webank.wedatasphere.qualitis.handler;

import com.alibaba.excel.context.AnalysisContext;

import com.alibaba.excel.event.AnalysisEventListener;

import com.alibaba.fastjson.JSON;

import com.webank.wedatasphere.qualitis.project.dao.repository.ProjectMuLuRepository;

import com.webank.wedatasphere.qualitis.project.dao.repository.ProjectRepository;

import com.webank.wedatasphere.qualitis.project.entity.Project;

import com.webank.wedatasphere.qualitis.project.entity.ProjectMuLu;

import com.webank.wedatasphere.qualitis.response.BiaoZunExcelDTO;

import com.webank.wedatasphere.qualitis.util.LocalTimeUtils;

import org.apache.commons.lang3.StringUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import java.util.ArrayList;

import java.util.List;

public class ExcelBiaoZunListener extends AnalysisEventListener<BiaoZunExcelDTO> {

   private Logger logger = LoggerFactory.getLogger(this.getClass());

    /**

     * 每隔5条存储数据库，实际使用中可以100条，然后清理list ，方便内存回收

     */

    private static final int BATCH_COUNT = 100;

    /**

     * 缓存的数据

     */

    private List<BiaoZunExcelDTO> cachedDataList = new ArrayList<>(BATCH_COUNT);

    /**

     * 假设这个是一个DAO，当然有业务逻辑这个也可以是一个service。当然如果不用存储这个对象没用。

     */

    private ProjectRepository projectRepository;

    private ProjectMuLuRepository projectMuLuRepository;

    private List<ProjectMuLu> muLuRepositoryAll;

    /**

     * 如果使用了spring,请使用这个构造方法。每次创建Listener的时候需要把spring管理的类传进来

     *

     * @param projectRepository

     */

    public ExcelBiaoZunListener(ProjectRepository projectRepository,ProjectMuLuRepository projectMuLuRepository) {

        this.projectRepository = projectRepository;

        this.projectMuLuRepository = projectMuLuRepository;

        this.muLuRepositoryAll = this.projectMuLuRepository.findAll();

    }

    /**

     * 这个每一条数据解析都会来调用

     *

     * @param data    one row value. Is is same as {@link AnalysisContext#()}

     * @param context

     */

    @Override

    public void invoke(BiaoZunExcelDTO data, AnalysisContext context) {

        logger.info("解析到一条数据:{}", JSON.toJSONString(data));

        cachedDataList.add(data);

        // 达到BATCH_COUNT了，需要去存储一次数据库，防止数据几万条数据在内存，容易OOM

        if (cachedDataList.size() >= BATCH_COUNT) {

            saveData();

            // 存储完成清理 list

            cachedDataList = new ArrayList<>(BATCH_COUNT);

        }

    }

    /**

     * 所有数据解析完成了 都会来调用

     *

     * @param context

     */

    @Override

    public void doAfterAllAnalysed(AnalysisContext context) {

        // 这里也要保存数据，确保最后遗留的数据也存储到数据库

        saveData();

        logger.info("所有数据解析完成！");

    }

    /**

     * 加上存储数据库

     */

    private void saveData() {

        logger.info("{}条数据，开始存储数据库！", cachedDataList.size());

        for (BiaoZunExcelDTO biaoZunExcelDTO : cachedDataList) {

            if (StringUtils.isBlank(biaoZunExcelDTO.getCnName())){

                continue;

            }

            for (ProjectMuLu muLu : muLuRepositoryAll) {

                if (muLu.getTitle().equals(biaoZunExcelDTO.getSjzlbzml())) {

                    biaoZunExcelDTO.setSjzlbzml(muLu.getId().toString());

                    break;

                }

            }

            // 处理数据

            Project project = new Project();

            project.setCnName(biaoZunExcelDTO.getCnName());

            project.setFbzt(biaoZunExcelDTO.getFbzt());

            project.setSjzlbzml(biaoZunExcelDTO.getSjzlbzml());

            project.setCkwx(biaoZunExcelDTO.getCkwx());

            project.setSyfw(biaoZunExcelDTO.getSyfw());

            project.setSjzllx(biaoZunExcelDTO.getSjzllx());

            project.setCreateTime(LocalTimeUtils.getCurrentTime());

            project.setCreateUser("admin");

            project.setCreateUserFullName("admin(管理员)");

            project.setDescription(biaoZunExcelDTO.getCkwx());

            project.setName(biaoZunExcelDTO.getCnName());

            project.setProjectType(1);

            project.setModifyTime(LocalTimeUtils.getCurrentTime());

            projectRepository.save(project);

        }

        logger.info("存储数据库成功！");

    }

}

通用的SpringBoot集成的文件上传与下载的更多相关文章

SpringBoot下文件上传与下载的实现
原文:http://blog.csdn.net/colton_null/article/details/76696674 SpringBoot后台如何实现文件上传下载? 最近做的一个项目涉及到文件上传 ...
SpringBoot项目实现文件上传和邮件发送
前言本篇文章主要介绍的是SpringBoot项目实现文件上传和邮件发送的功能. SpringBoot 文件上传说明:如果想直接获取工程那么可以直接跳到底部,通过链接下载工程代码. 开发准备环境要 ...
SpringBoot整合阿里云OSS文件上传、下载、查看、删除
1. 开发前准备 1.1 前置知识 java基础以及SpringBoot简单基础知识即可. 1.2 环境参数开发工具:IDEA 基础环境:Maven+JDK8 所用技术:SpringBoot.lom ...
springboot+web文件上传和下载
一.首先安装mysql数据库,开启web服务器. 二.pom.xml文件依赖包配置如下: <?xml version="1.0" encoding="UTF-8&q ...
SpringBoot 文件上传、下载、设置大小
本文使用SpringBoot的版本为2.0.3.RELEASE 1.上传单个文件 ①html对应的提交表单 <form action="uploadFile" method= ...
七、springBoot 简单优雅是实现文件上传和下载
前言好久没有更新spring Boot 这个项目了.最近看了一下docker 的知识,后期打算将spring boot 和docker 结合起来.刚好最近有一个上传文件的工作呢,刚好就想起这个脚手架 ...
Springboot如何启用文件上传功能
网上的文章在写 "springboot文件上传" 时,都让你加上模版引擎,我只想说,我用不上,加模版引擎,你是觉得我脑子坏了,还是觉得我拿不动刀了. springboot如何启用文 ...
19、文件上传与下载/JavaMail邮件开发
回顾: 一. 监听器生命周期监听器 ServletRequestListener HttpSessionListener ServletContextListener 属性监听器 ServletRe ...
Http服务器实现文件上传与下载(三)
一.引言在前2章的内容基本上已经讲解了整个的大致流程.在设计Http服务器时,我设计为四层的结构,最底层是网络传输层,就是socket编程.接着一层是请求和响应层,叫做Request和Respons ...
基于hap的文件上传和下载
序言现在,绝大部分的应用程序在很多的情况下都需要使用到文件上传与下载的功能,在本文中结合hap利用spirng mvc实现文件的上传和下载,包括上传下载图片.上传下载文档.前端所使用的技术不限,本文 ...

随机推荐

IM开发基础知识补课(十)：大型IM系统有多难？万字长文，搞懂异地多活！
本文由公众号"水滴与银弹"号主Kaito原创分享,原题"搞懂异地多活,看这篇就够了",为使文章更好理解,有修订. 1.引言前几天技术群里有群友问我手上有没有I ...
架构-初识DDD
引言继上一篇BFF的文章后,我又去网上学习了一下DDD(领域驱动设计),发现一篇不错的文章,参考并写了一些自己的理解分享在这里. DDD 是什么领域驱动设计(Domain Driven Desig ...
自动化滑动极验v3示例
import random import ddddocr from playwright.sync_api import sync_playwright import time import requ ...
工程目录下.git目录过大文件清理
1. 查看当前5个大文件 git rev-list --objects --all | grep "$(git verify-pack -v .git/objects/pack/*.idx ...
15个Linux Grep命令使用实例（实用、常用）
Grep命令主要用于从文件中查找指定的字符串.首先建一个demo_file: 复制代码代码如下: $ cat demo_fileTHIS LINE IS THE 1ST UPPER CASE LIN ...
rabbitmq部署及配置与验证-copy
1. 场景描述朋友项目需要弄个测试环境,稍微帮忙了下,系统不复杂,但是需要自己安装mysql.Reids.Es.RabbitMq等,Mq主要用在同步用户信息与发送站内消息和短信上,RabbitMq以 ...
w3cschool-Docker 入门到实践
https://www.w3cschool.cn/reqsgr/ 什么是 Docker Docker 是一个开源项目,诞生于 2013 年初,最初是 dotCloud 公司内部的一个业余项目.它基于 ...
类的内部成员之五：内部类（Person.Bird bird = p.new Bird();）
/* * 类的内部成员之五:内部类 * 1. Java中允许将一个类A声明在另一个类B中,则类A就是内部类,类B称为外部类 * * 2.内部类的分类:成员内部类(静态.非静态) vs 局部内部类(方法 ...
使用Vant做移动端对图片预览ImagePreview和List的理解
使用Vant3做移动端的感受最近在使用Vant3做移动端. 感觉还可以,使用起来也简单,但是也遇见一些坑. 图片预览ImagePreview的使用在使用图片预览的时候, 我们在main.js中进行 ...
Visual Studio各个版本密钥
1.VS2012 旗舰版:YKCW6-BPFPF-BT8C9-7DCTH-QXGWC 2.VS2013 旗舰版:BWG7X-J98B3-W34RT-33B3R-JVYW9 专业版:XDM3T-W3T3 ...

通用的SpringBoot集成的文件上传与下载

废话不多说--直接看代码

controller

Service

很老很老的easyexcel的API

通用的SpringBoot集成的文件上传与下载的更多相关文章

随机推荐

热门专题