Fix concurrency issue in aggregation

wso2 · Dec 6, 2024 · a85c622 · a85c622
1 parent b45b4ca
commit a85c622
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 12 deletions.
diff --git a/modules/core/src/main/java/org/apache/synapse/mediators/eip/aggregator/Aggregate.java b/modules/core/src/main/java/org/apache/synapse/mediators/eip/aggregator/Aggregate.java
@@ -115,9 +115,6 @@ public Aggregate(SynapseEnvironment synEnv, String corelation, long timeoutMilli
      */
     public synchronized boolean addMessage(MessageContext synCtx) {
         if (maxCount <= 0 || (maxCount > 0 && messages.size() < maxCount)) {
-            if (messages == null) {
-                return false;
-            }
             messages.add(synCtx);
             return true;
         } else {
@@ -261,12 +258,16 @@ public void run() {
                 break;
             }
             if (getLock()) {
-                if (log.isDebugEnabled()) {
-                    log.debug("Time : " + System.currentTimeMillis() + " and this aggregator " +
-                            "expired at : " + expiryTimeMillis);
+                try {
+                    if (log.isDebugEnabled()) {
+                        log.debug("Time : " + System.currentTimeMillis() + " and this aggregator " +
+                                "expired at : " + expiryTimeMillis);
+                    }
+                    synEnv.getExecutorService().execute(new AggregateTimeout(this));
+                    break;
+                } finally {
+                    releaseLock();
                 }
-                synEnv.getExecutorService().execute(new AggregateTimeout(this));
-                break;
             }
         }
     }
@@ -312,10 +313,14 @@ public void run() {
     }
 
     public synchronized boolean getLock() {
-        return !locked;
+        if (!locked) {
+            locked = true;
+            return true;
+        }
+        return false;
     }
 
-    public void releaseLock() {
+    public synchronized void releaseLock() {
         locked = false;
     }
 

diff --git a/modules/core/src/main/java/org/apache/synapse/mediators/v2/ScatterGather.java b/modules/core/src/main/java/org/apache/synapse/mediators/v2/ScatterGather.java
@@ -448,7 +448,7 @@ private boolean aggregateMessages(MessageContext synCtx, SynapseLog synLog) {
             }
         } else {
             synLog.traceOrDebug("Unable to find aggregation correlation property");
-            return true;
+            return false;
         }
         // if there is an aggregate continue on aggregation
         if (aggregate != null) {
@@ -471,7 +471,6 @@ private boolean aggregateMessages(MessageContext synCtx, SynapseLog synLog) {
             }
         } else {
             synLog.traceOrDebug("Unable to find an aggregate for this message - skip");
-            return true;
         }
         return false;
     }