Changeset 104 for branches/supervisor/src/main/java/omq/client/proxy/Proxymq.java

Timestamp:

10/16/13 13:01:04 (11 years ago)

Author:

stoda

Message:

MultiMethod? without numWait -> it depends on the timeout

File:

• branches/supervisor/src/main/java/omq/client/proxy/Proxymq.java (modified) (8 diffs)

branches/supervisor/src/main/java/omq/client/proxy/Proxymq.java

@@ -4 +4 @@
 import java.lang.reflect.InvocationHandler;
 import java.lang.reflect.Method;
+import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 import java.util.Properties;
@@ -201 +203 @@
                                 publishMessage(request, replyQueueName);
                                 if (request.isMulti()) {
-                                        return getResults(corrId, request.getWait(), timeout, type);
+                                        return getResults(corrId, timeout, type);
                                 } else {
                                         return getResult(corrId, timeout, type);
@@ -228 +230 @@
                 String methodName = method.getName();
                 boolean multi = false;
-                int wait = 0;
 
                 if (method.getAnnotation(MultiMethod.class) != null) {
                         multi = true;
-                        wait = method.getAnnotation(MultiMethod.class).waitNum();
                 }
 
@@ -246 +246 @@
                                 timeout = sync.timeout();
                         }
-                        return Request.newSyncRequest(corrId, methodName, arguments, retries, timeout, multi, wait);
+                        return Request.newSyncRequest(corrId, methodName, arguments, retries, timeout, multi);
                 } else {
                         return Request.newAsyncRequest(corrId, methodName, arguments, multi);
@@ -291 +291 @@
          * @param corrId
          *            - Correlation Id of the request
-         * @param wait
-         *            - Array length
          * @param timeout
          *            - Timeout read in @SyncMethod.timeout(). If the timeout is set
@@ -302 +300 @@
          * @throws Exception
          */
-        private Object getResults(String corrId, int wait, long timeout, Class<?> type) throws Exception {
+        private Object getResults(String corrId, long timeout, Class<?> type) throws Exception {
                 Response resp = null;
                 // Get the component type of an array
                 Class<?> actualType = type.getComponentType();
 
-                Object array = Array.newInstance(actualType, wait);
+                List<Object> list = new ArrayList<Object>();
 
                 int i = 0;
@@ -313 +311 @@
                 long start = System.currentTimeMillis();
 
-                while (i < wait) {
+                while (true) {
                         synchronized (results) {
                                 // Due to we are using notifyAll(), we need to control the real
@@ -322 +320 @@
                                 }
                                 if ((timeout - localTimeout) <= 0) {
-                                        throw new TimeoutException("Timeout exception time: " + timeout);
+                                        break;
                                 }
                                 // Remove the corrId to receive new replies
                                 resp = serializer.deserializeResponse(results.remove(corrId), actualType);
-                                Array.set(array, i, resp.getResult());
+                                list.add(resp.getResult());
                         }
                         i++;
                 }
+
+                if (i == 0) {
+                        results.remove(corrId);
+                        throw new TimeoutException("Timeout exception time: " + timeout);
+                }
+
                 synchronized (results) {
                         results.put(corrId, null);
+                }
+
+                Object array = Array.newInstance(actualType, i);
+                i = 0;
+                for (Object o : list) {
+                        Array.set(array, i++, o);
                 }