Package io.druid.query

Examples of io.druid.query.QueryRunner

745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
        .setGranularity(new PeriodGranularity(new Period("P1M"), null, null))
        .setLimitSpec(orderBySpec);

    final GroupByQuery fullQuery = builder.build();

    QueryRunner mergedRunner = factory.getToolchest().mergeResults(
        new QueryRunner<Row>()
        {
          @Override
          public Sequence<Row> run(
              Query<Row> query, Map<String, Object> context
          )
          {
            // simulate two daily segments
            final Query query1 = query.withQuerySegmentSpec(
                new MultipleIntervalSegmentSpec(Lists.newArrayList(new Interval("2011-04-02/2011-04-03")))
            );
            final Query query2 = query.withQuerySegmentSpec(
                new MultipleIntervalSegmentSpec(Lists.newArrayList(new Interval("2011-04-03/2011-04-04")))
            );
            return Sequences.concat(runner.run(query1, context), runner.run(query2, context));
          }
        }
    );

    Map<String, Object> context = Maps.newHashMap();
    TestHelper.assertExpectedObjects(expectedResults, mergedRunner.run(fullQuery, context), "merged");
  }
View Full Code Here
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
            )
        );

    final GroupByQuery fullQuery = builder.build();

    QueryRunner mergedRunner = factory.getToolchest().mergeResults(
        new QueryRunner<Row>()
        {
          @Override
          public Sequence<Row> run(
              Query<Row> query, Map<String, Object> context
          )
          {
            // simulate two daily segments
            final Query query1 = query.withQuerySegmentSpec(
                new MultipleIntervalSegmentSpec(Lists.newArrayList(new Interval("2011-04-02/2011-04-03")))
            );
            final Query query2 = query.withQuerySegmentSpec(
                new MultipleIntervalSegmentSpec(Lists.newArrayList(new Interval("2011-04-03/2011-04-04")))
            );
            return Sequences.concat(runner.run(query1, context), runner.run(query2, context));
          }
        }
    );

    Map<String, Object> context = Maps.newHashMap();
    TestHelper.assertExpectedObjects(expectedResults, mergedRunner.run(fullQuery, context), "merged");
  }
