Package org.apache.drill.exec.rpc.user

Examples of org.apache.drill.exec.rpc.user.QueryResultBatch

57
58
59
60
61
62
63
64
65
66
67
68
                    Files.toString(FileUtils.getResourceAsFile("/record/vector/test_date.json"), Charsets.UTF_8)
                            .replace("#{TEST_FILE}", "/test_simple_date.json"));

            RecordBatchLoader batchLoader = new RecordBatchLoader(bit.getContext().getAllocator());

            QueryResultBatch batch = results.get(1);
            assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));

            for (VectorWrapper<?> v : batchLoader) {

                ValueVector.Accessor accessor = v.getValueVector().getAccessor();
View Full Code Here
91
92
93
94
95
96
97
98
99
100
101
102
                    Files.toString(FileUtils.getResourceAsFile("/record/vector/test_sort_date.json"), Charsets.UTF_8)
                            .replace("#{TEST_FILE}", "/test_simple_date.json"));

            RecordBatchLoader batchLoader = new RecordBatchLoader(bit.getContext().getAllocator());

            QueryResultBatch batch = results.get(1);
            assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));

            for (VectorWrapper<?> v : batchLoader) {

                ValueVector.Accessor accessor = v.getValueVector().getAccessor();
View Full Code Here
125
126
127
128
129
130
131
132
133
134
135
136
                    Files.toString(FileUtils.getResourceAsFile("/record/vector/test_timestamp.json"), Charsets.UTF_8)
                            .replace("#{TEST_FILE}", "/test_simple_date.json"));

            RecordBatchLoader batchLoader = new RecordBatchLoader(bit.getContext().getAllocator());

            QueryResultBatch batch = results.get(1);
            assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));

            for (VectorWrapper<?> v : batchLoader) {

                ValueVector.Accessor accessor = v.getValueVector().getAccessor();
View Full Code Here
159
160
161
162
163
164
165
166
167
168
169
170
                    Files.toString(FileUtils.getResourceAsFile("/record/vector/test_interval.json"), Charsets.UTF_8)
                            .replace("#{TEST_FILE}", "/test_simple_interval.json"));

            RecordBatchLoader batchLoader = new RecordBatchLoader(bit.getContext().getAllocator());

            QueryResultBatch batch = results.get(1);
            assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));

            Iterator<VectorWrapper<?>> itr = batchLoader.iterator();

            ValueVector.Accessor accessor = itr.next().getValueVector().getAccessor();
View Full Code Here
217
218
219
220
221
222
223
224
225
226
227
228
                    Files.toString(FileUtils.getResourceAsFile("/record/vector/test_all_date_literals.json"), Charsets.UTF_8)
                            .replace("#{TEST_FILE}", "/test_simple_date.json"));

            RecordBatchLoader batchLoader = new RecordBatchLoader(bit.getContext().getAllocator());

            QueryResultBatch batch = results.get(1);
            assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));

            String result[] = {"2008-02-27",
                               "2008-02-27 01:02:03.000",
                               "2008-02-27 01:02:03.000 UTC",
                               "10:11:13.999",
View Full Code Here
257
258
259
260
261
262
263
264
265
266
267
268
                    Files.toString(FileUtils.getResourceAsFile("/record/vector/test_date_add.json"), Charsets.UTF_8)
                            .replace("#{TEST_FILE}", "/test_simple_date.json"));

            RecordBatchLoader batchLoader = new RecordBatchLoader(bit.getContext().getAllocator());

            QueryResultBatch batch = results.get(1);
            assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));

            for (VectorWrapper<?> v : batchLoader) {

                ValueVector.Accessor accessor = v.getValueVector().getAccessor();
View Full Code Here
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
    List<QueryResultBatch> result = testSqlWithResults("select sum(a) as total_sum from dfs.`/tmp/parquet_with_nulls_should_sum_100000_nulls_first.parquet`");
    assertEquals("Only expected one batch with data, and then the empty finishing batch.", 2, result.size());
    RecordBatchLoader loader = new RecordBatchLoader(bit.getContext()
        .getAllocator());

    QueryResultBatch b = result.get(0);
    loader.load(b.getHeader().getDef(), b.getData());

    VectorWrapper vw = loader.getValueAccessorById(
        NullableBigIntVector.class, //
        loader.getValueVectorId(SchemaPath.getCompoundPath("total_sum")).getFieldIds() //
    );
    assertEquals(4999950000l, vw.getValueVector().getAccessor().getObject(0));
    b.release();
    loader.clear();
  }
View Full Code Here
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
    List<QueryResultBatch> result = testSqlWithResults("select count(wr_return_quantity) as row_count from dfs.`/tmp/web_returns` where wr_return_quantity = 1");
    assertEquals("Only expected one batch with data, and then the empty finishing batch.", 2, result.size());
    RecordBatchLoader loader = new RecordBatchLoader(bit.getContext()
        .getAllocator());

    QueryResultBatch b = result.get(0);
    loader.load(b.getHeader().getDef(), b.getData());

    VectorWrapper vw = loader.getValueAccessorById(
        BigIntVector.class, //
        loader.getValueVectorId(SchemaPath.getCompoundPath("row_count")).getFieldIds() //
    );
    assertEquals(3573l, vw.getValueVector().getAccessor().getObject(0));
    b.release();
    loader.clear();
  }
View Full Code Here
60
61
62
63
64
65
66
67
68
69
70
71
            List<QueryResultBatch> results = client.runQuery(org.apache.drill.exec.proto.UserBitShared.QueryType.PHYSICAL,
                    Files.toString(FileUtils.getResourceAsFile("/functions/multi_input_add_test.json"), Charsets.UTF_8));

            RecordBatchLoader batchLoader = new RecordBatchLoader(bit.getContext().getAllocator());

            QueryResultBatch batch = results.get(1);
            assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));

            for (VectorWrapper<?> v : batchLoader) {

                ValueVector.Accessor accessor = v.getValueVector().getAccessor();
View Full Code Here
57
58
59
60
61
62
63
64
65
66
67
68
69
              inputDataFile));

      RecordBatchLoader batchLoader = new RecordBatchLoader(bit
          .getContext().getAllocator());

      QueryResultBatch batch = results.get(1);
      assertTrue(batchLoader.load(batch.getHeader().getDef(),
          batch.getData()));

      int i = 0;
      for (VectorWrapper<?> v : batchLoader) {
        ValueVector.Accessor accessor = v.getValueVector().getAccessor();
        System.out.println((accessor.getObject(0)));
View Full Code Here
TOP

Related Classes of org.apache.drill.exec.rpc.user.QueryResultBatch

  • org.apache.drill.exec.fn.impl.TestAggregateFunction
  • org.apache.drill.exec.fn.impl.TestDateFunctions
  • org.apache.drill.exec.fn.impl.TestMultiInputAdd
  • org.apache.drill.exec.fn.impl.TestNewAggregateFunctions
  • org.apache.drill.exec.physical.impl.join.TestHashJoin
  • org.apache.drill.exec.physical.impl.TestCastFunctions
  • org.apache.drill.exec.physical.impl.TestCastVarCharToBigInt
  • org.apache.drill.exec.physical.impl.TestDecimal
  • org.apache.drill.exec.physical.impl.TestExtractFunctions
  • org.apache.drill.exec.physical.impl.TestReverseImplicitCast
    • 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.