public OrderedPartitionRecordBatch(
OrderedPartitionSender pop, RecordBatch incoming, FragmentContext context)
throws OutOfMemoryException {
super(pop, context);
this.incoming = incoming;
this.partitions = pop.getDestinations().size();
this.sendingMajorFragmentWidth = pop.getSendingWidth();
this.recordsToSample = pop.getRecordsToSample();
this.samplingFactor = pop.getSamplingFactor();
this.completionFactor = pop.getCompletionFactor();
DistributedCache cache = context.getDrillbitContext().getCache();
this.mmap = cache.getMultiMap(MULTI_CACHE_CONFIG);
this.tableMap = cache.getMap(SINGLE_CACHE_CONFIG);
Preconditions.checkNotNull(tableMap);
this.mapKey =
String.format(
"%s_%d", context.getHandle().getQueryId(), context.getHandle().getMajorFragmentId());
this.minorFragmentSampleCount = cache.getCounter(mapKey);
SchemaPath outputPath = popConfig.getRef();
MaterializedField outputField =
MaterializedField.create(outputPath, Types.required(TypeProtos.MinorType.INT));
this.partitionKeyVector =
(IntVector) TypeHelper.getNewVector(outputField, oContext.getAllocator());
}
public void run() throws Exception {
coord.start(10000);
DrillbitEndpoint md = engine.start();
cache.run();
manager.start(md, cache, engine.getBitCom(), coord);
handle = coord.register(md);
}
protected void testSqlPlan(String sqlCommands) throws Exception {
String[] sqlStrings = sqlCommands.split(";");
final DistributedCache cache = new LocalCache();
cache.run();
final SystemOptionManager opt = new SystemOptionManager(cache);
opt.init();
final OptionManager sess = new SessionOptionManager(opt);
new NonStrictExpectations() {
{
dbContext.getMetrics();
result = new MetricRegistry();
dbContext.getAllocator();
result = new TopLevelAllocator();
dbContext.getConfig();
result = config;
dbContext.getOptionManager();
result = opt;
dbContext.getCache();
result = cache;
}
};
final StoragePluginRegistry registry = new StoragePluginRegistry(dbContext);
registry.init();
final FunctionImplementationRegistry functionRegistry =
new FunctionImplementationRegistry(config);
final SchemaPlus root = Frameworks.createRootSchema(false);
registry.getSchemaFactory().registerSchemas(new UserSession(null, null, null), root);
new NonStrictExpectations() {
{
context.getNewDefaultSchema();
result = root;
context.getStorage();
result = registry;
context.getFunctionRegistry();
result = functionRegistry;
context.getSession();
result = new UserSession(null, null, null);
context.getCurrentEndpoint();
result = DrillbitEndpoint.getDefaultInstance();
context.getActiveEndpoints();
result = ImmutableList.of(DrillbitEndpoint.getDefaultInstance());
context.getPlannerSettings();
result = new PlannerSettings(sess);
context.getOptions();
result = sess;
context.getConfig();
result = config;
context.getCache();
result = cache;
}
};
for (String sql : sqlStrings) {
if (sql.trim().isEmpty()) continue;
DrillSqlWorker worker = new DrillSqlWorker(context);
PhysicalPlan p = worker.getPlan(sql);
}
}