edu.jhu.hlt.concrete.serialization.TarCompactCommunicationSerializer.java Source code

Introduction

Here is the source code for edu.jhu.hlt.concrete.serialization.TarCompactCommunicationSerializer.java

Source

/*
 * Copyright 2012-2016 Johns Hopkins University HLTCOE. All rights reserved.
 * See LICENSE in the project root directory.
 */
package edu.jhu.hlt.concrete.serialization;

import java.io.BufferedOutputStream;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Collection;
import java.util.Iterator;

import org.apache.commons.compress.archivers.tar.TarArchiveEntry;
import org.apache.commons.compress.archivers.tar.TarArchiveOutputStream;
import org.apache.commons.io.IOUtils;

import edu.jhu.hlt.concrete.Communication;
import edu.jhu.hlt.concrete.serialization.iterators.TarArchiveEntryCommunicationIterator;
import edu.jhu.hlt.concrete.util.ConcreteException;

/**
 *
 */
public class TarCompactCommunicationSerializer extends CompactCommunicationSerializer
        implements CommunicationTarSerializer {

    /* (non-Javadoc)
     * @see edu.jhu.hlt.concrete.serialization.CommunicationTarSerializer#toTar(java.util.Collection, java.nio.file.Path)
     */
    @Override
    public void toTar(Collection<Communication> commColl, Path outPath) throws ConcreteException, IOException {
        try (OutputStream os = Files.newOutputStream(outPath);
                BufferedOutputStream bos = new BufferedOutputStream(os);
                TarArchiveOutputStream tos = new TarArchiveOutputStream(bos);) {
            for (Communication c : commColl) {
                TarArchiveEntry entry = new TarArchiveEntry(c.getId() + ".concrete");
                byte[] cbytes = this.toBytes(c);
                entry.setSize(cbytes.length);
                tos.putArchiveEntry(entry);
                try (ByteArrayInputStream bis = new ByteArrayInputStream(cbytes)) {
                    IOUtils.copy(bis, tos);
                    tos.closeArchiveEntry();
                }
            }

        } catch (IOException e) {
            throw new ConcreteException(e);
        }
    }

    /* (non-Javadoc)
     * @see edu.jhu.hlt.concrete.serialization.CommunicationTarSerializer#toTar(java.util.Collection, java.lang.String)
     */
    @Override
    public void toTar(Collection<Communication> commColl, String outPathString)
            throws ConcreteException, IOException {
        this.toTar(commColl, Paths.get(outPathString));
    }

    /* (non-Javadoc)
     * @see edu.jhu.hlt.concrete.serialization.CommunicationTarSerializer#fromTar(java.io.InputStream)
     */
    @Override
    public Iterator<Communication> fromTar(InputStream is) throws ConcreteException, IOException {
        return new TarArchiveEntryCommunicationIterator(is);
    }
}