unload hotfix (#132)

HadoukenIO · Aug 3, 2017 · e6ab4db · e6ab4db
1 parent 1f943ef
commit e6ab4db
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/browser/subscription_manager.js b/src/browser/subscription_manager.js
@@ -59,9 +59,14 @@ function SubscriptionManager() {
     }
 
     function removeSubscription(identity, ...args) {
-        let key = genSubscriptionKey.apply(null, args),
-            identityKey = identityToKey(identity),
-            subscription = subscriptionList.get(identityKey).get(key);
+        const key = genSubscriptionKey.apply(null, args);
+        const identityKey = identityToKey(identity);
+        const subscriptionId = subscriptionList.get(identityKey);
+        const subscription = subscriptionId && subscriptionId.get(key);
+
+        if (!subscription) {
+            return;
+        }
 
         subscription.refCount--;
         if (subscription.refCount <= 0) {