me.lightspeed7.mongofs.MongoFileWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongoFS Show documentation

An extension to the MongoDB Java Driver library that goes beyond what the GridFS feature supports. Compressed file storage, zip files, temporary files

There is a newer version: 0.10.0

Show newest version

package me.lightspeed7.mongofs;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import me.lightspeed7.mongofs.url.MongoFileUrl;
import me.lightspeed7.mongofs.util.BytesCopier;

import org.mongodb.Document;
import org.mongodb.MongoCollection;

public class MongoFileWriter {

    private MongoFile file;
    private me.lightspeed7.mongofs.url.MongoFileUrl url;
    private MongoCollection chunksCollection;
    private MongoFileStore store;

    public MongoFileWriter(final MongoFileStore store, final MongoFileUrl url, final MongoFile file,
            final MongoCollection chunksCollection) {

        this.store = store;
        this.url = url;
        this.file = file;
        this.chunksCollection = chunksCollection;
    }

    /**
     * Stream the data to the file
     * 
     * @param in
     * @return the file object
     * 
     * @throws IOException
     * @throws IllegalArgumentException
     */
    public MongoFile write(final InputStream in) throws IOException {

        if (in == null) {
            throw new IllegalArgumentException("passed inputStream cannot be null");
        }

        // transfer the data
        OutputStream out = getOutputStream();
        try {
            new BytesCopier(in, out).transfer(true);
        } finally {
            if (out != null) {
                out.close();
            }
        }

        // make sure all the bytes transferred correctly
        file.validate();

        // return the file object
        return store.findOne(file.getId());
    }

    /**
     * Returns an output stream to write data to
     * 
     * @return an OutputStream
     * @throws IOException
     */
    public OutputStream getOutputStream() throws IOException {

        MongoFileWriterAdapter adapter = new MongoFileWriterAdapter(file);

        FileChunksOutputStreamSink chunks = new FileChunksOutputStreamSink(//
                chunksCollection, file.getId(), adapter, file.getExpiresAt());

        OutputStream sink = new BufferedChunksOutputStream(chunks, file.getChunkSize());

        if (url.isStoredEncrypted()) {
            sink = new MongoEncryptionOutputStream(store.getConfig(), file, sink);
        }

        if (url.isStoredCompressed()) {
            return new MongoGZipOutputStream(file, sink);
        }
        else {
            return new CountingOutputStream(MongoFileConstants.length, file, sink);
        }
    }

    /**
     * Read the given stream as a zip file and split the files inside into separate file objects
     * 
     * @param in
     * 
     * @return MongoManifest file
     * 
     * @throws IOException
     */
    public MongoManifest uploadZipFile(final InputStream in) throws IOException {

        return new MongoZipFileExpander(store, file).expandFrom(in);
    }

    /**
     * The the MongoFile object to write to
     * 
     * @return the MongoFile object
     */
    public MongoFile getMongoFile() {

        return file;
    }

    @Override
    public String toString() {

        return String.format("MongoFileWriter [chunksCollection=%s, url=%s]", chunksCollection, url);
    }

}