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 find( DBObject query ){
        List files = new ArrayList();

        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 );
    }