/* * JBoss, Home of Professional Open Source * Copyright 2006, Red Hat Middleware LLC, and individual contributors * as indicated by the @author tags. * See the copyright.txt in the distribution for a full listing * of individual contributors. * This copyrighted material is made available to anyone wishing to use, * modify, copy, or redistribute it subject to the terms and conditions * of the GNU Lesser General Public License, v. 2.1. * This program is distributed in the hope that it will be useful, but WITHOUT A * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A * PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. * You should have received a copy of the GNU Lesser General Public License, * v.2.1 along with this distribution; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, * MA 02110-1301, USA. * * (C) 2005-2006, * @author JBoss Inc. */ package com.arjuna.wst.tests.arq; import java.util.HashMap; import java.util.Map; import com.arjuna.webservices11.wsarj.ArjunaContext; import com.arjuna.webservices11.wsat.processors.CompletionCoordinatorProcessor; import org.jboss.ws.api.addressing.MAP; import com.arjuna.wst11.CompletionCoordinatorParticipant; import org.oasis_open.docs.ws_tx.wsat._2006._06.Notification; public class TestCompletionCoordinatorProcessor extends CompletionCoordinatorProcessor { private Map messageIdMap = new HashMap() ; public CompletionCoordinatorDetails getCompletionCoordinatorDetails(final String messageId, final long timeout) { final long endTime = System.currentTimeMillis() + timeout ; synchronized(messageIdMap) { long now = System.currentTimeMillis() ; while(now < endTime) { final CompletionCoordinatorDetails details = (CompletionCoordinatorDetails)messageIdMap.remove(messageId) ; if (details != null) { return details ; } try { messageIdMap.wait(endTime - now) ; } catch (final InterruptedException ie) {} // ignore now = System.currentTimeMillis() ; } final CompletionCoordinatorDetails details = (CompletionCoordinatorDetails)messageIdMap.remove(messageId) ; if (details != null) { return details ; } } throw new NullPointerException("Timeout occurred waiting for id: " + messageId) ; } /** * Commit. * @param commit The commit notification. * @param map The addressing context. * @param arjunaContext The arjuna context. */ public void commit(final Notification commit, final MAP map, final ArjunaContext arjunaContext) { final String messageId = map.getMessageID(); final CompletionCoordinatorDetails details = new CompletionCoordinatorDetails(map, arjunaContext) ; details.setCommit(true) ; synchronized(messageIdMap) { messageIdMap.put(messageId, details) ; messageIdMap.notifyAll() ; } } /** * Rollback. * @param rollback The rollback notification. * @param map The addressing context. * @param arjunaContext The arjuna context. */ public void rollback(final Notification rollback, final MAP map, final ArjunaContext arjunaContext) { final String messageId = map.getMessageID() ; final CompletionCoordinatorDetails details = new CompletionCoordinatorDetails(map, arjunaContext) ; details.setRollback(true) ; synchronized(messageIdMap) { messageIdMap.put(messageId, details) ; messageIdMap.notifyAll() ; } } /** * Activate the participant. * @param participant The participant. * @param identifier The identifier. */ public void activateParticipant(final CompletionCoordinatorParticipant participant, final String identifier) { } /** * Deactivate the participant. * @param participant The participant. */ public void deactivateParticipant(final CompletionCoordinatorParticipant participant) { } public static class CompletionCoordinatorDetails { private final MAP map ; private final ArjunaContext arjunaContext ; private boolean commit ; private boolean rollback ; CompletionCoordinatorDetails(final MAP map, final ArjunaContext arjunaContext) { this.map = map ; this.arjunaContext = arjunaContext ; } public MAP getMAP() { return map ; } public ArjunaContext getArjunaContext() { return arjunaContext ; } public boolean hasCommit() { return commit ; } void setCommit(final boolean commit) { this.commit = commit ; } public boolean hasRollback() { return rollback ; } void setRollback(final boolean rollback) { this.rollback = rollback ; } } }