source: sample/hadoop-0.17/tw/org/nchc/util/LocalTupleRecordWriter.java @ 20

package tw.org.nchc.util;

import java.io.IOException;

import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.fs.Path;
import org.apache.hadoop.io.LongWritable;
import org.apache.hadoop.io.SequenceFile;
import org.apache.hadoop.mapred.JobConf;

import tw.org.nchc.tuple.Tuple;

public class LocalTupleRecordWriter {

  private LongWritable mLong = new LongWritable();

  private long mCnt = 0;

  private SequenceFile.Writer writer;

  public LocalTupleRecordWriter(String file) throws IOException {
    JobConf config = new JobConf();

    writer = SequenceFile.createWriter(FileSystem.get(config), config,
        new Path(file), LongWritable.class, Tuple.class);
  }

  public void add(Tuple tuple) throws IOException {
    mLong.set(mCnt);
    writer.append(mLong, tuple);
    mCnt++;
  }

  public long getRecordCount() {
    return mCnt;
  }

  public void close() throws IOException {
    writer.close();
  }
}