aboutsummaryrefslogtreecommitdiffhomepage
path: root/Firestore/Source/Local/FSTEagerGarbageCollector.mm
diff options
context:
space:
mode:
authorGravatar Gil <mcg@google.com>2018-01-31 11:23:55 -0800
committerGravatar GitHub <noreply@github.com>2018-01-31 11:23:55 -0800
commit729b8d176c75ecc0cbbd137cc6811116a64e310a (patch)
tree22b793b03611ce5ad615b7c7d9579f5ba5206b4a /Firestore/Source/Local/FSTEagerGarbageCollector.mm
parent693d0649bfcc9c32201e2431ae08ea85fdbdb617 (diff)
Move all Firestore Objective-C to Objective-C++ (#734)
* Move all Firestore files to Objective-C++ * Update project file references * Don't use module imports from Objective-C++ * Use extern "C" for C-accessible globals * Work around more stringent type checking in Objective-C++ * NSMutableDictionary ivars aren't implicitly casted to NSDictionary * FSTMaybeDocument callback can't be passed a function that accepts FSTDocument * NSComparisonResult can't be multiplied by -1 without casting * Add a #include <inttypes.h> where needed * Avoid using C++ keywords as variables * Remove #if __cplusplus guards
Diffstat (limited to 'Firestore/Source/Local/FSTEagerGarbageCollector.mm')
-rw-r--r--Firestore/Source/Local/FSTEagerGarbageCollector.mm89
1 files changed, 89 insertions, 0 deletions
diff --git a/Firestore/Source/Local/FSTEagerGarbageCollector.mm b/Firestore/Source/Local/FSTEagerGarbageCollector.mm
new file mode 100644
index 0000000..77a577e
--- /dev/null
+++ b/Firestore/Source/Local/FSTEagerGarbageCollector.mm
@@ -0,0 +1,89 @@
+/*
+ * Copyright 2017 Google
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#import "Firestore/Source/Local/FSTEagerGarbageCollector.h"
+
+#import "Firestore/Source/Model/FSTDocumentKey.h"
+
+NS_ASSUME_NONNULL_BEGIN
+
+#pragma mark - FSTMultiReferenceSet
+
+@interface FSTEagerGarbageCollector ()
+
+/** The garbage collectible sources to double-check during garbage collection. */
+@property(nonatomic, strong, readonly) NSMutableArray<id<FSTGarbageSource>> *sources;
+
+/** A set of potentially garbage keys. */
+@property(nonatomic, strong, readonly) NSMutableSet<FSTDocumentKey *> *potentialGarbage;
+
+@end
+
+@implementation FSTEagerGarbageCollector
+
+- (instancetype)init {
+ self = [super init];
+ if (self) {
+ _sources = [NSMutableArray array];
+ _potentialGarbage = [[NSMutableSet alloc] init];
+ }
+ return self;
+}
+
+- (BOOL)isEager {
+ return YES;
+}
+
+- (void)addGarbageSource:(id<FSTGarbageSource>)garbageSource {
+ [self.sources addObject:garbageSource];
+ garbageSource.garbageCollector = self;
+}
+
+- (void)removeGarbageSource:(id<FSTGarbageSource>)garbageSource {
+ [self.sources removeObject:garbageSource];
+ garbageSource.garbageCollector = nil;
+}
+
+- (void)addPotentialGarbageKey:(FSTDocumentKey *)key {
+ [self.potentialGarbage addObject:key];
+}
+
+- (NSMutableSet<FSTDocumentKey *> *)collectGarbage {
+ NSMutableArray<id<FSTGarbageSource>> *sources = self.sources;
+
+ NSMutableSet<FSTDocumentKey *> *actualGarbage = [NSMutableSet set];
+ for (FSTDocumentKey *key in self.potentialGarbage) {
+ BOOL isGarbage = YES;
+ for (id<FSTGarbageSource> source in sources) {
+ if ([source containsKey:key]) {
+ isGarbage = NO;
+ break;
+ }
+ }
+
+ if (isGarbage) {
+ [actualGarbage addObject:key];
+ }
+ }
+
+ // Clear locally retained potential keys and returned confirmed garbage.
+ [self.potentialGarbage removeAllObjects];
+ return actualGarbage;
+}
+
+@end
+
+NS_ASSUME_NONNULL_END