com.tinkerpop.gremlin.hadoop.structure.io.kryo.KryoOutputFormat Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hadoop-gremlin Show documentation

The newest version!

package com.tinkerpop.gremlin.hadoop.structure.io.kryo;

import com.tinkerpop.gremlin.hadoop.structure.io.CommonFileOutputFormat;
import com.tinkerpop.gremlin.hadoop.structure.io.VertexWritable;
import org.apache.hadoop.io.NullWritable;
import org.apache.hadoop.mapreduce.RecordWriter;
import org.apache.hadoop.mapreduce.TaskAttemptContext;

import java.io.DataOutputStream;
import java.io.IOException;

/**
 * @author Marko A. Rodriguez (http://markorodriguez.com)
 */
public class KryoOutputFormat extends CommonFileOutputFormat {

    @Override
    public RecordWriter getRecordWriter(final TaskAttemptContext job) throws IOException, InterruptedException {
        return new KryoRecordWriter(getDataOuputStream(job));
    }

    public RecordWriter getRecordWriter(final TaskAttemptContext job, final DataOutputStream outputStream) throws IOException, InterruptedException {
        return new KryoRecordWriter(outputStream);
    }
}