Java tutorial
/******************************************************************************* * Copyright 2011 Clockwork * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. ******************************************************************************/ package nl.clockwork.mule.ebms.dao; import java.sql.Connection; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; import java.sql.Timestamp; import java.util.ArrayList; import java.util.Date; import java.util.List; import javax.activation.DataSource; import javax.mail.util.ByteArrayDataSource; import javax.xml.bind.JAXBException; import nl.clockwork.common.dao.DAOException; import nl.clockwork.common.util.XMLMessageBuilder; import nl.clockwork.mule.ebms.Constants; import nl.clockwork.mule.ebms.Constants.EbMSMessageStatus; import nl.clockwork.mule.ebms.Constants.EbMSMessageType; import nl.clockwork.mule.ebms.model.EbMSAcknowledgment; import nl.clockwork.mule.ebms.model.EbMSBaseMessage; import nl.clockwork.mule.ebms.model.EbMSMessage; import nl.clockwork.mule.ebms.model.EbMSMessageContext; import nl.clockwork.mule.ebms.model.EbMSMessageError; import nl.clockwork.mule.ebms.model.EbMSSendEvent; import nl.clockwork.mule.ebms.model.cpp.cpa.CollaborationProtocolAgreement; import nl.clockwork.mule.ebms.model.ebxml.AckRequested; import nl.clockwork.mule.ebms.model.ebxml.Acknowledgment; import nl.clockwork.mule.ebms.model.ebxml.ErrorList; import nl.clockwork.mule.ebms.model.ebxml.Manifest; import nl.clockwork.mule.ebms.model.ebxml.MessageHeader; import nl.clockwork.mule.ebms.model.ebxml.MessageOrder; import nl.clockwork.mule.ebms.model.ebxml.SyncReply; import nl.clockwork.mule.ebms.model.xml.xmldsig.ObjectFactory; import nl.clockwork.mule.ebms.model.xml.xmldsig.SignatureType; import org.apache.commons.io.IOUtils; import org.springframework.dao.DataAccessException; import org.springframework.dao.EmptyResultDataAccessException; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.jdbc.core.PreparedStatementCreator; import org.springframework.jdbc.core.simple.ParameterizedRowMapper; import org.springframework.jdbc.core.simple.SimpleJdbcTemplate; import org.springframework.jdbc.support.GeneratedKeyHolder; import org.springframework.jdbc.support.KeyHolder; import org.springframework.transaction.PlatformTransactionManager; import org.springframework.transaction.TransactionStatus; import org.springframework.transaction.support.TransactionCallbackWithoutResult; import org.springframework.transaction.support.TransactionTemplate; public abstract class AbstractEbMSDAO implements EbMSDAO { public class EbMSBaseMessageParameterizedRowMapper implements ParameterizedRowMapper<EbMSBaseMessage> { @Override public EbMSBaseMessage mapRow(ResultSet rs, int rowNum) throws SQLException { try { if (EbMSMessageType.MESSAGE_ERROR.action().getService().getValue().equals(rs.getString("service")) && EbMSMessageType.MESSAGE_ERROR.action().getAction().equals(rs.getString("action"))) return new EbMSMessageError( XMLMessageBuilder.getInstance(MessageHeader.class) .handle(rs.getString("message_header")), XMLMessageBuilder.getInstance(ErrorList.class).handle(rs.getString("content"))); else if (EbMSMessageType.ACKNOWLEDGMENT.action().getService().getValue() .equals(rs.getString("service")) && EbMSMessageType.ACKNOWLEDGMENT.action().getAction().equals(rs.getString("action"))) return new EbMSAcknowledgment( XMLMessageBuilder.getInstance(MessageHeader.class) .handle(rs.getString("message_header")), XMLMessageBuilder.getInstance(Acknowledgment.class).handle(rs.getString("content"))); else return new EbMSMessage( XMLMessageBuilder.getInstance(MessageHeader.class) .handle(rs.getString("message_header")), XMLMessageBuilder.getInstance(AckRequested.class).handle(rs.getString("ack_requested")), XMLMessageBuilder.getInstance(Manifest.class).handle(rs.getString("content")), getAttachments(rs.getLong("id"))); } catch (JAXBException e) { throw new SQLException(e); } catch (DAOException e) { throw new SQLException(e); } } } public class EbMSMessagePreparedStatement implements PreparedStatementCreator { private Date timestamp; private String cpaId; private String conversationId; private Long sequenceNr; private String messageId; private String refToMessageId; private String fromRole; private String toRole; private String serviceType; private String service; private String action; private byte[] original; private String signature; private String messageHeader; private String syncReply; private String messageOrder; private String ackRequested; private String content; private EbMSMessageStatus status; public EbMSMessagePreparedStatement(Date timestamp, String cpaId, String conversationId, String messageId, String refToMessageId, String fromRole, String toRole, String serviceType, String service, String action, String messageHeader, String content) { this(timestamp, cpaId, conversationId, null, messageId, refToMessageId, fromRole, toRole, serviceType, service, action, null, null, messageHeader, null, null, null, content, null); } public EbMSMessagePreparedStatement(Date timestamp, String cpaId, String conversationId, String messageId, String refToMessageId, String fromRole, String toRole, String serviceType, String service, String action, String messageHeader, String content, EbMSMessageStatus status) { this(timestamp, cpaId, conversationId, null, messageId, refToMessageId, fromRole, toRole, serviceType, service, action, null, null, messageHeader, null, null, null, content, status); } public EbMSMessagePreparedStatement(Date timestamp, String cpaId, String conversationId, Long sequenceNr, String messageId, String refToMessageId, String fromRole, String toRole, String serviceType, String service, String action, String messageHeader, String syncReply, String messageOrder, String ackRequested, String content) { this(timestamp, cpaId, conversationId, sequenceNr, messageId, refToMessageId, fromRole, toRole, serviceType, service, action, null, null, messageHeader, syncReply, messageOrder, ackRequested, content, null); } public EbMSMessagePreparedStatement(Date timestamp, String cpaId, String conversationId, Long sequenceNr, String messageId, String refToMessageId, String fromRole, String toRole, String serviceType, String service, String action, byte[] original, String signature, String messageHeader, String syncReply, String messageOrder, String ackRequested, String content, EbMSMessageStatus status) { this.timestamp = timestamp; this.cpaId = cpaId; this.conversationId = conversationId; this.sequenceNr = sequenceNr; this.messageId = messageId; this.refToMessageId = refToMessageId; this.fromRole = fromRole; this.toRole = toRole; this.serviceType = serviceType; this.service = service; this.action = action; this.original = original; this.signature = signature; this.messageHeader = messageHeader; this.syncReply = syncReply; this.messageOrder = messageOrder; this.ackRequested = ackRequested; this.content = content; this.status = status; } public PreparedStatement createPreparedStatement(Connection connection) throws SQLException { PreparedStatement ps = connection.prepareStatement( "insert into ebms_message (" + "time_stamp," + "cpa_id," + "conversation_id," + "sequence_nr," + "message_id," + "ref_to_message_id," + "from_role," + "to_role," + "service_type," + "service," + "action," + "original," + "signature," + "message_header," + "sync_reply," + "message_order," + "ack_requested," + "content," + "status," + "status_time" + ") values (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?," + (status == null ? "null" : getTimestampFunction()) + ")", //new String[]{"id"} new int[] { 1 }); //ps.setDate(1,new java.sql.Date(timestamp.getTime())); //ps.setString(1,String.format(getDateFormat(),timestamp)); ps.setTimestamp(1, new Timestamp(timestamp.getTime())); //ps.setObject(1,timestamp,Types.TIMESTAMP); //ps.setObject(1,timestamp); ps.setString(2, cpaId); ps.setString(3, conversationId); if (sequenceNr == null) ps.setNull(4, java.sql.Types.BIGINT); else ps.setLong(4, sequenceNr); ps.setString(5, messageId); ps.setString(6, refToMessageId); ps.setString(7, fromRole); ps.setString(8, toRole); ps.setString(9, serviceType); ps.setString(10, service); ps.setString(11, action); ps.setBytes(12, original); ps.setString(13, signature); ps.setString(14, messageHeader); ps.setString(15, syncReply); ps.setString(16, messageOrder); ps.setString(17, ackRequested); ps.setString(18, content); if (status == null) ps.setNull(19, java.sql.Types.INTEGER); else ps.setInt(19, status.id()); //ps.setString(20,status == null ? null : String.format(getDateFormat(),timestamp)); //ps.setTimestamp(20,status == null ? null : new Timestamp(timestamp.getTime())); //ps.setObject(20,status == null ? null : timestamp,Types.TIMESTAMP); //ps.setObject(20,status == null ? null : timestamp); return ps; } } protected TransactionTemplate transactionTemplate; protected JdbcTemplate jdbcTemplate; protected SimpleJdbcTemplate simpleJdbcTemplate; //public abstract String getDateFormat(); public abstract String getTimestampFunction(); public AbstractEbMSDAO(PlatformTransactionManager transactionManager, javax.sql.DataSource dataSource) { transactionTemplate = new TransactionTemplate(transactionManager); jdbcTemplate = new JdbcTemplate(dataSource); simpleJdbcTemplate = new SimpleJdbcTemplate(dataSource); } public AbstractEbMSDAO(TransactionTemplate transactionTemplate, javax.sql.DataSource dataSource) { this.transactionTemplate = transactionTemplate; jdbcTemplate = new JdbcTemplate(dataSource); simpleJdbcTemplate = new SimpleJdbcTemplate(dataSource); } @Override public boolean existsCPA(String cpaId) throws DAOException { try { return simpleJdbcTemplate.queryForInt("select count(cpa_id)" + " from cpa" + " where cpa_id = ?", cpaId) > 0; } catch (DataAccessException e) { throw new DAOException(e); } } @Override public CollaborationProtocolAgreement getCPA(String cpaId) throws DAOException { try { String result = simpleJdbcTemplate.queryForObject("select cpa" + " from cpa" + " where cpa_id = ?", String.class, cpaId); return XMLMessageBuilder.getInstance(CollaborationProtocolAgreement.class).handle(result); } catch (EmptyResultDataAccessException e) { return null; } catch (DataAccessException e) { throw new DAOException(e); } catch (JAXBException e) { throw new DAOException(e); } } @SuppressWarnings("unchecked") @Override public List<String> getCPAIds() throws DAOException { try { return jdbcTemplate.queryForList("select cpa_id" + " from cpa" + " order by cpa_id desc", String.class); } catch (DataAccessException e) { throw new DAOException(e); } } @Override public boolean insertCPA(CollaborationProtocolAgreement cpa) throws DAOException { try { simpleJdbcTemplate.update("insert into cpa (" + "cpa_id," + "cpa" + ") values (?,?)", cpa.getCpaid(), XMLMessageBuilder.getInstance(CollaborationProtocolAgreement.class).handle(cpa)); return true; } catch (DataAccessException e) { return false; } catch (JAXBException e) { throw new DAOException(e); } } @Override public boolean updateCPA(CollaborationProtocolAgreement cpa) throws DAOException { try { simpleJdbcTemplate.update("update cpa set" + " cpa = ?" + " where cpa_id = ?", XMLMessageBuilder.getInstance(CollaborationProtocolAgreement.class).handle(cpa), cpa.getCpaid()); return true; } catch (DataAccessException e) { return false; } catch (JAXBException e) { throw new DAOException(e); } } @Override public boolean deleteCPA(String cpaId) throws DAOException { try { return simpleJdbcTemplate.update("delete from cpa" + " where cpa_id = ?", cpaId) > 0; } catch (DataAccessException e) { return false; } } @Override public boolean existsMessage(String messageId) throws DAOException { try { return simpleJdbcTemplate.queryForInt( "select count(message_id)" + " from ebms_message" + " where message_id = ?", messageId) > 0; } catch (DataAccessException e) { throw new DAOException(e); } } @Override public Long getEbMSMessageId(String messageId) throws DAOException { try { return simpleJdbcTemplate.queryForLong("select id" + " from ebms_message" + " where message_id = ?", messageId); } catch (EmptyResultDataAccessException e) { return null; } catch (DataAccessException e) { throw new DAOException(e); } } @Override public Long getEbMSMessageResponseId(String messageId) throws DAOException { try { return simpleJdbcTemplate.queryForLong("select id" + " from ebms_message" + " where ref_to_message_id = ?" + " and service = '" + Constants.EBMS_SERVICE_URI + "'" + " and (action = '" + EbMSMessageType.MESSAGE_ERROR.action().getAction() + "'" + " or action = '" + EbMSMessageType.ACKNOWLEDGMENT.action().getAction() + "')", messageId); } catch (EmptyResultDataAccessException e) { return null; } catch (DataAccessException e) { throw new DAOException(e); } } @Override public MessageHeader getMessageHeader(String messageId) throws DAOException { try { String result = simpleJdbcTemplate.queryForObject( "select message_header" + " from ebms_message" + " where message_id = ?", String.class, messageId); return XMLMessageBuilder.getInstance(MessageHeader.class).handle(result); } catch (EmptyResultDataAccessException e) { return null; } catch (DataAccessException e) { throw new DAOException(e); } catch (JAXBException e) { throw new DAOException(e); } } private List<DataSource> getAttachments(long messageId) throws DAOException { try { return simpleJdbcTemplate.query( "select name, content_type, content" + " from ebms_attachment" + " where ebms_message_id = ?", new ParameterizedRowMapper<DataSource>() { @Override public DataSource mapRow(ResultSet rs, int rowNum) throws SQLException { ByteArrayDataSource result = new ByteArrayDataSource(rs.getBytes("content"), rs.getString("content_type")); result.setName(rs.getString("name")); return result; } }, messageId); } catch (DataAccessException e) { throw new DAOException(e); } } @Override public EbMSBaseMessage getMessage(final long id) throws DAOException { try { return simpleJdbcTemplate .queryForObject( "select id, service, action, message_header, ack_requested, content" + " from ebms_message" + " where id = ?", new EbMSBaseMessageParameterizedRowMapper(), id); } catch (EmptyResultDataAccessException e) { return null; } catch (DataAccessException e) { throw new DAOException(e); } } @Override public EbMSMessageStatus getMessageStatus(String messageId) throws DAOException { try { return EbMSMessageStatus.get(simpleJdbcTemplate.queryForObject( "select status" + " from ebms_message" + " where message_id = ?", new ParameterizedRowMapper<Integer>() { @Override public Integer mapRow(ResultSet rs, int rowNum) throws SQLException { return rs.getObject("status") == null ? null : rs.getInt("status"); } }, messageId)); } catch (EmptyResultDataAccessException e) { return EbMSMessageStatus.NOT_RECOGNIZED; } catch (DataAccessException e) { throw new DAOException(e); } } protected PreparedStatementCreator getEbMSMessagePreparedStatement(Date timestamp, String cpaId, String conversationId, String messageId, String refToMessageId, String fromRole, String toRole, String serviceType, String service, String action, String messageHeader, String content) { return new EbMSMessagePreparedStatement(timestamp, cpaId, conversationId, messageId, refToMessageId, fromRole, toRole, serviceType, service, action, messageHeader, content); } protected PreparedStatementCreator getEbMSMessagePreparedStatement(Date timestamp, String cpaId, String conversationId, String messageId, String refToMessageId, String fromRole, String toRole, String serviceType, String service, String action, String messageHeader, String content, EbMSMessageStatus status) { return new EbMSMessagePreparedStatement(timestamp, cpaId, conversationId, messageId, refToMessageId, fromRole, toRole, serviceType, service, action, messageHeader, content, status); } protected PreparedStatementCreator getEbMSMessagePreparedStatement(Date timestamp, String cpaId, String conversationId, Long sequenceNr, String messageId, String refToMessageId, String fromRole, String toRole, String serviceType, String service, String action, String messageHeader, String syncReply, String messageOrder, String ackRequested, String Manifest) { return new EbMSMessagePreparedStatement(timestamp, cpaId, conversationId, sequenceNr, messageId, refToMessageId, fromRole, toRole, serviceType, service, action, messageHeader, syncReply, messageOrder, ackRequested, Manifest); } protected PreparedStatementCreator getEbMSMessagePreparedStatement(Date timestamp, String cpaId, String conversationId, Long sequenceNr, String messageId, String refToMessageId, String fromRole, String toRole, String serviceType, String service, String action, byte[] original, String signature, String messageHeader, String syncReply, String messageOrder, String ackRequested, String Manifest, EbMSMessageStatus status) { return new EbMSMessagePreparedStatement(timestamp, cpaId, conversationId, sequenceNr, messageId, refToMessageId, fromRole, toRole, serviceType, service, action, original, signature, messageHeader, syncReply, messageOrder, ackRequested, Manifest, status); } @Override public void insertMessage(final EbMSMessage message, final List<EbMSSendEvent> sendEvents) throws DAOException { try { transactionTemplate.execute(new TransactionCallbackWithoutResult() { @Override public void doInTransactionWithoutResult(TransactionStatus transactionStatus) { try { Date timestamp = new Date(); KeyHolder keyHolder = new GeneratedKeyHolder(); jdbcTemplate.update(getEbMSMessagePreparedStatement(timestamp, message.getMessageHeader().getCPAId(), message.getMessageHeader().getConversationId(), message.getMessageOrder() == null ? null : message.getMessageOrder().getSequenceNumber().getValue().longValue(), message.getMessageHeader().getMessageData().getMessageId(), message.getMessageHeader().getMessageData().getRefToMessageId(), message.getMessageHeader().getFrom().getRole(), message.getMessageHeader().getTo().getRole(), message.getMessageHeader().getService().getType(), message.getMessageHeader().getService().getValue(), message.getMessageHeader().getAction(), XMLMessageBuilder.getInstance(MessageHeader.class) .handle(message.getMessageHeader()), XMLMessageBuilder.getInstance(SyncReply.class).handle(message.getSyncReply()), XMLMessageBuilder.getInstance(MessageOrder.class).handle(message.getMessageOrder()), XMLMessageBuilder.getInstance(AckRequested.class).handle(message.getAckRequested()), XMLMessageBuilder.getInstance(Manifest.class).handle(message.getManifest())), keyHolder); for (DataSource attachment : message.getAttachments()) { simpleJdbcTemplate.update("insert into ebms_attachment (" + "ebms_message_id," + "name," + "content_type," + "content" + ") values (?,?,?,?)", keyHolder.getKey().longValue(), attachment.getName() == null ? Constants.DEFAULT_FILENAME : attachment.getName(), attachment.getContentType().split(";")[0].trim(), IOUtils.toByteArray(attachment.getInputStream())); } List<Object[]> events = new ArrayList<Object[]>(); for (EbMSSendEvent sendEvent : sendEvents) { //events.add(new Object[]{keyHolder.getKey().longValue(),String.format(getDateFormat(),sendEvent.getTime())}); events.add(new Object[] { keyHolder.getKey().longValue(), sendEvent.getTime() }); } simpleJdbcTemplate.batchUpdate( "insert into ebms_send_event (" + "ebms_message_id," + "time" + ") values (?,?)", events); } catch (Exception e) { throw new RuntimeException(e); } } }); } catch (RuntimeException e) { throw new DAOException(e); } } @Override public void insertMessage(final EbMSMessage message, final EbMSMessageStatus status) throws DAOException { try { transactionTemplate.execute(new TransactionCallbackWithoutResult() { @Override public void doInTransactionWithoutResult(TransactionStatus transactionStatus) { try { Date timestamp = new Date(); KeyHolder keyHolder = new GeneratedKeyHolder(); jdbcTemplate.update(getEbMSMessagePreparedStatement(timestamp, message.getMessageHeader().getCPAId(), message.getMessageHeader().getConversationId(), message.getMessageOrder() == null ? null : message.getMessageOrder().getSequenceNumber().getValue().longValue(), message.getMessageHeader().getMessageData().getMessageId(), message.getMessageHeader().getMessageData().getRefToMessageId(), message.getMessageHeader().getFrom().getRole(), message.getMessageHeader().getTo().getRole(), message.getMessageHeader().getService().getType(), message.getMessageHeader().getService().getValue(), message.getMessageHeader().getAction(), message.getOriginal(), XMLMessageBuilder.getInstance(SignatureType.class) .handle(new ObjectFactory().createSignature(message.getSignature())), XMLMessageBuilder.getInstance(MessageHeader.class) .handle(message.getMessageHeader()), XMLMessageBuilder.getInstance(SyncReply.class).handle(message.getSyncReply()), XMLMessageBuilder.getInstance(MessageOrder.class).handle(message.getMessageOrder()), XMLMessageBuilder.getInstance(AckRequested.class).handle(message.getAckRequested()), XMLMessageBuilder.getInstance(Manifest.class).handle(message.getManifest()), status), keyHolder); for (DataSource attachment : message.getAttachments()) { simpleJdbcTemplate.update("insert into ebms_attachment (" + "ebms_message_id," + "name," + "content_type," + "content" + ") values (?,?,?,?)", keyHolder.getKey().longValue(), attachment.getName() == null ? Constants.DEFAULT_FILENAME : attachment.getName(), attachment.getContentType().split(";")[0].trim(), IOUtils.toByteArray(attachment.getInputStream())); } } catch (Exception e) { throw new RuntimeException(e); } } }); } catch (RuntimeException e) { throw new DAOException(e); } } @Override public void insertMessage(final EbMSMessage message, final EbMSMessageStatus status, final EbMSMessageError messageError, final EbMSSendEvent sendEvent) throws DAOException { try { transactionTemplate.execute(new TransactionCallbackWithoutResult() { @Override public void doInTransactionWithoutResult(TransactionStatus transactionStatus) { try { Date timestamp = new Date(); KeyHolder keyHolder = new GeneratedKeyHolder(); jdbcTemplate.update(getEbMSMessagePreparedStatement(timestamp, message.getMessageHeader().getCPAId(), message.getMessageHeader().getConversationId(), message.getMessageOrder() == null ? null : message.getMessageOrder().getSequenceNumber().getValue().longValue(), message.getMessageHeader().getMessageData().getMessageId(), message.getMessageHeader().getMessageData().getRefToMessageId(), message.getMessageHeader().getFrom().getRole(), message.getMessageHeader().getTo().getRole(), message.getMessageHeader().getService().getType(), message.getMessageHeader().getService().getValue(), message.getMessageHeader().getAction(), message.getOriginal(), XMLMessageBuilder.getInstance(SignatureType.class) .handle(new ObjectFactory().createSignature(message.getSignature())), XMLMessageBuilder.getInstance(MessageHeader.class) .handle(message.getMessageHeader()), XMLMessageBuilder.getInstance(SyncReply.class).handle(message.getSyncReply()), XMLMessageBuilder.getInstance(MessageOrder.class).handle(message.getMessageOrder()), XMLMessageBuilder.getInstance(AckRequested.class).handle(message.getAckRequested()), XMLMessageBuilder.getInstance(Manifest.class).handle(message.getManifest()), status), keyHolder); for (DataSource attachment : message.getAttachments()) { simpleJdbcTemplate.update("insert into ebms_attachment (" + "ebms_message_id," + "name," + "content_type," + "content" + ") values (?,?,?,?)", keyHolder.getKey().longValue(), attachment.getName() == null ? Constants.DEFAULT_FILENAME : attachment.getName(), attachment.getContentType().split(";")[0].trim(), IOUtils.toByteArray(attachment.getInputStream())); } jdbcTemplate.update(getEbMSMessagePreparedStatement(timestamp, messageError.getMessageHeader().getCPAId(), messageError.getMessageHeader().getConversationId(), messageError.getMessageHeader().getMessageData().getMessageId(), messageError.getMessageHeader().getMessageData().getRefToMessageId(), messageError.getMessageHeader().getFrom().getRole(), messageError.getMessageHeader().getTo().getRole(), messageError.getMessageHeader().getService().getType(), messageError.getMessageHeader().getService().getValue(), messageError.getMessageHeader().getAction(), XMLMessageBuilder.getInstance(MessageHeader.class) .handle(messageError.getMessageHeader()), XMLMessageBuilder.getInstance(ErrorList.class).handle(messageError.getErrorList())), keyHolder); simpleJdbcTemplate.update( "insert into ebms_send_event (" + "ebms_message_id," + "time" + ") values (?,?)", keyHolder.getKey().longValue(), //String.format(getDateFormat(),sendEvent.getTime()) sendEvent.getTime()); } catch (Exception e) { throw new RuntimeException(e); } } }); } catch (RuntimeException e) { throw new DAOException(e); } } @Override public void insertMessage(final EbMSMessage message, final EbMSMessageStatus status, final EbMSAcknowledgment acknowledgment, final EbMSSendEvent sendEvent) throws DAOException { try { transactionTemplate.execute(new TransactionCallbackWithoutResult() { @Override public void doInTransactionWithoutResult(TransactionStatus transactionStatus) { try { Date timestamp = new Date(); KeyHolder keyHolder = new GeneratedKeyHolder(); jdbcTemplate.update(getEbMSMessagePreparedStatement(timestamp, message.getMessageHeader().getCPAId(), message.getMessageHeader().getConversationId(), message.getMessageOrder() == null ? null : message.getMessageOrder().getSequenceNumber().getValue().longValue(), message.getMessageHeader().getMessageData().getMessageId(), message.getMessageHeader().getMessageData().getRefToMessageId(), message.getMessageHeader().getFrom().getRole(), message.getMessageHeader().getTo().getRole(), message.getMessageHeader().getService().getType(), message.getMessageHeader().getService().getValue(), message.getMessageHeader().getAction(), message.getOriginal(), XMLMessageBuilder.getInstance(SignatureType.class) .handle(new ObjectFactory().createSignature(message.getSignature())), XMLMessageBuilder.getInstance(MessageHeader.class) .handle(message.getMessageHeader()), XMLMessageBuilder.getInstance(SyncReply.class).handle(message.getSyncReply()), XMLMessageBuilder.getInstance(MessageOrder.class).handle(message.getMessageOrder()), XMLMessageBuilder.getInstance(AckRequested.class).handle(message.getAckRequested()), XMLMessageBuilder.getInstance(Manifest.class).handle(message.getManifest()), status), keyHolder); for (DataSource attachment : message.getAttachments()) { simpleJdbcTemplate.update("insert into ebms_attachment (" + "ebms_message_id," + "name," + "content_type," + "content" + ") values (?,?,?,?)", keyHolder.getKey().longValue(), attachment.getName() == null ? Constants.DEFAULT_FILENAME : attachment.getName(), attachment.getContentType().split(";")[0].trim(), IOUtils.toByteArray(attachment.getInputStream())); } jdbcTemplate.update(getEbMSMessagePreparedStatement(timestamp, acknowledgment.getMessageHeader().getCPAId(), acknowledgment.getMessageHeader().getConversationId(), acknowledgment.getMessageHeader().getMessageData().getMessageId(), acknowledgment.getMessageHeader().getMessageData().getRefToMessageId(), acknowledgment.getMessageHeader().getFrom().getRole(), acknowledgment.getMessageHeader().getTo().getRole(), acknowledgment.getMessageHeader().getService().getType(), acknowledgment.getMessageHeader().getService().getValue(), acknowledgment.getMessageHeader().getAction(), XMLMessageBuilder.getInstance(MessageHeader.class) .handle(acknowledgment.getMessageHeader()), XMLMessageBuilder.getInstance(Acknowledgment.class) .handle(acknowledgment.getAcknowledgment())), keyHolder); simpleJdbcTemplate.update( "insert into ebms_send_event (" + "ebms_message_id," + "time" + ") values (?,?)", keyHolder.getKey().longValue(), //String.format(getDateFormat(),sendTime) sendEvent.getTime()); } catch (Exception e) { throw new RuntimeException(e); } } }); } catch (RuntimeException e) { throw new DAOException(e); } } @Override public void insertMessage(final EbMSMessageError messageError, final EbMSMessageStatus status) throws DAOException { try { transactionTemplate.execute(new TransactionCallbackWithoutResult() { @Override public void doInTransactionWithoutResult(TransactionStatus transactionStatus) { try { Date timestamp = new Date(); KeyHolder keyHolder = new GeneratedKeyHolder(); jdbcTemplate.update(getEbMSMessagePreparedStatement(timestamp, messageError.getMessageHeader().getCPAId(), messageError.getMessageHeader().getConversationId(), messageError.getMessageHeader().getMessageData().getMessageId(), messageError.getMessageHeader().getMessageData().getRefToMessageId(), messageError.getMessageHeader().getFrom().getRole(), messageError.getMessageHeader().getTo().getRole(), messageError.getMessageHeader().getService().getType(), messageError.getMessageHeader().getService().getValue(), messageError.getMessageHeader().getAction(), XMLMessageBuilder.getInstance(MessageHeader.class) .handle(messageError.getMessageHeader()), XMLMessageBuilder.getInstance(ErrorList.class).handle(messageError.getErrorList()), status), keyHolder); Long id = getEbMSMessageId( messageError.getMessageHeader().getMessageData().getRefToMessageId()); if (id != null) simpleJdbcTemplate.update("delete from ebms_send_event" + " where ebms_message_id = ?" + " and status = 0", id); } catch (Exception e) { throw new RuntimeException(e); } } }); } catch (RuntimeException e) { throw new DAOException(e); } } @Override public void insertMessage(final EbMSAcknowledgment acknowledgment, final EbMSMessageStatus status) throws DAOException { try { transactionTemplate.execute(new TransactionCallbackWithoutResult() { @Override public void doInTransactionWithoutResult(TransactionStatus transactionStatus) { try { Date timestamp = new Date(); KeyHolder keyHolder = new GeneratedKeyHolder(); jdbcTemplate.update(getEbMSMessagePreparedStatement(timestamp, acknowledgment.getMessageHeader().getCPAId(), acknowledgment.getMessageHeader().getConversationId(), acknowledgment.getMessageHeader().getMessageData().getMessageId(), acknowledgment.getMessageHeader().getMessageData().getRefToMessageId(), acknowledgment.getMessageHeader().getFrom().getRole(), acknowledgment.getMessageHeader().getTo().getRole(), acknowledgment.getMessageHeader().getService().getType(), acknowledgment.getMessageHeader().getService().getValue(), acknowledgment.getMessageHeader().getAction(), XMLMessageBuilder.getInstance(MessageHeader.class) .handle(acknowledgment.getMessageHeader()), XMLMessageBuilder.getInstance(Acknowledgment.class) .handle(acknowledgment.getAcknowledgment()), status), keyHolder); Long id = getEbMSMessageId( acknowledgment.getMessageHeader().getMessageData().getRefToMessageId()); if (id != null) simpleJdbcTemplate.update("delete from ebms_send_event" + " where ebms_message_id = ?" + " and status = 0", id); } catch (Exception e) { throw new RuntimeException(e); } } }); } catch (RuntimeException e) { throw new DAOException(e); } } public String getMessageContextFilter(EbMSMessageContext messageContext, List<Object> parameters) { StringBuffer result = new StringBuffer(); if (messageContext != null) { if (messageContext.getCpaId() != null) { parameters.add(messageContext.getCpaId()); result.append(" and cpa_id = ?"); } if (messageContext.getFromRole() != null) { parameters.add(messageContext.getFromRole()); result.append(" and from_role = ?"); } if (messageContext.getToRole() != null) { parameters.add(messageContext.getToRole()); result.append(" and to_role = ?"); } if (messageContext.getServiceType() != null) { parameters.add(messageContext.getServiceType()); result.append(" and service_type = ?"); } if (messageContext.getService() != null) { parameters.add(messageContext.getService()); result.append(" and service = ?"); } if (messageContext.getAction() != null) { parameters.add(messageContext.getAction()); result.append(" and action = ?"); } if (messageContext.getConversationId() != null) { parameters.add(messageContext.getConversationId()); result.append(" and conversation_id = ?"); } if (messageContext.getMessageId() != null) { parameters.add(messageContext.getMessageId()); result.append(" and message_id = ?"); } if (messageContext.getRefToMessageId() != null) { parameters.add(messageContext.getRefToMessageId()); result.append(" and ref_to_message_id = ?"); } if (messageContext.getSequenceNr() != null) { parameters.add(messageContext.getSequenceNr()); result.append(" and sequence_nr = ?"); } } return result.toString(); } @SuppressWarnings("unchecked") @Override public List<String> getReceivedMessageIds(EbMSMessageContext messageContext) throws DAOException { try { List<Object> parameters = new ArrayList<Object>(); return jdbcTemplate.queryForList("select message_id" + " from ebms_message" + " where status = " + EbMSMessageStatus.RECEIVED.id() + getMessageContextFilter(messageContext, parameters) + " order by time_stamp asc", parameters.toArray(new Object[0]), String.class); } catch (DataAccessException e) { throw new DAOException(e); } } public abstract String getReceivedMessageIdsQuery(String messageContextFilter, int maxNr); @SuppressWarnings("unchecked") @Override public List<String> getReceivedMessageIds(EbMSMessageContext messageContext, int maxNr) throws DAOException { try { //TODO improve: add maxNr to parameters??? List<Object> parameters = new ArrayList<Object>(); String messageContextFilter = getMessageContextFilter(messageContext, parameters); return jdbcTemplate.queryForList(getReceivedMessageIdsQuery(messageContextFilter, maxNr), parameters.toArray(new Object[0]), String.class); } catch (DataAccessException e) { throw new DAOException(e); } } @Override public EbMSBaseMessage getMessage(String messageId) throws DAOException { try { return simpleJdbcTemplate.queryForObject( "select id, service, action, message_header, ack_requested, content" + " from ebms_message" + " where message_id = ?", // and status=" + EbMSMessageStatus.RECEIVED.id(), new EbMSBaseMessageParameterizedRowMapper(), messageId); } catch (EmptyResultDataAccessException e) { return null; } catch (DataAccessException e) { throw new DAOException(e); } } @Override public void processReceivedMessage(String messageId) throws DAOException { try { simpleJdbcTemplate.update("update ebms_message" + " set status = " + EbMSMessageStatus.PROCESSED.id() + "," + " status_time = " + getTimestampFunction() + " where message_id = ?" + " and status = " + EbMSMessageStatus.RECEIVED.id(), messageId); } catch (DataAccessException e) { throw new DAOException(e); } } @Override public void processReceivedMessages(final List<String> messageIds) throws DAOException { try { List<Object[]> ids = new ArrayList<Object[]>(); for (String messageId : messageIds) ids.add(new Object[] { messageId }); simpleJdbcTemplate.batchUpdate("update ebms_message" + " set status = " + EbMSMessageStatus.PROCESSED.id() + "," + " status_time = " + getTimestampFunction() + " where message_id = ?" + " and status = " + EbMSMessageStatus.RECEIVED.id(), ids); } catch (DataAccessException e) { throw new DAOException(e); } } }