读写mongo GridFs中的文件

mongodb里面自带有一个分布式文件系统gridFs,它是以块的方式来存储文件的,一般的存储都够用了,国内一个使用例子是视觉中国使用它来进行上亿数据级的图片存储,可以看出这套文件系统还是挺强大的。下面介绍下如何用spring-data-mongodb来对其进行操作,其实spring-data-mongodb并没有对gridfs进行再次封装,我们只能自己根据需要简单封装下接口,mongodb java api中操作gridfs也是很简单的,1.得到DB对象,2.new 一个GridFSInputFile对象,这个对象相当于一个文件记录,包含文件和与这个文件相关的信息。3.调用save方法保存。读取文件时可以根据文件名和id来对文件进行查询,查询出来一个GridFSDBFile 对象,再把这个对象输出到流或文件中。

先注入MongoDbFactory

1
2
@Autowired
  private MongoDbFactory mongoDbFactory;

获得DB对象

1
DB db = mongoDbFactory.getDb();

保存文件(其中FS_NAME相当于gridfs文件系统的表名,可以有多个)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
public boolean saveFile(File file, String fileName, String contentType,
      DBObject metaData) {
 
    GridFSInputFile gfsInput;
    try {   
      gfsInput = new GridFS(db, FS_NAME).createFile(file);
      gfsInput.setFilename(fileName);
      gfsInput.setContentType(contentType);
      gfsInput.setMetaData(metaData);
      gfsInput.save();
    } catch (IOException e) {
      log.error(e, e);
      return false;
    }
    return true;
  }

通过文件名读取文件

1
2
3
4
5
6
7
8
9
10
public GridFSDBFile findFileByName(String fileName){
    GridFSDBFile gfsFile ;
    try {      
      gfsFile = new GridFS(db, FS_NAME).findOne(fileName);
    } catch (Exception e) {
      log.error(e, e);
      return null;
    }
    return gfsFile;
  }

通过id读取文件

1
2
3
4
5
6
7
8
9
10
public GridFSDBFile findFileById(String id){
    GridFSDBFile gfsFile ;
    try {      
      gfsFile = new GridFS(db, FS_NAME).find(new ObjectId(id));
    } catch (Exception e) {
      log.error(e, e);
      return null;
    }
    return gfsFile;
  }

输出文件到OutputStream

1
2
3
4
5
6
7
8
9
private boolean writeToOutputStream(OutputStream out, GridFSDBFile gfsFile) {
    try {     
      gfsFile.writeTo(out);
    } catch (IOException e) {
      log.error(e, e);
      return false;
    }
    return true;
  }

 

猜你喜欢

转载自k1121.iteye.com/blog/2083543