com.thinkaurelius.titan.hadoop.formats.TitanH1OutputCommitter Maven / Gradle / Ivy

Go to download
package com.thinkaurelius.titan.hadoop.formats;

import org.apache.hadoop.mapreduce.JobContext;
import org.apache.hadoop.mapreduce.OutputCommitter;
import org.apache.hadoop.mapreduce.TaskAttemptContext;

import java.io.IOException;

public class TitanH1OutputCommitter extends OutputCommitter {
    private final TitanH1OutputFormat tof;

    public TitanH1OutputCommitter(TitanH1OutputFormat tof) {
        this.tof = tof;
    }

    @Override
    public void setupJob(JobContext jobContext) throws IOException {

    }

    @Override
    public void setupTask(TaskAttemptContext taskContext) throws IOException {

    }

    @Override
    public boolean needsTaskCommit(TaskAttemptContext taskContext) throws IOException {
        return tof.hasModifications(taskContext.getTaskAttemptID());
    }

    @Override
    public void commitTask(TaskAttemptContext taskContext) throws IOException {
        tof.commit(taskContext.getTaskAttemptID());
    }

    @Override
    public void abortTask(TaskAttemptContext taskContext) throws IOException {
        tof.abort(taskContext.getTaskAttemptID());
    }
}