source: sample/hadoop-0.17/tw/org/nchc/util/LocalTupleRecordReader.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 LocalTupleRecordReader {
  private LongWritable mKey = new LongWritable();
  private SequenceFile.Reader mReader;

  private long cnt = 0;

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

    mReader = new SequenceFile.Reader(FileSystem.get(config),
        new Path(file), config);
  }

  public boolean read(Tuple tuple) throws IOException {
    if (mReader.next(mKey, tuple) == true) {
      cnt++;
    } else {
      return false;
    }

    return true;
  }

  public long getRecordCount() {
    return cnt;
  }

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

}