java mongodb 文件上传,下载,删除方法

package com.stylefeng.guns.config.mongo;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import org.apache.log4j.Logger;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Service;
import com.mongodb.DB;
import com.mongodb.gridfs.GridFS;
import com.mongodb.gridfs.GridFSDBFile;
import com.mongodb.gridfs.GridFSInputFile;
/**
 * 
 * 类描述:mongodb文件处理方法类
 * @author lh
 * 2018年1月25日 下午2:59:08
 */

@Service
public class MongoFileService {
private static Logger logger = Logger.getLogger(MongoFileService.class);
@Autowired
private MongoTemplate mongoTemplate;
/**
* 上传
* @param file 
* @param fileName 文件名
* @param type 表名
* @return mongoid
*/
public String saveFile(File file, String fileName, MongoDbTypeEnum type) {
String id = "";
FileInputStream fileInputStream = null;
try {
GridFS fs = getGridFS(type);
GridFSInputFile mongofile = null;
fileInputStream = new FileInputStream(file);
mongofile = fs.createFile(fileInputStream, fileName);
mongofile.setFilename(fileName);
mongofile.put("lastModified", file.lastModified());
mongofile.save();
id = this.generalDbKey(mongofile.getId().toString(), type);
} catch (Exception e) {
logger.error("saveFile  error=======>" + e);
} finally {
if (!(null == fileInputStream)) {
try {
fileInputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
return id;
}

/**
*  下载文件到指定路径
* @param type 表名
* @param id 主键
* @param Path 下载路径
* @return
*/
public int downloadFile(MongoDbTypeEnum type, String id, File Path) {
//result=1代表成功,0失败
int result=1;
try {
// 获取fs的根节点
GridFS fs = getGridFS(type);
// 解析id
String[] key = parseDbKey(id);
// 取objectId
ObjectId objectId = new ObjectId(key[0]);
// 取表名
// String CollectionsName=key[1];
// 通过id查询
GridFSDBFile file = fs.find(objectId);
// 通过文件名称查询
// GridFSDBFile file = fs.findOne("lvxing");
// 输出文件
file.writeTo(Path + File.separator + file.getFilename());
} catch (Exception e) {
result=0;
logger.error("downloadFile  error=======>" + e);
}
return result;
}
// 删除文件
public int deleteFile(MongoDbTypeEnum type, String id) {
//result=1代表成功,0失败
int result=1;
try {
// 获取fs的根节点
GridFS fs = getGridFS(type);
// 解析id
String[] key = parseDbKey(id);
// 取objectId
ObjectId objectId = new ObjectId(key[0]);
// 取表名
// String CollectionsName=key[1];
// 删除文件
fs.remove(objectId);
} catch (Exception e) {
result=0;
logger.error("deleteFile  error=======>" + e);
}
return result;
}
/**
* 获取数据库.
* 
* @param type
* @return
*/
private GridFS getGridFS(MongoDbTypeEnum type) {
DB db = mongoTemplate.getDb();
String dbName = db.getName();
GridFS gridFS = new GridFS(db, type.toString());
return gridFS;
}

/**
* 生成标记数据库的dbkey.
* 
* @param objId
* @param type
* @return
*/
private String generalDbKey(String objId, MongoDbTypeEnum type) {
return objId + "-" + type;
}

/**
* 解析dbkey.
* 
* @param dbKey
* @return
*/
private String[] parseDbKey(String dbKey) {
return dbKey.split("-");
}

}

/**
 * 
 * 类描述:MongoDb数据库分类枚举.
 * @author lh
 * 2018年1月19日 下午3:11:24
 */
public enum MongoDbTypeEnum {
//测试用的表
test

}

如果你是湖南的 欢迎加入 湖南人在深圳-Java群:557651502

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值