@Override protected List<Driver> createDrivers(TaskContext taskContext) { return localQueryRunner.createDrivers(session, query, new NullOutputFactory(), taskContext); } }
public List<Driver> createDrivers(@Language("SQL") String sql, OutputFactory outputFactory, TaskContext taskContext) { return createDrivers(defaultSession, sql, outputFactory, taskContext); }
public List<Driver> createDrivers(Session session, @Language("SQL") String sql, OutputFactory outputFactory, TaskContext taskContext) { Plan plan = createPlan(session, sql, WarningCollector.NOOP); return createDrivers(session, plan, outputFactory, taskContext); }
private void setUpCountStarFromOrdersWithJoin() { // query will reserve all memory in the user pool and discard the output setUp(() -> { OutputFactory outputFactory = new PageConsumerOutputFactory(types -> (page -> {})); return localQueryRunner.createDrivers("SELECT COUNT(*) FROM orders JOIN lineitem ON CAST(orders.orderkey AS VARCHAR) = CAST(lineitem.orderkey AS VARCHAR)", outputFactory, taskContext); }); }
List<Driver> drivers = createDrivers(session, plan, outputFactory, taskContext); drivers.forEach(closer::register);
List<Driver> drivers = localQueryRunner.createDrivers( query, new PageConsumerOperator.PageConsumerOutputFactory(types -> output::add),
@Override protected List<Driver> createDrivers(TaskContext taskContext) { return localQueryRunner.createDrivers(session, query, new NullOutputFactory(), taskContext); } }
@Override protected List<Driver> createDrivers(TaskContext taskContext) { return localQueryRunner.createDrivers(query, new NullOutputFactory(), taskContext); } }
public List<Driver> createDrivers(@Language("SQL") String sql, OutputFactory outputFactory, TaskContext taskContext) { return createDrivers(defaultSession, sql, outputFactory, taskContext); }
private MaterializedResult executeInternal(Session session, @Language("SQL") String sql) { lock.readLock().lock(); try { MaterializedOutputFactory outputFactory = new MaterializedOutputFactory(); TaskContext taskContext = createTaskContext(executor, session); List<Driver> drivers = createDrivers(session, sql, outputFactory, taskContext); boolean done = false; while (!done) { boolean processed = false; for (Driver driver : drivers) { if (!driver.isFinished()) { driver.process(); processed = true; } } done = !processed; } return outputFactory.getMaterializedResult(); } finally { lock.readLock().unlock(); } }
LocalQueryRunner.MaterializedOutputFactory outputFactory = new LocalQueryRunner.MaterializedOutputFactory(); TaskContext taskContext = createTaskContext(queryContext, localQueryRunner.getExecutor(), session, new DataSize(0, BYTE)); Driver driver = Iterables.getOnlyElement(localQueryRunner.createDrivers("SELECT COUNT(*), clerk FROM orders GROUP BY clerk", outputFactory, taskContext));
List<Driver> drivers = localQueryRunner.createDrivers( query, new PageConsumerOperator.PageConsumerOutputFactory(types -> output::add),