From 8f96cc1b7e185109a903d70f1c115454ef6695d5 Mon Sep 17 00:00:00 2001
From: slaws <slaws@13f79535-47bb-0310-9956-ffa450edef68>
Date: Thu, 23 Jun 2011 14:31:42 +0000
Subject: TUSCANY-3881 - create a context bean to hold a more consistent of
 Axis2 context in the reference and service side chains.

git-svn-id: http://svn.us.apache.org/repos/asf/tuscany@1138899 13f79535-47bb-0310-9956-ffa450edef68
---
 .../binding/ws/axis2/transport/TransportReferenceInterceptor.java | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'sca-java-2.x/trunk/modules/binding-ws-runtime-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/transport')

diff --git a/sca-java-2.x/trunk/modules/binding-ws-runtime-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/transport/TransportReferenceInterceptor.java b/sca-java-2.x/trunk/modules/binding-ws-runtime-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/transport/TransportReferenceInterceptor.java
index 7daa58bf29..888a9f6a79 100644
--- a/sca-java-2.x/trunk/modules/binding-ws-runtime-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/transport/TransportReferenceInterceptor.java
+++ b/sca-java-2.x/trunk/modules/binding-ws-runtime-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/transport/TransportReferenceInterceptor.java
@@ -28,6 +28,7 @@ import org.apache.axis2.client.OperationClient;
 import org.apache.axis2.client.Options;
 import org.apache.axis2.context.MessageContext;
 import org.apache.axis2.transport.http.HTTPConstants;
+import org.apache.tuscany.sca.binding.ws.axis2.context.WSAxis2BindingContext;
 import org.apache.tuscany.sca.interfacedef.util.FaultException;
 import org.apache.tuscany.sca.invocation.Interceptor;
 import org.apache.tuscany.sca.invocation.Invoker;
@@ -72,7 +73,8 @@ public class TransportReferenceInterceptor implements Interceptor {
     }
 
     protected Object invokeTarget(Message msg) throws AxisFault {
-        final OperationClient operationClient = msg.getBindingContext();
+        WSAxis2BindingContext bindingContext = msg.getBindingContext();
+        final OperationClient operationClient = bindingContext.getAxisOperationClient();
 
         // ensure connections are tracked so that they can be closed by the reference binding
         MessageContext requestMC = operationClient.getMessageContext("Out");
@@ -95,6 +97,7 @@ public class TransportReferenceInterceptor implements Interceptor {
         }
 
         MessageContext responseMC = operationClient.getMessageContext("In");
+        bindingContext.setAxisInMessageContext(responseMC);
                 
         OMElement response = responseMC.getEnvelope().getBody().getFirstElement();
 
@@ -111,7 +114,8 @@ public class TransportReferenceInterceptor implements Interceptor {
     }
     
     protected Object invokeTargetOneWay(Message msg) throws AxisFault {
-        OperationClient operationClient = msg.getBindingContext();
+        WSAxis2BindingContext bindingContext = msg.getBindingContext();
+        OperationClient operationClient = bindingContext.getAxisOperationClient();
 
         // ensure connections are tracked so that they can be closed by the reference binding
         MessageContext requestMC = operationClient.getMessageContext("Out");
-- 
cgit v1.2.3