public JdbcMessageReceiver(
UMOConnector connector,
UMOComponent component,
UMOEndpoint endpoint,
String readStmt,
String ackStmt)
throws InitialisationException {
super(
connector,
component,
endpoint,
new Long(((JdbcConnector) connector).getPollingFrequency()));
this.receiveMessagesInTransaction = false;
this.connector = (JdbcConnector) connector;
this.readParams = new ArrayList();
this.readStmt = JdbcUtils.parseStatement(readStmt, this.readParams);
this.ackParams = new ArrayList();
this.ackStmt = JdbcUtils.parseStatement(ackStmt, this.ackParams);
}
/*
* (non-Javadoc)
*
* @see org.mule.providers.AbstractMessageDispatcher#doDispatch(org.mule.umo.UMOEvent)
*/
protected void doDispatch(UMOEvent event) throws Exception {
if (logger.isDebugEnabled()) {
logger.debug("Dispatch event: " + event);
}
UMOImmutableEndpoint endpoint = event.getEndpoint();
String writeStmt = endpoint.getEndpointURI().getAddress();
String str;
if ((str = this.connector.getQuery(endpoint, writeStmt)) != null) {
writeStmt = str;
}
if (StringUtils.isEmpty(writeStmt)) {
throw new IllegalArgumentException("Write statement should not be null");
}
if (!"insert".equalsIgnoreCase(writeStmt.substring(0, 6))
&& !"update".equalsIgnoreCase(writeStmt.substring(0, 6))
&& !"delete".equalsIgnoreCase(writeStmt.substring(0, 6))) {
throw new IllegalArgumentException(
"Write statement should be an insert / update / delete sql statement");
}
List paramNames = new ArrayList();
writeStmt = JdbcUtils.parseStatement(writeStmt, paramNames);
Object[] paramValues = JdbcUtils.getParams(endpoint, paramNames, event.getTransformedMessage());
UMOTransaction tx = TransactionCoordination.getInstance().getTransaction();
Connection con = null;
try {
con = this.connector.getConnection();
int nbRows = connector.createQueryRunner().update(con, writeStmt, paramValues);
if (nbRows != 1) {
logger.warn("Row count for write should be 1 and not " + nbRows);
}
if (tx == null) {
JdbcUtils.commitAndClose(con);
}
logger.debug("Event dispatched succesfuly");
} catch (Exception e) {
logger.debug("Error dispatching event: " + e.getMessage(), e);
if (tx == null) {
JdbcUtils.rollbackAndClose(con);
}
throw e;
}
}
/**
* Make a specific request to the underlying transport
*
* @param endpoint the endpoint to use when connecting to the resource
* @param timeout the maximum time the operation should block before returning. The call should
* return immediately if there is data available. If no data becomes available before the
* timeout elapses, null will be returned
* @return the result of the request wrapped in a UMOMessage object. Null will be returned if no
* data was avaialable
* @throws Exception if the call to the underlying protocal cuases an exception
*/
protected UMOMessage doReceive(UMOImmutableEndpoint endpoint, long timeout) throws Exception {
if (logger.isDebugEnabled()) {
logger.debug("Trying to receive a message with a timeout of " + timeout);
}
String[] stmts = this.connector.getReadAndAckStatements(endpoint);
String readStmt = stmts[0];
String ackStmt = stmts[1];
List readParams = new ArrayList();
List ackParams = new ArrayList();
readStmt = JdbcUtils.parseStatement(readStmt, readParams);
ackStmt = JdbcUtils.parseStatement(ackStmt, ackParams);
Connection con = null;
long t0 = System.currentTimeMillis();
try {
con = this.connector.getConnection();
if (timeout < 0) {
timeout = Long.MAX_VALUE;
}
Object result = null;
do {
result =
connector
.createQueryRunner()
.query(
con,
readStmt,
JdbcUtils.getParams(endpoint, readParams, null),
new MapHandler());
if (result != null) {
if (logger.isDebugEnabled()) {
logger.debug("Received: " + result);
}
break;
}
long sleep =
Math.min(
this.connector.getPollingFrequency(), timeout - (System.currentTimeMillis() - t0));
if (sleep > 0) {
if (logger.isDebugEnabled()) {
logger.debug("No results, sleeping for " + sleep);
}
Thread.sleep(sleep);
} else {
logger.debug("Timeout");
return null;
}
} while (true);
if (result != null && ackStmt != null) {
int nbRows =
connector
.createQueryRunner()
.update(con, ackStmt, JdbcUtils.getParams(endpoint, ackParams, result));
if (nbRows != 1) {
logger.warn("Row count for ack should be 1 and not " + nbRows);
}
}
UMOMessageAdapter msgAdapter = this.connector.getMessageAdapter(result);
UMOMessage message = new MuleMessage(msgAdapter);
JdbcUtils.commitAndClose(con);
return message;
} catch (Exception e) {
JdbcUtils.rollbackAndClose(con);
throw e;
}
}