/* | |
* Copyright 2004 The Apache Software Foundation. | |
* | |
* 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 org.apache.kandula.initiator; | |
import java.io.IOException; | |
import java.rmi.RemoteException; | |
import org.apache.axis2.AxisFault; | |
import org.apache.axis2.addressing.EndpointReference; | |
import org.apache.axis2.context.ConfigurationContext; | |
import org.apache.axis2.context.ConfigurationContextFactory; | |
import org.apache.axis2.databinding.types.URI; | |
import org.apache.axis2.databinding.types.URI.MalformedURIException; | |
import org.apache.axis2.deployment.DeploymentException; | |
import org.apache.kandula.Constants; | |
import org.apache.kandula.Status; | |
import org.apache.kandula.context.CoordinationContext; | |
import org.apache.kandula.context.impl.ADBCoordinationContext; | |
import org.apache.kandula.faults.AbstractKandulaException; | |
import org.apache.kandula.faults.KandulaGeneralException; | |
import org.apache.kandula.utility.EndpointReferenceFactory; | |
import org.apache.kandula.wsat.completion.CompletionCoordinatorPortTypeRawXMLStub; | |
import org.apache.kandula.wsat.completion.CompletionInitiatorServiceListener; | |
import org.apache.kandula.wscoor.ActivationServiceStub; | |
import org.apache.kandula.wscoor.RegistrationServiceStub; | |
import org.oasis_open.docs.ws_tx.wscoor._2006._06.CreateCoordinationContext; | |
import org.oasis_open.docs.ws_tx.wscoor._2006._06.CreateCoordinationContextResponse; | |
import org.oasis_open.docs.ws_tx.wscoor._2006._06.CreateCoordinationContextResponseType; | |
import org.oasis_open.docs.ws_tx.wscoor._2006._06.CreateCoordinationContextType; | |
import org.oasis_open.docs.ws_tx.wscoor._2006._06.Register; | |
import org.oasis_open.docs.ws_tx.wscoor._2006._06.RegisterResponse; | |
import org.oasis_open.docs.ws_tx.wscoor._2006._06.RegisterType; | |
public class TransactionManager { | |
private static ThreadLocal threadInfo = new ThreadLocal(); | |
private ConfigurationContext configurationContext; | |
public TransactionManager(String axis2Home, String axis2Xml) | |
throws AbstractKandulaException { | |
try { | |
configurationContext = ConfigurationContextFactory | |
.createConfigurationContextFromFileSystem(axis2Home, | |
axis2Xml); | |
} catch (DeploymentException e) { | |
throw new KandulaGeneralException(e); | |
} catch (AxisFault e1) { | |
throw new KandulaGeneralException(e1); | |
} | |
} | |
/** | |
* @throws Exception | |
*/ | |
public void begin(String coordinatorAddress) throws Exception { | |
if (threadInfo.get()!=null) | |
{ | |
throw new IllegalStateException(); | |
} | |
InitiatorContext initiatorTransaction = new InitiatorContext(Constants.WS_AT,coordinatorAddress); | |
CoordinationContext coordinationContext = createTransaction(initiatorTransaction); | |
initiatorTransaction.setCoordinationContext(coordinationContext); | |
threadInfo.set(initiatorTransaction); | |
} | |
public void commit() throws Exception { | |
InitiatorContext initiatorTransaction = getTransaction(); | |
CompletionCallback completionCallback = new CompletionCallback(initiatorTransaction); | |
// Register for completion | |
EndpointReference coordinationEPR = registerForCompletion(initiatorTransaction,completionCallback); | |
initiatorTransaction.setCoordinationEPR(coordinationEPR); | |
CompletionCoordinatorPortTypeRawXMLStub stub = new CompletionCoordinatorPortTypeRawXMLStub( | |
configurationContext, coordinationEPR); | |
stub.commitOperation(); | |
while (!completionCallback.isComplete()) | |
Thread.sleep(10); | |
if ((completionCallback.getResult() == Status.CoordinatorStatus.STATUS_ABORTING)) { | |
forgetTransaction(); | |
throw new Exception("Aborted"); | |
} | |
forgetTransaction(); | |
} | |
public void rollback() throws Exception { | |
InitiatorContext initiatorTransaction = getTransaction(); | |
// Register for completion | |
CompletionCallback completionCallback = new CompletionCallback(initiatorTransaction); | |
// Register for completion | |
EndpointReference coordinationEPR = registerForCompletion(initiatorTransaction,completionCallback); | |
initiatorTransaction.setCoordinationEPR(coordinationEPR); | |
CompletionCoordinatorPortTypeRawXMLStub stub = new CompletionCoordinatorPortTypeRawXMLStub( | |
configurationContext, coordinationEPR); | |
stub.rollbackOperation(); | |
while (!completionCallback.isComplete()) | |
Thread.sleep(10); | |
forgetTransaction(); | |
} | |
// public Transaction suspend() { | |
// Transaction tx= getTransaction(); | |
// forget(); | |
// return tx; | |
// } | |
// | |
// public void resume(Transaction tx) { | |
// if (threadInfo.get() != null) | |
// throw new IllegalStateException(); | |
// else | |
// threadInfo.set(tx); | |
// } | |
// | |
// public void forget() { | |
// threadInfo.set(null); | |
// } | |
private static InitiatorContext getTransaction() | |
throws AbstractKandulaException { | |
Object key = threadInfo.get(); | |
if (key!= null) | |
{ | |
return (InitiatorContext)key; | |
} | |
return null; | |
} | |
public static void forgetTransaction() { | |
threadInfo.set(null); | |
} | |
private EndpointReference registerForCompletion(InitiatorContext initiatorTransaction, CompletionCallback completionCallback) throws AxisFault, IOException, MalformedURIException, RemoteException { | |
RegistrationServiceStub registrationCoordinator = new RegistrationServiceStub( | |
configurationContext,null); | |
registrationCoordinator._getServiceClient().setTargetEPR(initiatorTransaction | |
.getCoordinationContext().getRegistrationService()); | |
//setup the listener | |
CompletionInitiatorServiceListener listener = new CompletionInitiatorServiceListener(); | |
EndpointReference registrationRequeterPortEPR =listener.getEpr(completionCallback); | |
Register register = new Register(); | |
RegisterType registerType = new RegisterType(); | |
registerType.setProtocolIdentifier(new URI(Constants.WS_AT_COMPLETION)); | |
registerType.setParticipantProtocolService(EndpointReferenceFactory.getADBEPRTypeFromEPR(registrationRequeterPortEPR)); | |
register.setRegister(registerType); | |
//Actual WS call for registeration | |
RegisterResponse registerResponse = registrationCoordinator | |
.RegisterOperation(register); | |
EndpointReference coordinationEPR = EndpointReferenceFactory | |
.getEPR(registerResponse.getRegisterResponse() | |
.getCoordinatorProtocolService()); | |
return coordinationEPR; | |
} | |
private CoordinationContext createTransaction(InitiatorContext initiatorTransaction) throws AxisFault, MalformedURIException, RemoteException { | |
ActivationServiceStub activationCoordinator = new ActivationServiceStub( | |
configurationContext,initiatorTransaction.getActivationEPR()); | |
CreateCoordinationContext context = new CreateCoordinationContext(); | |
CreateCoordinationContextType createCoordinationContextType = new CreateCoordinationContextType(); | |
createCoordinationContextType.setCoordinationType(new URI(initiatorTransaction.getCoordinationType())); | |
context.setCreateCoordinationContext(createCoordinationContextType); | |
CreateCoordinationContextResponse response = activationCoordinator.CreateCoordinationContextOperation(context); | |
CreateCoordinationContextResponseType createCoordinationContextResponse = response.getCreateCoordinationContextResponse(); | |
CoordinationContext coordinationContext = new ADBCoordinationContext(createCoordinationContextResponse.getCoordinationContext()); | |
return coordinationContext; | |
} | |
} |