Package org.teiid.query.processor

Examples of org.teiid.query.processor.BatchIterator

80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
        selectNode.addChild(child);
        selectNode.initialize(context, mgr, dataMgr);
       
        selectNode.open();
       
        BatchIterator iterator = new BatchIterator(selectNode);
       
        for (int i = 0; i < expected.length; i++) {
          while (true) {
            try {
              assertEquals("Rows don't match at " + i, expected[i], iterator.nextTuple()); //$NON-NLS-1$
              break;
            } catch (BlockedException e) {
              continue;
            } catch (QueryProcessor.ExpiredTimeSliceException e) {
              continue;
            }
          }
    } 
        assertFalse(iterator.hasNext());
  }
View Full Code Here
90
91
92
93
94
95
96
97
98
99
100
        return outputPhase();
    }

    private void sortPhase() throws BlockedException, TeiidComponentException, TeiidProcessingException {
      if (this.sortUtility == null) {
          this.sortUtility = new SortUtility(new BatchIterator(getChildren()[0]), items, this.mode, getBufferManager(),
                                              getConnectionID(), getChildren()[0].getElements());
    }
    this.output = this.sortUtility.sort();
    if (this.outputTs == null) {
      this.outputTs = this.output.createIndexedTupleSource();
View Full Code Here
476
477
478
479
480
481
482
483
484
485
486
            CommandContext subContext = getContext().clone();
            subContext.setVariableContext(this.currentVarContext);
            subContext.setTempTableStore(getTempTableStore());
            state = new CursorState();
            state.processor = new QueryProcessor(command, subContext, this.bufferMgr, this.dataMgr);
            state.ts = new BatchIterator(state.processor);
            if (procAssignments != null && state.processor.getOutputElements().size() - procAssignments.size() > 0) {
              state.resultsBuffer = bufferMgr.createTupleBuffer(state.processor.getOutputElements().subList(0, state.processor.getOutputElements().size() - procAssignments.size()), getContext().getConnectionID(), TupleSourceType.PROCESSOR);
            }
              this.currentState = state;
          }
View Full Code Here
131
132
133
134
135
136
137
138
139
140
141
142
143
        if (this.iterator == null) {
            if (this.buffer != null) {
                iterator = buffer.createIndexedTupleSource();
            } else {
                // return a TupleBatch tuplesource iterator
                BatchIterator bi = new BatchIterator(this.source);
                if (implicitBuffer != ImplicitBuffer.NONE) {
                  bi.setBuffer(createSourceTupleBuffer(), implicitBuffer == ImplicitBuffer.ON_MARK);
                }
                this.iterator = bi;
            }
        }
        return this.iterator;
View Full Code Here
192
193
194
195
196
197
198
199
200
201
202
        TupleSource ts = null;
        if (this.buffer != null) {
          this.buffer.setForwardOnly(true);
          ts = this.buffer.createIndexedTupleSource();
        } else {
          ts = new BatchIterator(this.source);
        }
        this.sortUtility = new SortUtility(ts, expressions, Collections.nCopies(expressions.size(), OrderBy.ASC),
            sortOption == SortOption.SORT_DISTINCT?Mode.DUP_REMOVE_SORT:Mode.SORT, this.source.getBufferManager(), this.source.getConnectionID(), source.getElements());
        this.markDistinct(sortOption == SortOption.SORT_DISTINCT && expressions.size() == this.getOuterVals().size());
    }
View Full Code Here
90
91
92
93
94
95
96
97
98
99
     * @see org.teiid.query.processor.xml.PlanExecutor#execute(java.util.Map)
     */
    public void execute(Map referenceValues) throws TeiidComponentException, BlockedException, TeiidProcessingException {       
        if (this.tupleSource == null) {
          setReferenceValues(referenceValues);
            this.tupleSource = new BatchIterator(internalProcessor);
        }
        //force execution
        this.tupleSource.hasNext();
    }   
View Full Code Here
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
@SuppressWarnings("nls")
public class TestBatchIterator {

  @Test public void testReset() throws Exception {
    BatchIterator bi = new BatchIterator(new FakeRelationalNode(1, new List[] {
      Arrays.asList(1),
      Arrays.asList(1),
      Arrays.asList(1)
    }, 1));
    BufferManager bm = BufferManagerFactory.getStandaloneBufferManager();
    bi.setBuffer(bm.createTupleBuffer(Arrays.asList(new ElementSymbol("x")), "test", TupleSourceType.PROCESSOR), true);  //$NON-NLS-1$
    bi.mark();
    bi.nextTuple();
    bi.nextTuple();
    bi.reset();
    bi.nextTuple();
  }
View Full Code Here
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
    bi.reset();
    bi.nextTuple();
  }
 
  @Test public void testReset1() throws Exception {
    BatchIterator bi = new BatchIterator(new FakeRelationalNode(1, new List[] {
      Arrays.asList(1),
      Arrays.asList(2),
      Arrays.asList(3)
    }, 2));
    BufferManager bm = BufferManagerFactory.getStandaloneBufferManager();
    TupleBuffer tb = bm.createTupleBuffer(Arrays.asList(new ElementSymbol("x")), "test", TupleSourceType.PROCESSOR);
    bi.setBuffer(tb, true);  //$NON-NLS-1$
    bi.nextTuple();
    bi.mark();
    bi.nextTuple();
    bi.reset();
    assertEquals(2, bi.getCurrentIndex());
    assertEquals(2, bi.nextTuple().get(0));
  }
View Full Code Here
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
    assertEquals(2, bi.getCurrentIndex());
    assertEquals(2, bi.nextTuple().get(0));
  }
 
  @Test public void testReset2() throws Exception {
    BatchIterator bi = new BatchIterator(new FakeRelationalNode(1, new List[] {
      Arrays.asList(1),
      Arrays.asList(2),
    }, 2));
    BufferManager bm = BufferManagerFactory.getStandaloneBufferManager();
    TupleBuffer tb = bm.createTupleBuffer(Arrays.asList(new ElementSymbol("x")), "test", TupleSourceType.PROCESSOR);
    bi.setBuffer(tb, true);  //$NON-NLS-1$
    bi.hasNext();
    bi.mark();
    bi.nextTuple();
    bi.nextTuple();
    assertNull(bi.nextTuple());
    bi.reset();
    bi.hasNext();
    assertEquals(1, bi.getCurrentIndex());
    assertEquals(1, bi.nextTuple().get(0));
  }
View Full Code Here
TOP

Related Classes of org.teiid.query.processor.BatchIterator

  • org.teiid.query.processor.proc.ProcedurePlan
  • org.teiid.query.processor.relational.SortNode
  • org.teiid.query.processor.relational.SourceState
  • org.teiid.query.processor.relational.TestSelectNode
  • org.teiid.query.processor.TestBatchIterator
  • org.teiid.query.processor.xml.RelationalPlanExecutor
    • 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.