springboot2.0x集成mongodb提供GridFsTemplate类实现文件上传下载

1.关于springboot2.0x集成MongoDB的步骤这里就不写了，博客搜索有非常详细的流程讲解
直接进入controller层进行操作
2.文件上传和下载

package com.yangfeng.controller.file;
 
import com.jp.framework.common.model.ServiceResultHelper;
import com.jp.framework.common.util.Constant;
import com.jp.zpzc.entity.customized.FileInfoAO;
import com.jp.zpzc.entity.customized.UserAO;
import com.mongodb.gridfs.GridFSDBFile;
import com.mongodb.gridfs.GridFSFile;
import org.apache.shiro.SecurityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.gridfs.GridFsTemplate;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
 
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.Part;
import java.io.InputStream;
 
 
/**
 * 文件上传下载
 *
 * @author yangfeng
 * @date 2018-06-04 12:57
 **/
@Controller
@RequestMapping("/file")
public class FileController {
 
 
    // 获得SpringBoot提供的mongodb的GridFS对象
    @Autowired
    private GridFsTemplate gridFsTemplate;
	@Resource
 	private MongoDbFactory mongoDbFactory;
    /**
     * 文件上传
     *
     * @param request
     * @return
     */
    @RequestMapping(value = "/uploadFile", method = RequestMethod.POST)
    @ResponseBody
    public Object uploadFile(@RequestParam（“file”）MultipartFile multiportFile) throws Exception {
   		 //获取当前用户（shiro权限框架的方法）
        User user = (User) SecurityUtils.getSubject().getPrincipal();
        // 获得提交的文件名
        String fileName = multiportFile.getOriginalFileName();
        // 获得文件输入流
        InputStream ins = multiportFile.getInputStream();
        // 获得文件类型
        String contentType = multiportFile.getContentType();
        // 将文件存储到mongodb中,mongodb 将会返回这个文件的具体信息
       ObjectId objectId = gridFsTemplate.store(ins, fileName, contentType);
       //将文件信息保存到关系型数据库中进行维护
        FileInfoAO fileInfo = new FileInfoAO();
        fileInfo.setContentType(contentType);
        fileInfo.setFileName(fileName);
        fileInfo.setLastUpdateBy(user != null ? user.getId() : null);
        fileInfo.setMongoFileId(objectId.toString());
        return ServiceResultHelper.genResultWithSuccess(fileInfo);
    }
 
    /**
     * 下载
     *
     * @param fileId   文件id
     * @param response
     * @return
     */
    @RequestMapping(value = "/downloadFile")
    public void downloadFile(@RequestParam(name = "file_id") String fileId, HttpServletRequest request, HttpServletResponse response) throws Exception {
        Query query = Query.query(Criteria.where("_id").is(fileId));
        // 查询单个文件
        GridFsFile gfsFile = gridFsTemplate.findOne(query);
      GridFsResource gridFsResource=new GridFsResource(gfsFile,        		 GridFsBuckets.create(mongoDbFactory.getDb)).openDownloadStream(gfsFile.getObjectId()));
        String fileName = gfsfile.getFilename().replace(",", "");
        //处理中文文件名乱码
        if (request.getHeader("User-Agent").toUpperCase().contains("MSIE") ||
                request.getHeader("User-Agent").toUpperCase().contains("TRIDENT")
                || request.getHeader("User-Agent").toUpperCase().contains("EDGE")) {
            fileName = java.net.URLEncoder.encode(fileName, "UTF-8");
        } else {
            //非IE浏览器的处理：
            fileName = new String(fileName.getBytes("UTF-8"), "ISO-8859-1");
        }
        // 通知浏览器进行文件下载
        response.setHeader("Content-Disposition", "attachment;filename=\"" + fileName + "\"");
       IOUtils.copy(gridFsResource.getInputStream(),response.getOutputStream());
    }
 
    /**
     * 删除文件
     *
     * @param fileId
     * @return
     */
    @RequestMapping(value = "/deleteFile", method = RequestMethod.POST)
    @ResponseBody
    public Object deleteFile(@RequestParam(name = "file_id") String fileId) {
        Query query = Query.query(Criteria.where("_id").is(fileId));
        // 查询单个文件
        GridFSFile gfsfile = gridFsTemplate.findOne(query);
        if (gfsfile == null) {
            return ServiceResultHelper.genResultWithFaild(Constant.ErrorCode.FILE_NOT_EXIST_ERROR_MSG, Constant.ErrorCode.FILE_NOT_EXIST_ERROR);
        }
        gridFsTemplate.delete(query);
        return ServiceResultHelper.genResultWithSuccess();
    }
    }