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