日期:2014-05-16  浏览次数:20507 次

MongoDB Java Driver 源码分析(11):GridFS 类
  GridFS 类在 com.mongodb.gridfs 包中的地位类似于 Mongo 类在 com.mongodb 包的作用,主要是提供对外的接口。
  它的各个方法是分别委托给不同的类执行的,尤其是 DBColletion 类的实例 _filesCollection  和 _chunkCollection 。

  我们先看看它的构造函数:
    /**
     * 在数据库中为指定的 bucket 创建 GridFS 实例
     * bucket 可以认为是标识符,用于区分不同的 GridFS 的实例,类似于传统数据库中的不同表
     * 默认的 bucket 是 "fs"
     */
    public GridFS(DB db, String bucket) {
        // DB 实例
        _db = db;

        // bucket 名称
        _bucketName = bucket;

       // files 集合,是 DBCollection 类的实例,用于存放文件的基本信息
        _filesCollection = _db.getCollection( _bucketName + ".files" );

        // chunks 集合,是 DBCollection 类的实例,用于存放文件的数据
        _chunkCollection = _db.getCollection( _bucketName + ".chunks" );

        // 如果数据集合较小的话,创建索引
        if (_filesCollection.count() < 1000)
            _filesCollection.ensureIndex( BasicDBObjectBuilder.start().add( "filename" , 1 ).add( "uploadDate" , 1 ).get() );
        if (_chunkCollection.count() < 1000)
            _chunkCollection.ensureIndex( BasicDBObjectBuilder.start().add( "files_id" , 1 ).add( "n" , 1 ).get() );

        // 设置 files 集合中存放的数据类型,这里指定为 GridFSDBFile
        _filesCollection.setObjectClass( GridFSDBFile.class );
    }

  可以看到,在构造函数中为属性 _db,_filesCollection ,_chunkCollection 进行了赋值,后续的操作都需要委托给这些属性。

_db 的相关操作 (数据库级别的操作)

    // 获取该 GRidFS 实例所属的 DB 实例
    public DB getDB(){
        return _db;
    }

_filesCollection 的相关操作(针对文件基本信息的操作)

    // 获取用于遍历文件列表的 DBCursor,按文件名称升序排列
    public DBCursor getFileList(){
        return _filesCollection.find().sort(new BasicDBObject("filename",1));
    }

   // 根据指定的query,获取用于遍历文件列表的 DBCursor,按文件名称升序排列
   public DBCursor getFileList( DBObject query ){
        return _filesCollection.find( query ).sort(new BasicDBObject("filename",1));
    }

    // 根据特定条件获取文件列表
    public List<GridFSDBFile> find( DBObject query ){
        List<GridFSDBFile> files = new ArrayList<GridFSDBFile>();

        DBCursor c = _filesCollection.find( query );
        while ( c.hasNext() ){
            files.add( _fix( c.next() ) );
        }
        return files;
    }

    // 对通过 find 方法返回的结果进行处理
    private GridFSDBFile _fix( Object o ){
        if ( o == null )
            return null;

        // 包含的数据类型不是 GridFSDBFile,抛异常
        if ( ! ( o instanceof GridFSDBFile ) )
            throw new RuntimeException( "somehow didn't get a GridFSDBFile" );

        // 设置属性 GridFSDBFile._fs ,指向 GridFS 实例本身
        GridFSDBFile f = (GridFSDBFile)o;
        f._fs = this;
        return f;
    }


_chunkCollection 的相关操作(针对文件数据的操作)

    // 根据指定的 OBjectId 删除文件
    // 需要同时删除文件基本信息和文件数据
    public void remove( ObjectId id ){
        _filesCollection.remove( new BasicDBObject( "_id" , id ) );
        _chunkCollection.remove( new BasicDBObject( "files_id" , id ) );
    }


GridFSDBFile 和 GridFSInputFile 的相关操作


  GridFSDBFile 和 GridFSInputFile 分别表示从数据库中读取的文件和将要保存到数据库中的文件。GridFS 类的部分操作是借助这两个类的实例实现的。

    // 根据特定查询删除文件
    // 调用了 GridFSDBFile.remove 方法
    public void remove( DBObject query ){
        for ( GridFSDBFile f : find( query ) ){
            f.remove();
        }
    }

    // 根据指定名称创建文件
    // 返回的是 GridFSInputFile 类的实例
    // 可以通过 GridFSInputFile 类的实例的 save 方法保存修改
    public GridFSInputFile createFile(String filename) {
        return new GridFSInputFile( this , filename );