Examples with JdbcExtractionNamespace org.apache.druid.query.lookup.namespace.JdbcExtractionNamespace used on opensource projects

Example 6 with JdbcExtractionNamespace

use of org.apache.druid.query.lookup.namespace.JdbcExtractionNamespace in project druid by druid-io.

the class JdbcExtractionNamespaceTest method testMappingWithFilter.

@Test(timeout = 60_000L)
public void testMappingWithFilter() throws InterruptedException {
    final JdbcExtractionNamespace extractionNamespace = new JdbcExtractionNamespace(derbyConnectorRule.getMetadataConnectorConfig(), TABLE_NAME, KEY_NAME, VAL_NAME, tsColumn, FILTER_COLUMN + "='1'", new Period(0), null, new JdbcAccessSecurityConfig());
    try (CacheScheduler.Entry entry = scheduler.schedule(extractionNamespace)) {
        CacheSchedulerTest.waitFor(entry);
        final Map<String, String> map = entry.getCache();
        for (Map.Entry<String, String[]> e : RENAMES.entrySet()) {
            String key = e.getKey();
            String[] val = e.getValue();
            String field = val[0];
            String filterVal = val[1];
            if ("1".equals(filterVal)) {
                Assert.assertEquals("non-null check", NullHandling.emptyToNullIfNeeded(field), NullHandling.emptyToNullIfNeeded(map.get(key)));
            } else {
                Assert.assertEquals("non-null check", null, NullHandling.emptyToNullIfNeeded(map.get(key)));
            }
        }
    }
}

Example 7 with JdbcExtractionNamespace

use of org.apache.druid.query.lookup.namespace.JdbcExtractionNamespace in project druid by druid-io.

the class JdbcExtractionNamespaceTest method testMappingWithoutFilter.

@Test(timeout = 60_000L)
public void testMappingWithoutFilter() throws InterruptedException {
    final JdbcExtractionNamespace extractionNamespace = new JdbcExtractionNamespace(derbyConnectorRule.getMetadataConnectorConfig(), TABLE_NAME, KEY_NAME, VAL_NAME, tsColumn, null, new Period(0), null, new JdbcAccessSecurityConfig());
    try (CacheScheduler.Entry entry = scheduler.schedule(extractionNamespace)) {
        CacheSchedulerTest.waitFor(entry);
        final Map<String, String> map = entry.getCache();
        for (Map.Entry<String, String[]> e : RENAMES.entrySet()) {
            String key = e.getKey();
            String[] val = e.getValue();
            String field = val[0];
            Assert.assertEquals("non-null check", NullHandling.emptyToNullIfNeeded(field), NullHandling.emptyToNullIfNeeded(map.get(key)));
        }
        Assert.assertEquals("null check", null, map.get("baz"));
    }
}