use of org.teiid.client.util.ResultsFuture in project teiid by teiid.
the class TestAsynch method testAsynchAnon.
/**
* Runs sufficient iterations to ensure the temptablestore state is correct
* @throws Exception
*/
@Test
public void testAsynchAnon() throws Exception {
String query = "begin\n" + "insert into #mom_collectors select 'a' as hostname, 123 as port, 1 as id from sometable;\n" + "insert into #apm_collectors select 'a' as hostname, 124 as port, 12 as id from sometable;\n" + "select 'add', hostname, port, id from #mom_collectors\n" + "where (hostname, port) not in (select (hostname, port) from #apm_collectors)\n" + "union select 'delete', hostname, port, id from #apm_collectors\n" + "where (hostname, port) not in (select (hostname, port) from #mom_collectors) with return; end";
Statement stmt = this.internalConnection.createStatement();
TeiidStatement ts = stmt.unwrap(TeiidStatement.class);
ef.addData("SELECT someTable.col FROM someTable", Collections.nCopies(1, Arrays.asList(1)));
ts.setExecutionProperty("autoCommitTxn", "off");
final ResultsFuture<Integer> result = new ResultsFuture<Integer>();
ts.submitExecute(query, new StatementCallback() {
int rowCount;
@Override
public void onRow(Statement s, ResultSet rs) {
try {
rowCount++;
if (rowCount == 100000) {
s.close();
}
} catch (SQLException e) {
result.getResultsReceiver().exceptionOccurred(e);
throw new RuntimeException(e);
}
}
@Override
public void onException(Statement s, Exception e) {
result.getResultsReceiver().exceptionOccurred(e);
}
@Override
public void onComplete(Statement s) {
result.getResultsReceiver().receiveResults(rowCount);
}
}, new RequestOptions().continuous(true));
assertEquals(100000, result.get().intValue());
}
use of org.teiid.client.util.ResultsFuture in project teiid by teiid.
the class TestAsynch method testAsynch.
@Test
public void testAsynch() throws Exception {
Statement stmt = this.internalConnection.createStatement();
TeiidStatement ts = stmt.unwrap(TeiidStatement.class);
final ResultsFuture<Integer> result = new ResultsFuture<Integer>();
ts.submitExecute("select * from sys.tables a, sys.tables b, sys.tables c", new StatementCallback() {
int rowCount;
@Override
public void onRow(Statement s, ResultSet rs) {
rowCount++;
try {
if (!rs.isLast()) {
assertTrue(rs.unwrap(TeiidResultSet.class).available() > 0);
}
if (rowCount == 10000) {
s.close();
}
} catch (AsynchPositioningException e) {
try {
assertEquals(0, rs.unwrap(TeiidResultSet.class).available());
} catch (SQLException e1) {
result.getResultsReceiver().exceptionOccurred(e1);
}
} catch (SQLException e) {
result.getResultsReceiver().exceptionOccurred(e);
}
}
@Override
public void onException(Statement s, Exception e) {
result.getResultsReceiver().exceptionOccurred(e);
}
@Override
public void onComplete(Statement s) {
result.getResultsReceiver().receiveResults(rowCount);
}
}, new RequestOptions());
assertEquals(10000, result.get().intValue());
}
use of org.teiid.client.util.ResultsFuture in project teiid by teiid.
the class TestAsynch method testAsynchContinuous.
@Test
public void testAsynchContinuous() throws Exception {
Statement stmt = this.internalConnection.createStatement();
TeiidStatement ts = stmt.unwrap(TeiidStatement.class);
final ResultsFuture<Integer> result = new ResultsFuture<Integer>();
ts.submitExecute("select xmlelement(name x) from SYS.Schemas", new StatementCallback() {
int rowCount;
@Override
public void onRow(Statement s, ResultSet rs) throws SQLException {
rowCount++;
if (rowCount == 1024) {
s.close();
}
}
@Override
public void onException(Statement s, Exception e) {
result.getResultsReceiver().exceptionOccurred(e);
}
@Override
public void onComplete(Statement s) {
result.getResultsReceiver().receiveResults(rowCount);
}
}, new RequestOptions().continuous(true));
assertEquals(1024, result.get().intValue());
}
use of org.teiid.client.util.ResultsFuture in project teiid by teiid.
the class TestAsynch method testAsynchContinuousWithAlter.
@Test
public void testAsynchContinuousWithAlter() throws Exception {
Statement stmt = this.internalConnection.createStatement();
TeiidStatement ts = stmt.unwrap(TeiidStatement.class);
final ResultsFuture<Integer> result = new ResultsFuture<Integer>();
ts.submitExecute("select * from test", new StatementCallback() {
int rowCount;
@Override
public void onRow(Statement s, ResultSet rs) {
try {
rowCount++;
if (rowCount < 3) {
assertEquals(1, rs.getInt(1));
if (rowCount == 2) {
Statement st = internalConnection.createStatement();
st.execute("alter view v.test as select 2");
st.close();
try {
// we only track down to millisecond resolution
Thread.sleep(100);
} catch (InterruptedException e) {
result.getResultsReceiver().exceptionOccurred(e);
}
}
} else {
// new value
assertEquals(2, rs.getInt(1));
s.close();
}
} catch (SQLException e) {
result.getResultsReceiver().exceptionOccurred(e);
throw new RuntimeException(e);
}
}
@Override
public void onException(Statement s, Exception e) {
result.getResultsReceiver().exceptionOccurred(e);
}
@Override
public void onComplete(Statement s) {
result.getResultsReceiver().receiveResults(rowCount);
}
}, new RequestOptions().continuous(true));
assertEquals(3, result.get().intValue());
}
Aggregations