Changeset 58 for trunk/src/main/java/omq/client/proxy/Proxymq.java

Timestamp:

06/25/13 16:38:31 (11 years ago)

Author:

stoda

Message:

@MultiMethod? + @SyncMethod? implemented and tested

File:

• trunk/src/main/java/omq/client/proxy/Proxymq.java (modified) (9 diffs)

trunk/src/main/java/omq/client/proxy/Proxymq.java

@@ -2 +2 @@
 
 import java.io.IOException;
+import java.lang.reflect.Array;
 import java.lang.reflect.InvocationHandler;
 import java.lang.reflect.Method;
 import java.lang.reflect.Proxy;
-import java.util.ArrayList;
 import java.util.Collection;
 import java.util.HashMap;
 import java.util.Hashtable;
-import java.util.List;
 import java.util.Map;
 import java.util.Properties;
-
-import org.apache.log4j.Logger;
 
 import omq.Remote;
@@ -34 +31 @@
 import omq.exception.TimeoutException;
 
+import org.apache.log4j.Logger;
+
 import com.rabbitmq.client.AMQP.BasicProperties;
 
@@ -190 +189 @@
                         try {
                                 publishMessage(request, replyQueueName);
-                                return getResult(corrId, timeout, type);
+                                if (request.isMulti()) {
+                                        return getResults(corrId, 2, timeout, type);
+                                } else {
+                                        return getResult(corrId, timeout, type);
+                                }
+
                         } catch (TimeoutException te) {
                                 logger.error(te);
@@ -203 +207 @@
                 String methodName = method.getName();
                 boolean multi = false;
+                int wait = 0;
 
                 if (method.getAnnotation(MultiMethod.class) != null) {
                         multi = true;
+                        wait = method.getAnnotation(MultiMethod.class).waitNum();
                 }
 
@@ -219 +225 @@
                                 timeout = sync.timeout();
                         }
-                        return Request.newSyncRequest(corrId, methodName, arguments, retries, timeout, multi);
+                        return Request.newSyncRequest(corrId, methodName, arguments, retries, timeout, multi, wait);
                 } else {
                         return Request.newAsyncRequest(corrId, methodName, arguments, multi);
@@ -229 +235 @@
 
                 // Wait for the results.
-                long localTimeout = 0;
+                long localTimeout = timeout;
                 long start = System.currentTimeMillis();
                 synchronized (results) {
                         // Due to we are using notifyAll(), we need to control the real time
                         while (!results.containsKey(corrId) && (timeout - localTimeout) >= 0) {
-                                results.wait(timeout);
+                                results.wait(localTimeout);
                                 localTimeout = System.currentTimeMillis() - start;
                         }
@@ -258 +264 @@
         }
 
-        @SuppressWarnings("unused")
         private Object getResults(String corrId, int wait, long timeout, Class<?> type) throws Exception {
                 Response resp = null;
-                List<Object> list = new ArrayList<Object>();
+                Class<?> actualType = type.getComponentType();
+
+                Object array = Array.newInstance(actualType, wait);
 
                 int i = 0;
@@ -279 +286 @@
                                 }
                                 // Remove the corrId to receive new replies
-                                resp = serializer.deserializeResponse(results.remove(corrId), type);
-                                list.add(resp.getResult());
-
+                                resp = serializer.deserializeResponse(results.remove(corrId), actualType);
+                                System.out.println("/n/n/n/n/nResult type: "+resp.getResult()+" /n/n/n/n/n");
+                                Array.set(array, i, resp.getResult());
                         }
                         i++;
@@ -289 +296 @@
                 }
 
-                return list;
+                return array;
         }