Retain class observers

zintus · Nov 13, 2015 · 52d374c · 52d374c
1 parent 2182390
commit 52d374c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Dynamic Code Injection/dyci/Classes/Notifications/SFInjectionsNotificationsCenter.m b/Dynamic Code Injection/dyci/Classes/Notifications/SFInjectionsNotificationsCenter.m
@@ -68,13 +68,13 @@ - (void)addObserver:(id<SFInjectionObserver>)observer forClass:(Class)aClass {
         return;
     }
     @synchronized (_observers) {
-        NSMutableSet * observersPerClass = [_observers objectForKey:aClass];
+        CFMutableSetRef observersPerClass = (__bridge CFMutableSetRef)([_observers objectForKey:aClass]);
         if (!observersPerClass) {
-            observersPerClass = (__bridge_transfer NSMutableSet *) CFSetCreateMutable(nil, 0, nil);
-            [_observers setObject:observersPerClass forKey:aClass];
+            observersPerClass = CFSetCreateMutable(nil, 0, nil);
+            [_observers setObject:(__bridge_transfer NSMutableSet *) observersPerClass forKey:aClass];
         }
-        @synchronized (observersPerClass) {
-            [observersPerClass addObject:observer];
+        @synchronized ((__bridge id)observersPerClass) {
+            CFSetSetValue(observersPerClass, (__bridge_retained void*) observer);
         }
     }
 }