cc.acs.mongofs.gridfs.GridFSFile.java Source code

Introduction

Here is the source code for cc.acs.mongofs.gridfs.GridFSFile.java

Source

// GridFSFile.java

/**
 *      Copyright (C) 2008 10gen Inc.
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 */

package cc.acs.mongofs.gridfs;

import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.BasicDBObjectBuilder;
import com.mongodb.BasicDBObject;
import com.mongodb.DBCursor;
import com.mongodb.MongoException;

import com.mongodb.util.JSON;

import org.bson.*;

import java.io.*;
import java.util.*;

public abstract class GridFSFile implements DBObject {

    // ------------------------------
    // --------- db           -------
    // ------------------------------

    public void save() {
        if (_fs == null)
            throw new MongoException("need _fs");
        _fs._filesCollection.save(this);
    }

    public void validate() {
    }

    public int numChunks() {
        double d = _length;
        d = d / _chunkSize;
        return (int) Math.ceil(d);
    }

    // ------------------------------
    // --------- getters      -------
    // ------------------------------

    public Object getId() {
        return _id;
    }

    public String getFilename() {
        return _filename;
    }

    public String getContentType() {
        return _contentType;
    }

    public long getLength() {
        return _length;
    }

    public long getChunkSize() {
        return _chunkSize;
    }

    public Date getUploadDate() {
        return _uploadDate;
    }

    public List<String> getAliases() {
        return (List<String>) _metadata.get("aliases");
    }

    public DBObject getMetaData() {
        return (DBObject) _metadata.get("metadata");
    }

    public String getMD5() {
        return _md5;
    }

    // ------------------------------
    // --------- DBOBject methods ---
    // ------------------------------

    public Object put(String key, Object v) {
        if (key == null)
            throw new RuntimeException("key should never be null");
        else if (key.equals("_id"))
            _id = v;
        else if (key.equals("filename"))
            _filename = v == null ? null : v.toString();
        else if (key.equals("contentType"))
            _contentType = (String) v;
        else if (key.equals("length"))
            _length = ((Number) v).longValue();
        else if (key.equals("chunkSize"))
            _chunkSize = ((Number) v).longValue();
        else if (key.equals("uploadDate"))
            _uploadDate = (Date) v;
        else if (key.equals("md5"))
            _md5 = (String) v;
        else
            _metadata.put(key, v);
        return v;
    }

    public Object get(String key) {
        if (key == null)
            throw new RuntimeException("key should never be null");
        else if (key.equals("_id"))
            return _id;
        else if (key.equals("filename"))
            return _filename;
        else if (key.equals("contentType"))
            return _contentType;
        else if (key.equals("length"))
            return _length;
        else if (key.equals("chunkSize"))
            return _chunkSize;
        else if (key.equals("uploadDate"))
            return _uploadDate;
        else if (key.equals("md5"))
            return _md5;
        return _metadata.get(key);
    }

    public void putAll(BSONObject o) {
        throw new UnsupportedOperationException();
    }

    public void putAll(Map m) {
        throw new UnsupportedOperationException();
    }

    public Map toMap() {
        throw new UnsupportedOperationException();
    }

    public Object removeField(String key) {
        throw new UnsupportedOperationException();
    }

    public boolean containsKey(String s) {
        return containsField(s);
    }

    public boolean containsField(String s) {
        return keySet().contains(s);
    }

    public Set<String> keySet() {
        Set<String> keys = new HashSet();
        keys.addAll(VALID_FIELDS);
        keys.addAll(_metadata.keySet());
        return keys;
    }

    public boolean isPartialObject() {
        return false;
    }

    public void markAsPartialObject() {
        throw new RuntimeException("can't load partial GridFSFile file");
    }

    // ----------------------
    // ------- fields -------
    // ----------------------

    public String toString() {
        return JSON.serialize(this);
    }

    protected void setGridFS(GridFS fs) {
        _fs = fs;
    }

    protected GridFS _fs = null;

    Object _id;
    String _filename;
    String _contentType;
    long _length;
    long _chunkSize;
    Date _uploadDate;
    List<String> _aliases;
    DBObject _metadata = new BasicDBObject();
    String _md5;

    final static Set<String> VALID_FIELDS = Collections.unmodifiableSet(new HashSet(Arrays.asList(new String[] {
            "_id", "filename", "contentType", "length", "chunkSize", "uploadDate", "aliases", "md5" })));
}