com.aliyun.odps.data.ResultSet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of odps-sdk-core Show documentation

The newest version!

/**
 *
 */
package com.aliyun.odps.data;

import java.util.Iterator;

import com.aliyun.odps.TableSchema;

/**
 * ResultSet for SQLTask
 *
 * @author emerson
 *
 */
public class ResultSet implements Iterable, Iterator {

  private Iterator recordIterator;
  private long recordCount;
  private TableSchema schema;

  /**
   * @param recordIterator
   * @param schema
   * @param recordCount -1 when we don't know iterator size.
   */
  public ResultSet(Iterator recordIterator, TableSchema schema, long recordCount) {
    this.recordIterator = recordIterator;
    this.recordCount = recordCount;
    this.schema = schema;
  }

  @Override
  public Iterator iterator() {
    return recordIterator;
  }

  public TableSchema getTableSchema() {
    return schema;
  }

  @Override
  public boolean hasNext() {
    return recordIterator.hasNext();
  }

  @Override
  public Record next() {
    return recordIterator.next();
  }

  @Override
  public void remove() {
    recordIterator.remove();
  }

  public long getRecordCount() {
    return recordCount;
  }
}