@BeforeMethod public void setUp() throws Exception { SessionBuilder sessionBuilder = testSessionBuilder(); generatedEvents = new EventsBuilder(); queryRunner = DistributedQueryRunner.builder(sessionBuilder.build()) .setExtraProperties(ImmutableMap.of("testing-warning-collector.preloaded-warnings", String.valueOf(TEST_WARNINGS))) .setNodeCount(1) .build(); queryRunner.installPlugin(new TestingEventListenerPlugin(generatedEvents)); generatedEvents.initialize(EXPECTED_EVENTS); }
private static DistributedQueryRunner createThriftQueryRunnerInternal(List<DriftServer> servers, int nodeCount, Map<String, String> properties) throws Exception { String addresses = servers.stream() .map(server -> "localhost:" + driftServerPort(server)) .collect(joining(",")); Session defaultSession = testSessionBuilder() .setCatalog("thrift") .setSchema("tiny") .build(); DistributedQueryRunner queryRunner = DistributedQueryRunner.builder(defaultSession) .setNodeCount(nodeCount) .setExtraProperties(properties) .build(); queryRunner.installPlugin(new ThriftPlugin()); Map<String, String> connectorProperties = ImmutableMap.<String, String>builder() .put("presto.thrift.client.addresses", addresses) .put("presto.thrift.client.connect-timeout", "30s") .put("presto-thrift.lookup-requests-concurrency", "2") .build(); queryRunner.createCatalog("thrift", "presto-thrift", connectorProperties); return queryRunner; }
throws Exception queryRunner = DistributedQueryRunner.builder(session).build(); queryRunner.installPlugin(new Plugin()
public static QueryRunner createKuduQueryRunner(String schema) throws Exception { QueryRunner runner = null; String kuduSchema = isSchemaEmulationEnabled() ? schema : "default"; try { runner = DistributedQueryRunner.builder(createSession(kuduSchema)).setNodeCount(3).build(); installKuduConnector(runner, kuduSchema); return runner; } catch (Throwable e) { closeAllSuppress(e, runner); throw e; } }
public static DistributedQueryRunner createQueryRunner(Session session) throws Exception { DistributedQueryRunner queryRunner = DistributedQueryRunner.builder(session) .setNodeCount(2) .build(); try { queryRunner.installPlugin(new TpchPlugin()); queryRunner.createCatalog("tpch", "tpch"); return queryRunner; } catch (Exception e) { queryRunner.close(); throw e; } } }
public static DistributedQueryRunner createQueryRunner() throws Exception { DistributedQueryRunner queryRunner = DistributedQueryRunner.builder(testSessionBuilder().build()) .setNodeCount(2) .build(); try { queryRunner.installPlugin(new TpchPlugin()); queryRunner.createCatalog("tpch", "tpch"); return queryRunner; } catch (Exception e) { queryRunner.close(); throw e; } } }
public static DistributedQueryRunner createQueryRunner(String dbConfigUrl, H2ResourceGroupsDao dao, String environment) throws Exception { DistributedQueryRunner queryRunner = DistributedQueryRunner .builder(testSessionBuilder().setCatalog("tpch").setSchema("tiny").build()) .setNodeCount(2) .setEnvironment(environment) .build(); try { Plugin h2ResourceGroupManagerPlugin = new H2ResourceGroupManagerPlugin(); queryRunner.installPlugin(h2ResourceGroupManagerPlugin); queryRunner.getCoordinator().getResourceGroupManager().get() .setConfigurationManager(CONFIGURATION_MANAGER_TYPE, ImmutableMap.of("resource-groups.config-db-url", dbConfigUrl, "node.environment", environment)); queryRunner.installPlugin(new TpchPlugin()); queryRunner.createCatalog("tpch", "tpch"); setup(queryRunner, dao, environment); return queryRunner; } catch (Exception e) { queryRunner.close(); throw e; } }
public static QueryRunner createKuduQueryRunnerTpch(Iterable<TpchTable<?>> tables) throws Exception { DistributedQueryRunner runner = null; String kuduSchema = isSchemaEmulationEnabled() ? "tpch" : "default"; try { runner = DistributedQueryRunner.builder(createSession(kuduSchema)).setNodeCount(3).build(); runner.installPlugin(new TpchPlugin()); runner.createCatalog("tpch", "tpch"); installKuduConnector(runner, kuduSchema); copyTpchTables(runner, "tpch", TINY_SCHEMA_NAME, createSession(kuduSchema), tables); return runner; } catch (Throwable e) { closeAllSuppress(e, runner); throw e; } }
public static DistributedQueryRunner createQueryRunner(Map<String, String> extraProperties, Map<String, String> coordinatorProperties) throws Exception { Session session = testSessionBuilder() .setSource("test") .setCatalog("tpcds") .setSchema("sf1") .build(); DistributedQueryRunner queryRunner = DistributedQueryRunner.builder(session) .setNodeCount(4) .setExtraProperties(extraProperties) .setCoordinatorProperties(coordinatorProperties) .build(); try { queryRunner.installPlugin(new TpcdsPlugin()); queryRunner.createCatalog("tpcds", "tpcds"); return queryRunner; } catch (Exception e) { queryRunner.close(); throw e; } }
public static DistributedQueryRunner createElasticsearchQueryRunner(EmbeddedElasticsearchNode embeddedElasticsearchNode, Iterable<TpchTable<?>> tables) throws Exception { DistributedQueryRunner queryRunner = null; try { queryRunner = DistributedQueryRunner.builder(createSession()) .setNodeCount(NODE_COUNT) .build(); queryRunner.installPlugin(new TpchPlugin()); queryRunner.createCatalog("tpch", "tpch"); embeddedElasticsearchNode.start(); ElasticsearchTableDescriptionProvider tableDescriptions = createTableDescriptions(queryRunner.getCoordinator().getMetadata()); TestingElasticsearchConnectorFactory testFactory = new TestingElasticsearchConnectorFactory(tableDescriptions); installElasticsearchPlugin(queryRunner, testFactory); TestingPrestoClient prestoClient = queryRunner.getClient(); LOG.info("Loading data..."); long startTime = System.nanoTime(); for (TpchTable<?> table : tables) { loadTpchTopic(embeddedElasticsearchNode, prestoClient, table); } LOG.info("Loading complete in %s", nanosSince(startTime).toString(SECONDS)); return queryRunner; } catch (Exception e) { closeAllSuppress(e, queryRunner, embeddedElasticsearchNode); throw e; } }
@BeforeMethod public void setUp() throws Exception { SessionBuilder sessionBuilder = testSessionBuilder(); generatedEvents = new EventsBuilder(); queryRunner = DistributedQueryRunner.builder(sessionBuilder.build()) .setExtraProperties(ImmutableMap.of("testing-warning-collector.preloaded-warnings", String.valueOf(TEST_WARNINGS))) .setNodeCount(1) .build(); queryRunner.installPlugin(new TestingEventListenerPlugin(generatedEvents)); generatedEvents.initialize(EXPECTED_EVENTS); }
throws Exception queryRunner = DistributedQueryRunner.builder(session).build(); queryRunner.installPlugin(new Plugin()
public static DistributedQueryRunner createQueryRunner(Session session) throws Exception { DistributedQueryRunner queryRunner = DistributedQueryRunner.builder(session) .setNodeCount(2) .build(); try { queryRunner.installPlugin(new TpchPlugin()); queryRunner.createCatalog("tpch", "tpch"); return queryRunner; } catch (Exception e) { queryRunner.close(); throw e; } } }
public static DistributedQueryRunner createQueryRunner() throws Exception { DistributedQueryRunner queryRunner = DistributedQueryRunner.builder(testSessionBuilder().build()) .setNodeCount(2) .build(); try { queryRunner.installPlugin(new TpchPlugin()); queryRunner.createCatalog("tpch", "tpch"); return queryRunner; } catch (Exception e) { queryRunner.close(); throw e; } } }
public static DistributedQueryRunner createQueryRunner(String dbConfigUrl, H2ResourceGroupsDao dao, String environment) throws Exception { DistributedQueryRunner queryRunner = DistributedQueryRunner .builder(testSessionBuilder().setCatalog("tpch").setSchema("tiny").build()) .setNodeCount(2) .setEnvironment(environment) .build(); try { Plugin h2ResourceGroupManagerPlugin = new H2ResourceGroupManagerPlugin(); queryRunner.installPlugin(h2ResourceGroupManagerPlugin); queryRunner.getCoordinator().getResourceGroupManager().get() .setConfigurationManager(CONFIGURATION_MANAGER_TYPE, ImmutableMap.of("resource-groups.config-db-url", dbConfigUrl, "node.environment", environment)); queryRunner.installPlugin(new TpchPlugin()); queryRunner.createCatalog("tpch", "tpch"); setup(queryRunner, dao, environment); return queryRunner; } catch (Exception e) { queryRunner.close(); throw e; } }
public static DistributedQueryRunner createQueryRunner(Map<String, String> extraProperties, Map<String, String> coordinatorProperties) throws Exception { Session session = testSessionBuilder() .setSource("test") .setCatalog("tpcds") .setSchema("sf1") .build(); DistributedQueryRunner queryRunner = DistributedQueryRunner.builder(session) .setNodeCount(4) .setExtraProperties(extraProperties) .setCoordinatorProperties(coordinatorProperties) .build(); try { queryRunner.installPlugin(new TpcdsPlugin()); queryRunner.createCatalog("tpcds", "tpcds"); return queryRunner; } catch (Exception e) { queryRunner.close(); throw e; } }