private static void installElasticsearchPlugin(QueryRunner queryRunner, TestingElasticsearchConnectorFactory factory) throws Exception { ElasticsearchPlugin plugin = new ElasticsearchPlugin(); plugin.setConnectorFactory(factory); queryRunner.installPlugin(plugin); URL metadataUrl = getResource(ElasticsearchQueryRunner.class, "/queryrunner"); assertNotNull(metadataUrl, "metadataUrl is null"); URI metadataUri = metadataUrl.toURI(); Map<String, String> config = new HashMap<>(); config.put("elasticsearch.default-schema-name", TPCH_SCHEMA); config.put("elasticsearch.table-description-directory", metadataUri.toString()); config.put("elasticsearch.scroll-size", "1000"); config.put("elasticsearch.scroll-timeout", "1m"); config.put("elasticsearch.max-hits", "1000000"); config.put("elasticsearch.request-timeout", "2m"); config.put("elasticsearch.max-request-retries", "3"); config.put("elasticsearch.max-request-retry-time", "5s"); queryRunner.createCatalog("elasticsearch", "elasticsearch", config); }
public static void installKafkaPlugin(EmbeddedKafka embeddedKafka, QueryRunner queryRunner, Map<SchemaTableName, KafkaTopicDescription> topicDescriptions) { KafkaPlugin kafkaPlugin = new KafkaPlugin(); kafkaPlugin.setTableDescriptionSupplier(() -> topicDescriptions); queryRunner.installPlugin(kafkaPlugin); Map<String, String> kafkaConfig = ImmutableMap.of( "kafka.nodes", embeddedKafka.getConnectString(), "kafka.table-names", Joiner.on(",").join(topicDescriptions.keySet()), "kafka.connect-timeout", "120s", "kafka.default-schema", "default"); queryRunner.createCatalog("kafka", "kafka", kafkaConfig); }
public static void installRedisPlugin(EmbeddedRedis embeddedRedis, QueryRunner queryRunner, Map<SchemaTableName, RedisTableDescription> tableDescriptions) { RedisPlugin redisPlugin = new RedisPlugin(); redisPlugin.setTableDescriptionSupplier(() -> tableDescriptions); queryRunner.installPlugin(redisPlugin); Map<String, String> redisConfig = ImmutableMap.of( "redis.nodes", embeddedRedis.getConnectString() + ":" + embeddedRedis.getPort(), "redis.table-names", Joiner.on(",").join(tableDescriptions.keySet()), "redis.default-schema", "default", "redis.hide-internal-columns", "true", "redis.key-prefix-schema-table", "true"); queryRunner.createCatalog("redis", "redis", redisConfig); }
@BeforeClass public void setUp() { metadata = new TestMetadata(); getQueryRunner().installPlugin(new TestPlugin(metadata)); getQueryRunner().installPlugin(new TpchPlugin()); getQueryRunner().createCatalog("test", "test", ImmutableMap.of()); getQueryRunner().createCatalog("tpch", "tpch", ImmutableMap.of()); }
@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 void installKuduConnector(QueryRunner runner, String schema) { String masterAddresses = System.getProperty("kudu.client.master-addresses", "localhost:7051"); Map<String, String> properties; if (!isSchemaEmulationEnabled()) { properties = ImmutableMap.of( "kudu.schema-emulation.enabled", "false", "kudu.client.master-addresses", masterAddresses); } else { properties = ImmutableMap.of( "kudu.schema-emulation.enabled", "true", "kudu.schema-emulation.prefix", getSchemaEmulationPrefix(), "kudu.client.master-addresses", masterAddresses); } runner.installPlugin(new KuduPlugin()); runner.createCatalog("kudu", "kudu", properties); if (isSchemaEmulationEnabled()) { runner.execute("DROP SCHEMA IF EXISTS " + schema); runner.execute("CREATE SCHEMA " + schema); } }
queryRunner.installPlugin(new Plugin()
@Test public void testCharTrailingSpace() throws Exception { execute("CREATE TABLE tpch.char_trailing_space (x char(10))"); assertUpdate("INSERT INTO char_trailing_space VALUES ('test')", 1); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test'", "VALUES 'test'"); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test '", "VALUES 'test'"); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test '", "VALUES 'test'"); assertEquals(getQueryRunner().execute("SELECT * FROM char_trailing_space WHERE x = char ' test'").getRowCount(), 0); Map<String, String> properties = ImmutableMap.of("deprecated.legacy-char-to-varchar-coercion", "true"); Map<String, String> connectorProperties = ImmutableMap.of("connection-url", postgreSqlServer.getJdbcUrl()); try (QueryRunner queryRunner = new DistributedQueryRunner(getSession(), 3, properties);) { queryRunner.installPlugin(new PostgreSqlPlugin()); queryRunner.createCatalog("postgresql", "postgresql", connectorProperties); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test'").getRowCount(), 0); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '").getRowCount(), 0); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '").getRowCount(), 0); MaterializedResult result = queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '"); assertEquals(result.getRowCount(), 1); assertEquals(result.getMaterializedRows().get(0).getField(0), "test "); } assertUpdate("DROP TABLE char_trailing_space"); }
@Test public void testCharTrailingSpace() throws Exception { execute("CREATE TABLE tpch.char_trailing_space (x char(10))"); assertUpdate("INSERT INTO char_trailing_space VALUES ('test')", 1); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test'", "VALUES 'test'"); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test '", "VALUES 'test'"); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test '", "VALUES 'test'"); assertEquals(getQueryRunner().execute("SELECT * FROM char_trailing_space WHERE x = char ' test'").getRowCount(), 0); Map<String, String> properties = ImmutableMap.of("deprecated.legacy-char-to-varchar-coercion", "true"); Map<String, String> connectorProperties = ImmutableMap.of("connection-url", mysqlServer.getJdbcUrl()); try (QueryRunner queryRunner = new DistributedQueryRunner(getSession(), 3, properties);) { queryRunner.installPlugin(new MySqlPlugin()); queryRunner.createCatalog("mysql", "mysql", connectorProperties); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test'").getRowCount(), 0); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '").getRowCount(), 0); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '").getRowCount(), 0); MaterializedResult result = queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '"); assertEquals(result.getRowCount(), 1); assertEquals(result.getMaterializedRows().get(0).getField(0), "test "); } assertUpdate("DROP TABLE char_trailing_space"); }
@BeforeClass public void setUp() { metadata = new TestMetadata(); getQueryRunner().installPlugin(new TestPlugin(metadata)); getQueryRunner().installPlugin(new TpchPlugin()); getQueryRunner().createCatalog("test", "test", ImmutableMap.of()); getQueryRunner().createCatalog("tpch", "tpch", ImmutableMap.of()); }
@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); }
queryRunner.installPlugin(new Plugin()
@Test public void testCharTrailingSpace() throws Exception { execute("CREATE TABLE tpch.char_trailing_space (x char(10))"); assertUpdate("INSERT INTO char_trailing_space VALUES ('test')", 1); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test'", "VALUES 'test'"); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test '", "VALUES 'test'"); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test '", "VALUES 'test'"); assertEquals(getQueryRunner().execute("SELECT * FROM char_trailing_space WHERE x = char ' test'").getRowCount(), 0); Map<String, String> properties = ImmutableMap.of("deprecated.legacy-char-to-varchar-coercion", "true"); Map<String, String> connectorProperties = ImmutableMap.of("connection-url", postgreSqlServer.getJdbcUrl()); try (QueryRunner queryRunner = new DistributedQueryRunner(getSession(), 3, properties);) { queryRunner.installPlugin(new PostgreSqlPlugin()); queryRunner.createCatalog("postgresql", "postgresql", connectorProperties); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test'").getRowCount(), 0); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '").getRowCount(), 0); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '").getRowCount(), 0); MaterializedResult result = queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '"); assertEquals(result.getRowCount(), 1); assertEquals(result.getMaterializedRows().get(0).getField(0), "test "); } assertUpdate("DROP TABLE char_trailing_space"); }
@Test public void testCharTrailingSpace() throws Exception { execute("CREATE TABLE tpch.char_trailing_space (x char(10))"); assertUpdate("INSERT INTO char_trailing_space VALUES ('test')", 1); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test'", "VALUES 'test'"); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test '", "VALUES 'test'"); assertQuery("SELECT * FROM char_trailing_space WHERE x = char 'test '", "VALUES 'test'"); assertEquals(getQueryRunner().execute("SELECT * FROM char_trailing_space WHERE x = char ' test'").getRowCount(), 0); Map<String, String> properties = ImmutableMap.of("deprecated.legacy-char-to-varchar-coercion", "true"); Map<String, String> connectorProperties = ImmutableMap.of("connection-url", mysqlServer.getJdbcUrl()); try (QueryRunner queryRunner = new DistributedQueryRunner(getSession(), 3, properties);) { queryRunner.installPlugin(new MySqlPlugin()); queryRunner.createCatalog("mysql", "mysql", connectorProperties); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test'").getRowCount(), 0); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '").getRowCount(), 0); assertEquals(queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '").getRowCount(), 0); MaterializedResult result = queryRunner.execute("SELECT * FROM char_trailing_space WHERE x = char 'test '"); assertEquals(result.getRowCount(), 1); assertEquals(result.getMaterializedRows().get(0).getField(0), "test "); } assertUpdate("DROP TABLE char_trailing_space"); }