Package org.teiid.client

Examples of org.teiid.client.ResultsMessage


  @Test public void testBatchExecution() throws Exception {
    ConnectionImpl conn = Mockito.mock(ConnectionImpl.class);
    DQP dqp = Mockito.mock(DQP.class);
    ResultsFuture<ResultsMessage> results = new ResultsFuture<ResultsMessage>();
    Mockito.stub(dqp.executeRequest(Mockito.anyLong(), (RequestMessage)Mockito.anyObject())).toReturn(results);
    ResultsMessage rm = new ResultsMessage();
    rm.setResults(new List<?>[] {Arrays.asList(1), Arrays.asList(2)});
    rm.setUpdateResult(true);
    results.getResultsReceiver().receiveResults(rm);
    Mockito.stub(conn.getDQP()).toReturn(dqp);
    StatementImpl statement = new StatementImpl(conn, ResultSet.TYPE_FORWARD_ONLY, ResultSet.CONCUR_READ_ONLY);
    statement.addBatch("delete from table"); //$NON-NLS-1$
    statement.addBatch("delete from table1"); //$NON-NLS-1$
View Full Code Here


        cs.close();     
    }
   
    @Test public void testOutputParameter() throws Exception {
        StatementImpl statement = createMockStatement(ResultSet.TYPE_FORWARD_ONLY);
        ResultsMessage resultsMsg = new ResultsMessage();
        resultsMsg.setResults(new List<?>[] {Arrays.asList(1, null, null), Arrays.asList(null, 2, 3)});
        resultsMsg.setLastRow(2);
        resultsMsg.setFirstRow(1);
        resultsMsg.setFinalRow(2);
        resultsMsg.setColumnNames(new String[] {"x", "out1", "out2"}); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
        resultsMsg.setDataTypes(new String[] {"integer", "integer", "integer"}); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
        ResultSetImpl cs = new ResultSetImpl(resultsMsg, statement, null, 2);
       
        int count = 0;
        while (cs.next()) {
            cs.getObject(1);
View Full Code Here

      StatementImpl statement = createMockStatement(ResultSet.TYPE_FORWARD_ONLY);
      ResultsFuture<LobChunk> future = new ResultsFuture<LobChunk>();
      future.getResultsReceiver().receiveResults(new LobChunk("<a/>".getBytes(Charset.forName("UTF-8")), true));
      XMLType result = new XMLType();
      Mockito.stub(statement.getDQP().requestNextLobChunk(0, 0, result.getReferenceStreamId())).toReturn(future);
        ResultsMessage resultsMsg = new ResultsMessage();
        result.setEncoding("UTF-8");
        resultsMsg.setResults(new List<?>[] {Arrays.asList(result)});
        resultsMsg.setLastRow(1);
        resultsMsg.setFirstRow(1);
        resultsMsg.setFinalRow(1);
        resultsMsg.setColumnNames(new String[] {"x"}); //$NON-NLS-1$
        resultsMsg.setDataTypes(new String[] {"xml"}); //$NON-NLS-1$
        ResultSetImpl cs = new ResultSetImpl(resultsMsg, statement);
        cs.next();
        assertEquals("<a/>", cs.getString(1));
    }
View Full Code Here

  public void testGetOutputParameter() throws Exception {
    CallableStatementImpl mmcs = getCallableStatement();
   
    RequestMessage request = new RequestMessage();
    request.setExecutionId(1);
    ResultsMessage resultsMsg = new ResultsMessage(request);
    List[] results = new List[] {Arrays.asList(null, null, null), Arrays.asList(null, 1, 2)};
    resultsMsg.setResults(results);
    resultsMsg.setColumnNames(new String[] { "IntNum", "Out1", "Out2" }); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
    resultsMsg.setDataTypes(new String[] { JDBCSQLTypeInfo.INTEGER, JDBCSQLTypeInfo.INTEGER, JDBCSQLTypeInfo.INTEGER });
    resultsMsg.setFinalRow(results.length);
    resultsMsg.setLastRow(results.length);
    resultsMsg.setFirstRow(1);
    resultsMsg.setParameters(Arrays.asList(new ParameterInfo(ParameterInfo.RESULT_SET, 1), new ParameterInfo(ParameterInfo.OUT, 1), new ParameterInfo(ParameterInfo.OUT, 1)));
    mmcs.createResultSet(resultsMsg);
    assertEquals(1, mmcs.getInt(1));
    assertEquals(2, mmcs.getInt(2));
  }
View Full Code Here

   
    public Batch requestBatch(int beginRow) throws SQLException{
      checkClosed();
        try {
          ResultsFuture<ResultsMessage> results = submitRequestBatch(beginRow);
          ResultsMessage currentResultMsg = getResults(results);
            return processBatch(currentResultMsg);
    } catch (InterruptedException e) {
      throw TeiidSQLException.create(e);
    } catch (ExecutionException e) {
      throw TeiidSQLException.create(e);
View Full Code Here

      TimeoutException {
    int timeoutSeconds = statement.getQueryTimeout();
    if (timeoutSeconds == 0) {
      timeoutSeconds = Integer.MAX_VALUE;
    }
    ResultsMessage currentResultMsg = results.get(timeoutSeconds, TimeUnit.SECONDS);
    return currentResultMsg;
  }
View Full Code Here

TOP

Related Classes of org.teiid.client.ResultsMessage

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.