com.tinkerpop.gremlin.hadoop.structure.io.script.ScriptOutputFormat 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.script;

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 Daniel Kuppitz (daniel at thinkaurelius.com)
 */
public class ScriptOutputFormat extends CommonFileOutputFormat {

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

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