fixing subscription issue

wso2 · Feb 18, 2024 · d943cc9 · d943cc9
1 parent a05a87d
commit d943cc9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/common-controller/internal/utils/event_utils.go b/common-controller/internal/utils/event_utils.go
@@ -194,6 +194,7 @@ func sendApplicationKeyMappingEvent(applicationUUID string, applicationSpec cpv1
 					ApplicationIdentifier: env.AppID,
 					KeyType:               env.KeyType,
 					EnvID:                 env.EnvID,
+					Organization:          applicationSpec.Organization,
 				},
 			}
 			sendEvent(&event)

diff --git a/....enforcer/src/main/java/org/wso2/apk/enforcer/subscription/SubscriptionDataStoreImpl.java b/....enforcer/src/main/java/org/wso2/apk/enforcer/subscription/SubscriptionDataStoreImpl.java
@@ -308,10 +308,11 @@ public void addApplicationMapping(org.wso2.apk.enforcer.discovery.subscription.A
         resolvedApplicationMapping.setUuid(applicationMapping.getUuid());
         resolvedApplicationMapping.setApplicationUUID(applicationMapping.getApplicationRef());
         resolvedApplicationMapping.setSubscriptionUUID(applicationMapping.getSubscriptionRef());
-        if (applicationMappingMap.containsKey(resolvedApplicationMapping.getUuid())) {
-            applicationMappingMap.replace(resolvedApplicationMapping.getUuid(), resolvedApplicationMapping);
+        resolvedApplicationMapping.setOrganization(applicationMapping.getOrganization());
+        if (applicationMappingMap.containsKey(resolvedApplicationMapping.getCacheKey())) {
+            applicationMappingMap.replace(resolvedApplicationMapping.getCacheKey(), resolvedApplicationMapping);
         } else {
-            applicationMappingMap.put(resolvedApplicationMapping.getUuid(), resolvedApplicationMapping);
+            applicationMappingMap.put(resolvedApplicationMapping.getCacheKey(), resolvedApplicationMapping);
         }
     }
 
@@ -342,7 +343,8 @@ public void removeApplicationMapping(org.wso2.apk.enforcer.discovery.subscriptio
         resolvedApplicationMapping.setUuid(applicationMapping.getUuid());
         resolvedApplicationMapping.setApplicationUUID(applicationMapping.getApplicationRef());
         resolvedApplicationMapping.setSubscriptionUUID(applicationMapping.getSubscriptionRef());
-        applicationMappingMap.remove(resolvedApplicationMapping.getUuid());
+        resolvedApplicationMapping.setOrganization(applicationMapping.getOrganization());
+        applicationMappingMap.remove(resolvedApplicationMapping.getCacheKey());
     }
 
     @Override