contrib/storage-phoenix/src/test/java/org/apache/drill/exec/store/phoenix/TestRowSet.java

package org.apache.drill.exec.store.phoenix;

import org.apache.drill.common.types.TypeProtos.MinorType;
import org.apache.drill.exec.physical.rowSet.DirectRowSet;
import org.apache.drill.exec.physical.rowSet.RowSet;
import org.apache.drill.exec.physical.rowSet.RowSet.SingleRowSet;
import org.apache.drill.exec.physical.rowSet.RowSetBuilder;
import org.apache.drill.exec.physical.rowSet.RowSetWriter;
import org.apache.drill.exec.record.metadata.SchemaBuilder;
import org.apache.drill.exec.record.metadata.TupleMetadata;
import org.apache.drill.test.SubOperatorTest;
import org.junit.Test;

public class TestRowSet extends SubOperatorTest {

  @Test
  public void testRowSet() {
    final TupleMetadata schema = new SchemaBuilder()
        .add("id", MinorType.INT)
        .add("name",MinorType.VARCHAR)
        .buildSchema();

    final RowSet rowSet = new RowSetBuilder(fixture.allocator(), schema)
        .addRow(1, "luocong")
        .addRow(2, "sunny")
        .build();

    rowSet.print();
    rowSet.clear();

    DirectRowSet directRowSet = DirectRowSet.fromSchema(fixture.allocator(), schema);
    RowSetWriter writer = directRowSet.writer();
    writer.scalar("id").setInt(1);
    writer.scalar("name").setString("luocong");
    writer.scalar("id").setInt(2);
    writer.scalar("name").setString("sunny");
    writer.save();
    SingleRowSet record = writer.done();
    record.print();
    record.clear();
  }
}