温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

如何使用Springboot整合GridFS实现文件操作

发布时间:2021-10-23 13:41:04 来源:亿速云 阅读:163 作者:iii 栏目:开发技术

本篇内容介绍了“如何使用Springboot整合GridFS实现文件操作”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!

目录
  • GridFsOperations,实现GridFS文件上传下载删除

    • 上传下载删除功能实现

  • 测试

    • 上传

    • 下载

    • 删除

GridFsOperations,实现GridFS文件上传下载删除

最近学习GridFS,想用它整合springboot弄个文件的上传下载。

网上查到的很多资料都是使用GridFsTemplate,还有GridFSBucket来实现的,需要各种额外配置Bean。但是看了spring-data-mongodb的官方文档,以及示例代码,他们只用到了GridFsOperations,无需其他任何配置。

然后就用GridFsOperations写了个文件上传下载的demo,用起来还是很方便的,给大家个参考。

上传下载删除功能实现

pom.xml

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>

application.properties

#文件上传下载配置
spring.servlet.multipart.max-file-size=1024MB
spring.servlet.multipart.max-request-size=1024MB

FileController

package com.example.tryRedis.controller;
import static org.springframework.data.mongodb.core.query.Query.*;
import static org.springframework.data.mongodb.gridfs.GridFsCriteria.*;
import com.mongodb.client.gridfs.model.GridFSFile;
import io.swagger.v3.oas.annotations.Parameter;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.gridfs.GridFsOperations;
import org.springframework.data.mongodb.gridfs.GridFsResource;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.HashMap;
import java.util.Map;
@RestController
@RequestMapping("/file")
public class FileController {
    @Autowired
    GridFsOperations gridFsOperations;
    //上传文件
    @PostMapping(value = "/upload",consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
    public Map<String , ObjectId> upload(@Parameter @RequestPart(value = "file") MultipartFile file){
        //开始时间
        long begin = System.nanoTime();
        Map<String,ObjectId> map = new HashMap<>();
        try{
            InputStream streamForUpload = file.getInputStream();
            ObjectId objectId = gridFsOperations.store(streamForUpload,file.getOriginalFilename(),file.getContentType());
            //上传结束
            long end = System.nanoTime();
            long time = end-begin;
            System.out.println("本次上传共耗时: "+ time);
            System.out.println("上传成功!文件名: "+file.getOriginalFilename()+". 文件ID: "+objectId);
            map.put(file.getOriginalFilename(),objectId);
        }catch (Exception e){
            e.printStackTrace();
        }
        return map;
    }
    //查询并下载文件
    @GetMapping("/download")
    public String download(String filename, HttpServletResponse response) throws IOException {
        //开始时间
        long begin = System.nanoTime();
        //查询文件
        GridFSFile result  = gridFsOperations.findOne(query(whereFilename().is(filename)));
        GridFsResource gridFsResource= gridFsOperations.getResource(result);
        String contentType = gridFsResource.getContentType();
        System.out.println("contentType: "+contentType);
        System.out.println("filename: "+gridFsResource.getFilename());
        response.reset();
        response.setContentType(contentType);
        //注意: 如果没有下面这行设置header, 结果会将文件的内容作为响应的 body 直接输出在页面上, 而不是下载文件
        response.setHeader("Content-Disposition","attachment;filename="+filename);  //指定下载文件名
        ServletOutputStream outputStream = response.getOutputStream();
        InputStream is = gridFsResource.getInputStream();
        byte[] bytes = new byte[1024];
        int len = 0;
        while ((len=is.read(bytes))!=-1){
            outputStream.write(bytes,0,len);
        }
        is.close();
        outputStream.close();
        //下载结束
        long end = System.nanoTime();
        long time = end-begin;
        System.out.println("本次下载共耗时: "+ time);
        return contentType;
    }
    @DeleteMapping("/delete")
    public String deleteFile(@Parameter @RequestParam("filename") String filename){
        gridFsOperations.delete(query(whereFilename().is(filename)));
        return "delete success";
    }
}

测试

上传

如何使用Springboot整合GridFS实现文件操作 如何使用Springboot整合GridFS实现文件操作

下载

红色圈内点击download就可以下载啦。或者在地址栏直接输入localhost:8080/file/download?filename=todo.txt 也可以直接下载文件(这里的todo.txt是我测试的文件,你们填自己上传的文件名,不要忘了加上后缀名!)

如何使用Springboot整合GridFS实现文件操作

删除

如何使用Springboot整合GridFS实现文件操作

上面这些上传删除功能测试的时候,大家也可以结合mongodb的数据库去看看。

如何使用Springboot整合GridFS实现文件操作

“如何使用Springboot整合GridFS实现文件操作”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注亿速云网站,小编将为大家输出更多高质量的实用文章!

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI