Add multiple response support to binding.websocket.

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

1 files changed, 24 insertions, 4 deletions

diff --git a/sca-java-2.x/contrib/modules/binding-websocket/src/main/java/org/apache/tuscany/sca/binding/websocket/runtime/WebsocketServiceInvoker.java b/sca-java-2.x/contrib/modules/binding-websocket/src/main/java/org/apache/tuscany/sca/binding/websocket/runtime/WebsocketServiceInvoker.java
index 7615be916c..ca859e38f4 100644
--- a/sca-java-2.x/contrib/modules/binding-websocket/src/main/java/org/apache/tuscany/sca/binding/websocket/runtime/WebsocketServiceInvoker.java
+++ b/sca-java-2.x/contrib/modules/binding-websocket/src/main/java/org/apache/tuscany/sca/binding/websocket/runtime/WebsocketServiceInvoker.java
@@ -40,7 +40,13 @@ package org.apache.tuscany.sca.binding.websocket.runtime;
 

 import java.lang.reflect.InvocationTargetException;

 

+import org.apache.tuscany.sca.assembly.EndpointReference;

+import org.apache.tuscany.sca.core.assembly.impl.RuntimeEndpointImpl;

+import org.apache.tuscany.sca.core.assembly.impl.RuntimeEndpointReferenceImpl;

+import org.apache.tuscany.sca.core.invocation.Constants;

+import org.apache.tuscany.sca.core.invocation.impl.MessageImpl;

 import org.apache.tuscany.sca.interfacedef.Operation;

+import org.apache.tuscany.sca.invocation.Message;

 import org.apache.tuscany.sca.runtime.RuntimeEndpoint;

 

 public class WebsocketServiceInvoker {

@@ -53,21 +59,35 @@ public class WebsocketServiceInvoker {
         this.endpoint = endpoint;

     }

 

-    public WebSocketBindingMessage invokeSync(WebSocketBindingMessage request) {

+    public WebsocketBindingMessage invokeSync(WebsocketBindingMessage request) {

         String jsonParams = request.getPayload();

         Object[] args = JSONUtil.decodePayloadForOperation(jsonParams, operation);

         try {

             Object operationResponse = endpoint.invoke(operation, args);

             String payload = JSONUtil.encodePayload(operationResponse);

-            WebSocketBindingMessage response = new WebSocketBindingMessage(request.getOperation(), payload);

+            WebsocketBindingMessage response = new WebsocketBindingMessage(request.getOperation(), payload);

             return response;

         } catch (InvocationTargetException e) {

             throw new RuntimeException(e);

         }

     }

 

-    public void invokeAsync(WebSocketBindingMessage request, TuscanyWebSocket channel) {

-        // TODO add multiple response support

+    public void invokeAsync(WebsocketBindingMessage request, TuscanyWebsocket channel) {

+        String jsonParams = request.getPayload();

+        Object[] args = JSONUtil.decodePayloadForOperation(jsonParams, operation);

+        Message msg = new MessageImpl();

+        msg.getHeaders().put(Constants.MESSAGE_ID, channel.getId());

+        msg.setBody(args);

+        EndpointReference re = new RuntimeEndpointReferenceImpl();

+        RuntimeEndpointImpl callbackEndpoint = new RuntimeEndpointImpl();

+        callbackEndpoint.setURI(request.getOperation());

+        re.setCallbackEndpoint(callbackEndpoint);

+        msg.setFrom(re);

+        endpoint.invoke(operation, msg);

+    }

+

+    public boolean isNonBlocking() {

+        return operation.isNonBlocking();

     }

 

 }