Back RKEntityByAttributeCache with an NSMutableSet instead of an …

…`NSMutableDictionary` for performance. refs #1232
RestKit · Apr 24, 2013 · f728d95 · f728d95
1 parent cfb4f7b
commit f728d95
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Code/CoreData/RKEntityByAttributeCache.m b/Code/CoreData/RKEntityByAttributeCache.m
@@ -249,9 +249,9 @@ - (NSSet *)objectsWithAttributeValues:(NSDictionary *)attributeValues inContext:
     NSMutableSet *objects = [NSMutableSet set];
     NSArray *cacheKeys = RKCacheKeysForEntityFromAttributeValues(self.entity, attributeValues);
     for (NSString *cacheKey in cacheKeys) {
-        NSArray *objectIDs = nil;
+        NSSet *objectIDs = nil;
         @synchronized(self.cacheKeysToObjectIDs) {
-            objectIDs = [[NSArray alloc] initWithArray:[self.cacheKeysToObjectIDs objectForKey:cacheKey] copyItems:YES];
+            objectIDs = [[NSSet alloc] initWithSet:[self.cacheKeysToObjectIDs objectForKey:cacheKey] copyItems:YES];
         }
         if ([objectIDs count]) {
             /**
@@ -279,13 +279,13 @@ - (void)setObjectID:(NSManagedObjectID *)objectID forAttributeValues:(NSDictiona
     [self.lock lock];
     if (attributeValues && [attributeValues count]) {
         NSString *cacheKey = RKCacheKeyForEntityWithAttributeValues(self.entity, attributeValues);
-        NSMutableArray *objectIDs = [self.cacheKeysToObjectIDs objectForKey:cacheKey];
+        NSMutableSet *objectIDs = [self.cacheKeysToObjectIDs objectForKey:cacheKey];
         if (objectIDs) {
             if (! [objectIDs containsObject:objectID]) {
                 [objectIDs addObject:objectID];
             }
         } else {
-            objectIDs = [NSMutableArray arrayWithObject:objectID];
+            objectIDs = [NSMutableSet setWithObject:objectID];
         }
 
         if (nil == self.cacheKeysToObjectIDs) self.cacheKeysToObjectIDs = [NSMutableDictionary dictionary];
@@ -302,7 +302,7 @@ - (void)removeObjectID:(NSManagedObjectID *)objectID forAttributeValues:(NSDicti
     if (attributeValues && [attributeValues count]) {
         NSArray *cacheKeys = RKCacheKeysForEntityFromAttributeValues(self.entity, attributeValues);
         for (NSString *cacheKey in cacheKeys) {
-            NSMutableArray *objectIDs = [self.cacheKeysToObjectIDs objectForKey:cacheKey];
+            NSMutableSet *objectIDs = [self.cacheKeysToObjectIDs objectForKey:cacheKey];
             if (objectIDs && [objectIDs containsObject:objectID]) {
                 [objectIDs removeObject:objectID];
             }