org.kurento.repository.internal.repoimpl.mongo.MongoRepositoryItem.java Source code

Introduction

Here is the source code for org.kurento.repository.internal.repoimpl.mongo.MongoRepositoryItem.java

Source

/*
 * (C) Copyright 2013 Kurento (http://kurento.org/)
 *
 * 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 org.kurento.repository.internal.repoimpl.mongo;

import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

import org.kurento.commons.exception.KurentoException;
import org.kurento.repository.RepositoryItemAttributes;
import org.kurento.repository.internal.repoimpl.AbstractRepositoryItem;

import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;
import com.mongodb.gridfs.GridFSDBFile;
import com.mongodb.gridfs.GridFSFile;
import com.mongodb.gridfs.GridFSInputFile;

public class MongoRepositoryItem extends AbstractRepositoryItem {

    private GridFSFile dbFile;
    private OutputStream storingOutputStream;

    private MongoRepositoryItem(MongoRepository repository, GridFSFile dbFile, State state) {

        super(dbFile.getId().toString(), state, loadAttributes(dbFile), repository);

        this.dbFile = dbFile;
        // don't call ours setMetadata(...)
        super.setMetadata(new HashMap<String, String>());
    }

    private static RepositoryItemAttributes loadAttributes(GridFSFile file) {

        RepositoryItemAttributes attributes = new RepositoryItemAttributes();

        attributes.setContentLength(file.getLength());
        attributes.setLastModified(file.getUploadDate().getTime());
        attributes.setMimeType(file.getContentType());

        return attributes;
    }

    public MongoRepositoryItem(MongoRepository repository, GridFSDBFile dbFile) {
        this(repository, dbFile, State.STORED);
    }

    public MongoRepositoryItem(MongoRepository repository, GridFSInputFile dbFile) {
        this(repository, dbFile, State.NEW);
    }

    @Override
    public InputStream createInputStreamToRead() {
        checkState(State.STORED);
        return ((GridFSDBFile) dbFile).getInputStream();
    }

    @Override
    public OutputStream createOutputStreamToWrite() {
        checkState(State.NEW);

        storingOutputStream = new FilterOutputStream(((GridFSInputFile) dbFile).getOutputStream()) {

            @Override
            public void close() throws IOException {
                putMetadataInGridFS(false);
                super.close();
                refreshAttributesOnClose();
            }
        };

        return storingOutputStream;
    }

    @Override
    public void setMetadata(Map<String, String> metadata) {
        super.setMetadata(metadata);
        if (state.equals(State.STORED)) {
            putMetadataInGridFS(true);
        }
    }

    protected void refreshAttributesOnClose() {
        dbFile = ((MongoRepository) repository).getGridFS().findOne(getId());
        if (dbFile == null) {
            throw new KurentoException("Grid object not found for id " + getId());
        }
        state = State.STORED;
        attributes.setContentLength(dbFile.getLength());
    }

    // TODO Optimize this to use the GridFS metadata
    private void putMetadataInGridFS(boolean save) {
        DBObject metadataDBO = new BasicDBObject();
        for (Entry<String, String> entry : metadata.entrySet()) {
            metadataDBO.put(entry.getKey(), entry.getValue());
        }
        dbFile.setMetaData(metadataDBO);
        if (save) {
            dbFile.save();
        }
    }

}