spring+mongodb 檔案上傳與下載實現
阿新 • • 發佈：2019-02-11
專案需要用的mongodb來儲存檔案
首先配置在spring配置檔案中配置mongodb
1.需要的jar包(maven依賴)
<dependency>
<groupId>org.springframework.data</groupId>
<artifactId>spring-data-mongodb</artifactId>
<version>1.7.0.RELEASE</version>
</dependency>2.spring 配置檔案配置 applicationContext-mongodb.xml
<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xmlns:context="http://www.springframework.org/schema/context"
xmlns:mongo="http://www.springframework.org/schema/data/mongo"
xmlns:p="http://www.springframework.org/schema/p" 詳細配置可參見spring官方配置檔案
然後springMvc 上傳配置檔案配置
1.需要的jar包
<dependency>
<groupId>commons-fileupload</groupId>
<artifactId>commons-fileupload</artifactId>
<version>1.3.1</version>
</dependency>
<dependency>
<groupId>org.apache.commons</groupId>
<artifactId>commons-io</artifactId>
<version>1.3.2</version>
</dependency>2.springMvc 配置
<!-- 定義檔案上傳解析器 -->
<bean id="multipartResolver"
class="org.springframework.web.multipart.commons.CommonsMultipartResolver">
<!-- 設定預設編碼 -->
<property name="defaultEncoding" value="UTF-8"></property>
<!-- 設定檔案上傳的最大值5MB，5*1024*1024 -->
<property name="maxUploadSize" value="5242880"></property>
</bean>上傳工具類封裝FileUpAndDownLoadService .java
package com.r2cf.interactiveweb.service;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.math.BigDecimal;
import java.util.Date;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.MongoDbFactory;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Component;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.gridfs.GridFS;
import com.mongodb.gridfs.GridFSDBFile;
import com.mongodb.gridfs.GridFSFile;
/**
* 檔案上傳下載工具類
* @author zhangpeng
*
*/
/**
* @author zhangpeng
*
*/
@Service
public class FileUpAndDownLoadService {
@Autowired
private MongoTemplate mongoTemplate;
/**
* 獲取mongodb資料庫檔案操作物件
* @return
*/
private GridFS getGridFS(){
DB db = mongoTemplate.getDb();
GridFS gridFS= new GridFS(db,"fs");
return gridFS;
}
/**
* 檔案上傳
* @param request
* @return
* @throws IOException
*/
public void upload(HttpServletRequest request,BigDecimal userId) throws IOException{
GridFS gridFS= getGridFS();
GridFSFile file = null;
MultipartHttpServletRequest mRequest = (MultipartHttpServletRequest) request;
Map<String, MultipartFile> fileMap = mRequest.getFileMap();
Iterator<Map.Entry<String, MultipartFile>> it = fileMap.entrySet().iterator();
while(it.hasNext()){
Map.Entry<String, MultipartFile> entry = it.next();
MultipartFile mFile = entry.getValue();//獲取檔案
file = gridFS.createFile(mFile.getInputStream());// 建立gridfs檔案
file.put("filename", mFile.getOriginalFilename());
file.put("userId", userId.toString());
file.put("uploadDate", new Date());
file.put("contentType",mFile.getContentType());
file.save();
}
}
/**
* 檢視檔案
* @param params k-v 搜尋條件
* @return
*/
public List<GridFSDBFile> viewFiles(Map<String,String> params){
GridFS gridFS= getGridFS();
BasicDBObject basicDBObject = new BasicDBObject();
for(Map.Entry<String, String> entry:params.entrySet()){
String key= entry.getKey();
String value=entry.getValue();
basicDBObject.append(key, value);
}
List<GridFSDBFile> list = gridFS.find(basicDBObject);
return list;
}
/**
* 檔案檢視與下載
* @param id 檔案唯一標示id
* @param response
* @throws IOException
*/
public void downLoad(String id,HttpServletResponse response) throws IOException{
GridFS gridFS= getGridFS();
ObjectId objId = new ObjectId(id);
GridFSDBFile gridFSDBFile =(GridFSDBFile)gridFS.findOne(objId);
if (gridFSDBFile != null) {
OutputStream sos = response.getOutputStream();
response.setContentType("application/octet-stream");
// 獲取原檔名
String name = (String) gridFSDBFile.get("filename");
String fileName = new String(name.getBytes("utf-8"), "ISO8859-1");
// 設定下載檔名
response.addHeader("Content-Disposition", "attachment; filename=\""
+ fileName + "\"");
// 向客戶端輸出檔案
gridFSDBFile.writeTo(sos);
sos.flush();
sos.close();
}
}
/**
* 單檔案刪除
* @param id 檔案唯一標識
* @return
*/
public String delete(String id){
GridFS gridFS= getGridFS();
ObjectId objId = new ObjectId(id);
gridFS.remove(objId);
return "success";
}
}