Tweaks and extensions to better match the async service implementation in core + JMS binding

git-svn-id: http://svn.us.apache.org/repos/asf/tuscany@1051397 13f79535-47bb-0310-9956-ffa450edef68

1 files changed, 5 insertions, 1 deletions

diff --git a/sca-java-2.x/trunk/samples/extending-tuscany/implementation-sample/src/main/java/sample/impl/SampleAsyncResponseInvoker.java b/sca-java-2.x/trunk/samples/extending-tuscany/implementation-sample/src/main/java/sample/impl/SampleAsyncResponseInvoker.java
index 2366deeb33..aed21ecc4e 100644
--- a/sca-java-2.x/trunk/samples/extending-tuscany/implementation-sample/src/main/java/sample/impl/SampleAsyncResponseInvoker.java
+++ b/sca-java-2.x/trunk/samples/extending-tuscany/implementation-sample/src/main/java/sample/impl/SampleAsyncResponseInvoker.java
@@ -49,7 +49,11 @@ class SampleAsyncResponseInvoker implements InvokerAsyncResponse {
 
     public void invokeAsyncResponse(final Message msg) {
         try {
-            String messageID = (String) msg.getHeaders().get(Constants.MESSAGE_ID);
+        	String messageID = (String) msg.getHeaders().get("RELATES_TO");
+            if (messageID == null ) {
+            	messageID = (String) msg.getHeaders().get(Constants.MESSAGE_ID);
+			} // end if
+            
             String forwardOpName = (String)asyncMessageMap.get(messageID);
             
             // process the async response