public DataSourceMetadataQueryBuilder dataSource(String ds) { dataSource = new TableDataSource(ds); return this; }
public ScanQueryBuilder dataSource(String ds) { dataSource = new TableDataSource(ds); return this; }
public TopNQueryBuilder dataSource(String d) { dataSource = new TableDataSource(d); return this; }
public SelectQueryBuilder dataSource(String ds) { dataSource = new TableDataSource(ds); return this; }
public SearchQueryBuilder dataSource(String d) { dataSource = new TableDataSource(d); return this; }
public Builder setDataSource(String dataSource) { this.dataSource = new TableDataSource(dataSource); return this; }
public SegmentMetadataQueryBuilder dataSource(String ds) { dataSource = new TableDataSource(ds); return this; }
public TimeseriesQueryBuilder dataSource(String ds) { dataSource = new TableDataSource(ds); return this; }
public TimeBoundaryQueryBuilder dataSource(String ds) { dataSource = new TableDataSource(ds); return this; }
@Test public void testSerialization() throws IOException { DataSource dataSource = new TableDataSource("somedatasource"); String json = jsonMapper.writeValueAsString(dataSource); DataSource serdeDataSource = jsonMapper.readValue(json, DataSource.class); Assert.assertEquals(dataSource, serdeDataSource); }
@Test public void testLegacyDataSource() throws IOException { DataSource dataSource = jsonMapper.readValue("\"somedatasource\"", DataSource.class); Assert.assertEquals(new TableDataSource("somedatasource"), dataSource); }
@Test public void testMergeResultsEmptyResults() { List<Result<TimeBoundaryResultValue>> results = new ArrayList<>(); TimeBoundaryQuery query = new TimeBoundaryQuery(new TableDataSource("test"), null, null, null, null); Iterable<Result<TimeBoundaryResultValue>> actual = query.mergeResults(results); Assert.assertFalse(actual.iterator().hasNext()); } }
@Test public void testTableDataSource() throws IOException { DataSource dataSource = jsonMapper.readValue("{\"type\":\"table\", \"name\":\"somedatasource\"}", DataSource.class); Assert.assertEquals(new TableDataSource("somedatasource"), dataSource); }
@Override public Sequence<Result<TimeseriesResultValue>> run( QueryPlus<Result<TimeseriesResultValue>> queryPlus, Map<String, Object> responseContext ) { if (queryPlus.getQuery().getDataSource().equals(new TableDataSource("ds1"))) { return Sequences.simple(descending ? Lists.reverse(ds1) : ds1); } else { return Sequences.simple(descending ? Lists.reverse(ds2) : ds2); } } }
@Test public void testQueryMaxTimeout() { exception.expect(IAE.class); exception.expectMessage("configured [timeout = 1000] is more than enforced limit of maxQueryTimeout [100]."); Query<?> query = new TestQuery( new TableDataSource("test"), new MultipleIntervalSegmentSpec(ImmutableList.of(Intervals.of("0/100"))), false, ImmutableMap.of(QueryContexts.TIMEOUT_KEY, 1000) ); QueryContexts.verifyMaxQueryTimeout(query, 100); }
private ScanQuery.ScanQueryBuilder newTestQuery() { return ScanQuery.newScanQueryBuilder() .dataSource(new TableDataSource(QueryRunnerTestHelper.dataSource)) .columns(Collections.emptyList()) .intervals(QueryRunnerTestHelper.fullOnIntervalSpec) .limit(Integer.MAX_VALUE) .legacy(false); }
private ScanQuery.ScanQueryBuilder newTestQuery() { return ScanQuery.newScanQueryBuilder() .dataSource(new TableDataSource(QueryRunnerTestHelper.dataSource)) .columns(Collections.emptyList()) .intervals(QueryRunnerTestHelper.fullOnIntervalSpec) .limit(3) .legacy(legacy); }
private ScanQuery.ScanQueryBuilder newBuilder() { return ScanQuery.newScanQueryBuilder() .dataSource(new TableDataSource(QueryRunnerTestHelper.dataSource)) .intervals(SelectQueryRunnerTest.I_0112_0114_SPEC) .batchSize(batchSize) .columns(Collections.emptyList()) .legacy(false) .limit(limit); }
private Druids.SelectQueryBuilder newTestQuery() { return Druids.newSelectQueryBuilder() .dataSource(new TableDataSource(QueryRunnerTestHelper.dataSource)) .dimensionSpecs(DefaultDimensionSpec.toSpec(Collections.emptyList())) .metrics(Collections.emptyList()) .intervals(QueryRunnerTestHelper.fullOnIntervalSpec) .granularity(QueryRunnerTestHelper.allGran) .pagingSpec(PagingSpec.newSpec(3)) .descending(descending); }
private Druids.SelectQueryBuilder newBuilder() { return Druids.newSelectQueryBuilder() .dataSource(new TableDataSource(QueryRunnerTestHelper.dataSource)) .intervals(SelectQueryRunnerTest.I_0112_0114_SPEC) .granularity(QueryRunnerTestHelper.allGran) .dimensionSpecs(DefaultDimensionSpec.toSpec(QueryRunnerTestHelper.dimensions)) .pagingSpec(PagingSpec.newSpec(3)); }