org.apache.drill.exec.physical.unit

Class MiniPlanUnitTestBase.PopBuilder

java.lang.Object

org.apache.drill.exec.physical.unit.MiniPlanUnitTestBase.PopBuilder

Direct Known Subclasses:

MiniPlanUnitTestBase.ScanPopBuider

Enclosing class:

MiniPlanUnitTestBase

public class MiniPlanUnitTestBase.PopBuilder
extends Object

Similar to LegacyOperatorTestBuilder, build a physical operator (RecordBatch) and specify its input record batches. The input record batch could be a non-scan operator by calling addInputAsChild(), or a scan operator by calling addJsonScanAsChild() if it's SCAN operator. A miniplan rooted as join operator like following could be constructed in either the following way:

                 Join
                /    \
          JSON_T1    Filter
                       \
                     JSON_T2
 

 new PopBuilder()
  .physicalOperator(joinPopConfig)
  .addScanAsChild()
      .fileSystem(..)
      .columnsToRead(...)
      .inputPath(...)
      .buildAddAsInput()
  .addInputAsChild()
      .physicalOperator(filterPopConfig)
      .addScanAsChild()
          .fileSystem(...)
          .columnsToRead(...)
          .inputPath(...)
          .buildAddAsInput()
      .buildAddAsInput()
  .build();
 

   RecordBatch scan1 = new ScanPopBuilder()
                          .fileSystem(...)
                          .columnsToRead(..)
                          .inputPath(...)
                          .build();
   RecordBatch scan2 = ... ;

   RecordBatch filter = new PopBuilder()
                          .physicalOperator(filterPopConfig)
                          .addInput(scan2);
   RecordBatch join = new PopBuilder()
                          .physicalOperator(joinPopConfig)
                          .addInput(scan1)
                          .addInput(filter)
                          .build();

 

Field Summary

Fields

Modifier and Type	Field and Description
protected long	initReservation
protected long	maxAllocation
protected PhysicalOperator	popConfig

Constructor Summary

Constructors

Constructor and Description
PopBuilder()
PopBuilder(MiniPlanUnitTestBase.PopBuilder parent)

Method Summary

Modifier and Type	Method and Description
MiniPlanUnitTestBase.PopBuilder	addInput(RecordBatch batch)
MiniPlanUnitTestBase.PopBuilder	addInputAsChild() Return a nested PopBuilder to build a non-scan recordBatch, which will be added as input batch after call buildAddAsInput()
MiniPlanUnitTestBase.JsonScanBuilder	addJsonScanAsChild() Return a ScanPopBuilder to build a Scan recordBatch, which will be added as input batch after call buildAddAsInput()
MiniPlanUnitTestBase.ParquetScanBuilder	addParquetScanAsChild() Return a ScanPopBuilder to build a Scan recordBatch, which will be added as input batch after call buildAddAsInput()
RecordBatch	build()
MiniPlanUnitTestBase.PopBuilder	buildAddAsInput()
MiniPlanUnitTestBase.PopBuilder	initReservation(long initReservation) Set initial memory reservation used by this operator's allocator.
MiniPlanUnitTestBase.PopBuilder	maxAllocation(long maxAllocation) Set max memory reservation used by this operator's allocator.
MiniPlanUnitTestBase.PopBuilder	physicalOperator(PhysicalOperator popConfig)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

popConfig

protected PhysicalOperator popConfig

initReservation

protected long initReservation

maxAllocation

protected long maxAllocation

Constructor Detail

PopBuilder

public PopBuilder()

PopBuilder

public PopBuilder(MiniPlanUnitTestBase.PopBuilder parent)

Method Detail

physicalOperator

public MiniPlanUnitTestBase.PopBuilder physicalOperator(PhysicalOperator popConfig)

initReservation

public MiniPlanUnitTestBase.PopBuilder initReservation(long initReservation)

Set initial memory reservation used by this operator's allocator. Default is AbstractBase.INIT_ALLOCATION

Parameters:

initReservation -

Returns:

maxAllocation

public MiniPlanUnitTestBase.PopBuilder maxAllocation(long maxAllocation)

Set max memory reservation used by this operator's allocator. Default is AbstractBase.MAX_ALLOCATION

Parameters:

maxAllocation -

Returns:

addJsonScanAsChild

public MiniPlanUnitTestBase.JsonScanBuilder addJsonScanAsChild()

Return a ScanPopBuilder to build a Scan recordBatch, which will be added as input batch after call buildAddAsInput()

Returns:

ScanPopBuilder

addParquetScanAsChild

public MiniPlanUnitTestBase.ParquetScanBuilder addParquetScanAsChild()

Return a ScanPopBuilder to build a Scan recordBatch, which will be added as input batch after call buildAddAsInput()

Returns:

ScanPopBuilder

addInputAsChild

public MiniPlanUnitTestBase.PopBuilder addInputAsChild()

Return a nested PopBuilder to build a non-scan recordBatch, which will be added as input batch after call buildAddAsInput()

Returns:

a nested PopBuild for non-scan recordbatch.

addInput

public MiniPlanUnitTestBase.PopBuilder addInput(RecordBatch batch)

buildAddAsInput

public MiniPlanUnitTestBase.PopBuilder buildAddAsInput()
                                                throws Exception

Throws:

Exception

build

public RecordBatch build()
                  throws Exception

Throws:

Exception