1 files changed, 203 insertions, 0 deletions
diff --git a/sca-java-2.x/tags/2.0.1-RC1/modules/binding-rest-runtime/src/main/java/org/apache/tuscany/sca/binding/rest/operationselector/rpc/provider/RPCOperationSelectorInterceptor.java b/sca-java-2.x/tags/2.0.1-RC1/modules/binding-rest-runtime/src/main/java/org/apache/tuscany/sca/binding/rest/operationselector/rpc/provider/RPCOperationSelectorInterceptor.java
new file mode 100644
index 0000000000..53be473416
--- /dev/null
+++ b/sca-java-2.x/tags/2.0.1-RC1/modules/binding-rest-runtime/src/main/java/org/apache/tuscany/sca/binding/rest/operationselector/rpc/provider/RPCOperationSelectorInterceptor.java
@@ -0,0 +1,203 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you 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.tuscany.sca.binding.rest.operationselector.rpc.provider;
+
+import java.lang.annotation.Annotation;
+import java.lang.reflect.Array;
+import java.lang.reflect.Method;
+import java.net.URLDecoder;
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+
+import javax.servlet.http.HttpServletResponse;
+import javax.ws.rs.QueryParam;
+
+import org.apache.tuscany.sca.common.http.HTTPContext;
+import org.apache.tuscany.sca.common.http.HTTPUtils;
+import org.apache.tuscany.sca.core.ExtensionPointRegistry;
+import org.apache.tuscany.sca.core.UtilityExtensionPoint;
+import org.apache.tuscany.sca.databinding.SimpleTypeMapper;
+import org.apache.tuscany.sca.interfacedef.InterfaceContract;
+import org.apache.tuscany.sca.interfacedef.Operation;
+import org.apache.tuscany.sca.interfacedef.java.JavaOperation;
+import org.apache.tuscany.sca.interfacedef.util.TypeInfo;
+import org.apache.tuscany.sca.invocation.Interceptor;
+import org.apache.tuscany.sca.invocation.Invoker;
+import org.apache.tuscany.sca.invocation.Message;
+import org.apache.tuscany.sca.runtime.RuntimeComponentService;
+import org.apache.tuscany.sca.runtime.RuntimeEndpoint;
+
+/**
+ * RPC operation selector Interceptor.
+ *
+ * @version $Rev$ $Date$
+*/
+public class RPCOperationSelectorInterceptor implements Interceptor {
+    private ExtensionPointRegistry extensionPoints;
+    private SimpleTypeMapper simpleTypeMapper;
+
+    private RuntimeEndpoint endpoint;
+
+    private RuntimeComponentService service;
+    private InterfaceContract interfaceContract;
+    private List<Operation> serviceOperations;
+
+    private Invoker next;
+
+    public RPCOperationSelectorInterceptor(ExtensionPointRegistry extensionPoints, RuntimeEndpoint endpoint) {
+        this.extensionPoints = extensionPoints;
+
+        UtilityExtensionPoint utilityExtensionPoint = extensionPoints.getExtensionPoint(UtilityExtensionPoint.class);
+        this.simpleTypeMapper = utilityExtensionPoint.getUtility(SimpleTypeMapper.class);
+
+        this.endpoint = endpoint;
+
+        this.service = (RuntimeComponentService)endpoint.getService();
+        this.interfaceContract = service.getInterfaceContract();
+        this.serviceOperations = service.getInterfaceContract().getInterface().getOperations();
+    }
+
+    public Invoker getNext() {
+        return next;
+    }
+
+    public void setNext(Invoker next) {
+        this.next = next;
+    }
+
+    public Message invoke(Message msg) {
+        try {
+            HTTPContext bindingContext = (HTTPContext)msg.getBindingContext();
+
+            if(! "get".equalsIgnoreCase(bindingContext.getHttpRequest().getMethod())) {
+                throw new RuntimeException("RPC Invocation only allowed over HTTP GET operations");
+            }
+
+            String path = URLDecoder.decode(HTTPUtils.getRequestPath(bindingContext.getHttpRequest()), "UTF-8");
+
+            if (path.startsWith("/")) {
+                path = path.substring(1);
+            }
+
+
+            String operationName = bindingContext.getHttpRequest().getParameter("method");
+            Operation operation = findOperation( operationName );
+
+            if (operation == null) {
+                throw new RuntimeException("Invalid Operation '" + operationName + "'" );
+            }
+
+            final JavaOperation javaOperation = (JavaOperation)operation;
+            final Method method = javaOperation.getJavaMethod();
+
+            List<Object> messageParameters = new ArrayList<Object>();
+            for(int i=0; i<method.getParameterTypes().length; i++) {
+                for(Annotation annotation : method.getParameterAnnotations()[i]) {
+                    if (annotation instanceof QueryParam) {
+                        QueryParam queryParam = (QueryParam) annotation;
+                        String name = queryParam.value();
+                        String[] values = bindingContext.getHttpRequest().getParameterValues(name);
+
+                        if(values.length == 1) {
+                            //process value, making necessary map from string to expected value
+                            Class<?> clazz = method.getParameterTypes()[i];
+                            TypeInfo typeInfo = simpleTypeMapper.getXMLType(clazz);
+                            Object v = simpleTypeMapper.toJavaObject(typeInfo.getQName(), values[0], null);
+                            messageParameters.add(v);
+                        } else {
+                            //process value, making necessary map from string to expected value
+                            Class<?> clazz = (method.getParameterTypes()[i]).getComponentType();
+                            TypeInfo typeInfo = simpleTypeMapper.getXMLType(clazz);
+
+
+                            Object objectArray = Array.newInstance(clazz, values.length);
+                            for (int count = 0; count < values.length; ++count) {
+                                Object v = simpleTypeMapper.toJavaObject(typeInfo.getQName(), values[count], null);
+                                Array.set(objectArray, count, v);
+                            }
+
+                            messageParameters.add(objectArray);
+                        }
+                    }
+                }
+            }
+
+            Object[] body = new Object[messageParameters.size()];
+            messageParameters.toArray(body);
+
+            msg.setBody(body);
+            msg.setOperation(operation);
+
+            Message responseMessage = getNext().invoke(msg);
+
+            //set Cache-Control to no-cache to avoid intermediary
+            //proxy/reverse-proxy caches and always hit the server
+            //that would identify if the value was current or not
+            bindingContext.getHttpResponse().setHeader("Cache-Control", "no-cache");
+            bindingContext.getHttpResponse().setHeader("Expires", new Date(0).toGMTString());
+
+
+            String eTag = HTTPUtils.calculateHashETag(responseMessage.getBody().toString().getBytes("UTF-8"));
+
+            // Test request for predicates.
+            String predicate = bindingContext.getHttpRequest().getHeader( "If-Match" );
+            if (( predicate != null ) && ( !predicate.equals(eTag) )) {
+                // No match, should short circuit
+                bindingContext.getHttpResponse().sendError(HttpServletResponse.SC_PRECONDITION_FAILED);
+            }
+            predicate = bindingContext.getHttpRequest().getHeader( "If-None-Match" );
+            if (( predicate != null ) && ( predicate.equals(eTag) )) {
+                // Match, should short circuit
+                bindingContext.getHttpResponse().sendError(HttpServletResponse.SC_NOT_MODIFIED);
+            }
+
+            bindingContext.getHttpResponse().addHeader("ETag", eTag);
+
+            return responseMessage;
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    /**
+     * Find the operation from the component service contract
+     * @param componentService
+     * @param method
+     * @return
+     */
+    private Operation findOperation(String method) {
+        if (method.contains(".")) {
+            method = method.substring(method.lastIndexOf(".") + 1);
+        }
+
+        List<Operation> operations = endpoint.getComponentServiceInterfaceContract().getInterface().getOperations();
+
+        Operation result = null;
+        for (Operation o : operations) {
+            if (o.getName().equalsIgnoreCase(method)) {
+                result = o;
+                break;
+            }
+        }
+
+        return result;
+    }
+}