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();
}
}