View Full Code Here
95
96
97
98
99
100
101
102
103
104
105
   *
   * @return
   */
  private List<SegmentAnalysis> getSegmentAnalysises(Segment index)
  {
    final QueryRunner runner = QueryRunnerTestHelper.makeQueryRunner(
        (QueryRunnerFactory) new SegmentMetadataQueryRunnerFactory(
            new SegmentMetadataQueryQueryToolChest(),
            QueryRunnerTestHelper.NOOP_QUERYWATCHER
        ), index
    );
View Full Code Here
89
90
91
92
93
94
95
96
97
98
99
100
        Object[] queryRunnerArray = (Object[]) object;

        Preconditions.checkState(queryRunnerArray.length == 1);
        Preconditions.checkState(queryRunnerArray[0] instanceof QueryRunner);

        final QueryRunner groupByRunner = (QueryRunner) queryRunnerArray[0];
        QueryRunner timeseriesRunner = new QueryRunner()
        {
          @Override
          public Sequence run(Query query, Map metadata)
          {
            TimeseriesQuery tsQuery = (TimeseriesQuery) query;
View Full Code Here
40
41
42
43
44
45
46
47
48
49
50
51
52
{
  public static Iterable<Row> runQuery(QueryRunnerFactory factory, QueryRunner runner, GroupByQuery query)
  {

    QueryToolChest toolChest = factory.getToolchest();
    QueryRunner theRunner = new FinalizeResultsQueryRunner<>(
        toolChest.mergeResults(toolChest.preMergeQueryDecoration(runner)),
        toolChest
    );

    Sequence<Row> queryResult = theRunner.run(query, Maps.newHashMap());
    return Sequences.toList(queryResult, Lists.<Row>newArrayList());
  }
View Full Code Here
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
                                          )
                                      )
                                  )
                                  .build();
    QueryToolChest toolChest = new TimeseriesQueryQueryToolChest(new QueryConfig());
    QueryRunner mergingrunner = toolChest.mergeResults(
        new UnionQueryRunner<Result<TimeseriesResultValue>>(
            new QueryRunner<Result<TimeseriesResultValue>>()
            {
              @Override
              public Sequence<Result<TimeseriesResultValue>> run(Query<Result<TimeseriesResultValue>> query,
                                                                 Map<String, Object> context)
              {
                if (query.getDataSource().equals(new TableDataSource("ds1"))) {
                  return Sequences.simple(
                      Lists.newArrayList(
                          new Result<TimeseriesResultValue>(
                              new DateTime("2011-04-02"),
                              new TimeseriesResultValue(
                                  ImmutableMap.<String, Object>of(
                                      "rows",
                                      1L,
                                      "idx",
                                      2L
                                  )
                              )
                          ),
                          new Result<TimeseriesResultValue>(
                              new DateTime("2011-04-03"),
                              new TimeseriesResultValue(
                                  ImmutableMap.<String, Object>of(
                                      "rows",
                                      3L,
                                      "idx",
                                      4L
                                  )
                              )
                          )
                      )
                  );
                } else {
                  return Sequences.simple(
                      Lists.newArrayList(
                          new Result<TimeseriesResultValue>(
                              new DateTime("2011-04-01"),
                              new TimeseriesResultValue(
                                  ImmutableMap.<String, Object>of(
                                      "rows",
                                      5L,
                                      "idx",
                                      6L
                                  )
                              )
                          ),
                          new Result<TimeseriesResultValue>(
                              new DateTime("2011-04-02"),
                              new TimeseriesResultValue(
                                  ImmutableMap.<String, Object>of(
                                      "rows",
                                      7L,
                                      "idx",
                                      8L
                                  )
                              )
                          ),
                          new Result<TimeseriesResultValue>(
                              new DateTime("2011-04-04"),
                              new TimeseriesResultValue(
                                  ImmutableMap.<String, Object>of(
                                      "rows",
                                      9L,
                                      "idx",
                                      10L
                                  )
                              )
                          )
                      )
                  );
                }
              }
            },
            toolChest
        )
    );

    List<Result<TimeseriesResultValue>> expectedResults = Arrays.asList(
        new Result<TimeseriesResultValue>(
            new DateTime("2011-04-01"),
            new TimeseriesResultValue(
                ImmutableMap.<String, Object>of("rows", 5L, "idx", 6L)
            )
        ),
        new Result<TimeseriesResultValue>(
            new DateTime("2011-04-02"),
            new TimeseriesResultValue(
                ImmutableMap.<String, Object>of("rows", 8L, "idx", 10L)
            )
        ),
        new Result<TimeseriesResultValue>(
            new DateTime("2011-04-03"),
            new TimeseriesResultValue(
                ImmutableMap.<String, Object>of("rows", 3L, "idx", 4L)
            )
        ),
        new Result<TimeseriesResultValue>(
            new DateTime("2011-04-04"),
            new TimeseriesResultValue(
                ImmutableMap.<String, Object>of("rows", 9L, "idx", 10L)
            )
        )
    );

    Iterable<Result<TimeseriesResultValue>> results = Sequences.toList(
        mergingrunner.run(query, Maps.<String, Object>newHashMap()),
        Lists.<Result<TimeseriesResultValue>>newArrayList()
    );

    System.out.println(results);
    TestHelper.assertExpectedResults(expectedResults, results);
View Full Code Here
64
65
66
67
68
69
70
71
72
73
74
        "version",
        0
    );

    final SpecificSegmentQueryRunner queryRunner = new SpecificSegmentQueryRunner(
        new QueryRunner()
        {
          @Override
          public Sequence run(Query query, Map context)
          {
            return new Sequence()
View Full Code Here
144
145
146
147
148
149
150
151
152
153
154
    rows.aggregate();
    builder.addMetric(rows);
    final Result<TimeseriesResultValue> value = builder.build();

    final SpecificSegmentQueryRunner queryRunner = new SpecificSegmentQueryRunner(
        new QueryRunner()
        {
          @Override
          public Sequence run(Query query, Map context)
          {
            return Sequences.withEffect(
View Full Code Here
TOP

Related Classes of io.druid.query.QueryRunner

  • io.druid.client.CachingClusteredClient
  • io.druid.client.CachingClusteredClientTest
  • io.druid.client.CachingQueryRunnerTest
  • io.druid.query.groupby.GroupByQueryRunnerTest
  • io.druid.query.groupby.GroupByQueryRunnerTestHelper
  • io.druid.query.groupby.GroupByTimeseriesQueryRunnerTest
  • io.druid.query.metadata.SegmentAnalyzerTest
  • io.druid.query.spec.SpecificSegmentQueryRunnerTest
  • io.druid.query.timeseries.TimeSeriesUnionQueryRunnerTest
  • io.druid.segment.AppendTest
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.