258 lines
10 KiB
Objective-C
258 lines
10 KiB
Objective-C
/*
|
|
* 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 "FSyncPoint.h"
|
|
#import "FOperation.h"
|
|
#import "FWriteTreeRef.h"
|
|
#import "FNode.h"
|
|
#import "FEventRegistration.h"
|
|
#import "FIRDatabaseQuery.h"
|
|
#import "FChildrenNode.h"
|
|
#import "FTupleRemovedQueriesEvents.h"
|
|
#import "FView.h"
|
|
#import "FOperationSource.h"
|
|
#import "FQuerySpec.h"
|
|
#import "FQueryParams.h"
|
|
#import "FPath.h"
|
|
#import "FEmptyNode.h"
|
|
#import "FViewCache.h"
|
|
#import "FCacheNode.h"
|
|
#import "FPersistenceManager.h"
|
|
#import "FDataEvent.h"
|
|
|
|
/**
|
|
* SyncPoint represents a single location in a SyncTree with 1 or more event registrations, meaning we need to
|
|
* maintain 1 or more Views at this location to cache server data and raise appropriate events for server changes
|
|
* and user writes (set, transaction, update).
|
|
*
|
|
* It's responsible for:
|
|
* - Maintaining the set of 1 or more views necessary at this location (a SyncPoint with 0 views should be removed).
|
|
* - Proxying user / server operations to the views as appropriate (i.e. applyServerOverwrite,
|
|
* applyUserOverwrite, etc.)
|
|
*/
|
|
@interface FSyncPoint ()
|
|
/**
|
|
* The Views being tracked at this location in the tree, stored as a map where the key is a
|
|
* queryParams and the value is the View for that query.
|
|
*
|
|
* NOTE: This list will be quite small (usually 1, but perhaps 2 or 3; any more is an odd use case).
|
|
*
|
|
* Maps NSString -> FView
|
|
*/
|
|
@property (nonatomic, strong) NSMutableDictionary *views;
|
|
|
|
@property (nonatomic, strong) FPersistenceManager *persistenceManager;
|
|
@end
|
|
|
|
@implementation FSyncPoint
|
|
|
|
- (id) initWithPersistenceManager:(FPersistenceManager *)persistence {
|
|
self = [super init];
|
|
if (self) {
|
|
self.persistenceManager = persistence;
|
|
self.views = [[NSMutableDictionary alloc] init];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (BOOL) isEmpty {
|
|
return [self.views count] == 0;
|
|
}
|
|
|
|
- (NSArray *) applyOperation:(id<FOperation>)operation
|
|
toView:(FView *)view
|
|
writesCache:(FWriteTreeRef *)writesCache
|
|
serverCache:(id<FNode>)optCompleteServerCache {
|
|
FViewOperationResult *result = [view applyOperation:operation writesCache:writesCache serverCache:optCompleteServerCache];
|
|
if (!view.query.loadsAllData) {
|
|
NSMutableSet *removed = [NSMutableSet set];
|
|
NSMutableSet *added = [NSMutableSet set];
|
|
[result.changes enumerateObjectsUsingBlock:^(FChange *change, NSUInteger idx, BOOL *stop) {
|
|
if (change.type == FIRDataEventTypeChildAdded) {
|
|
[added addObject:change.childKey];
|
|
} else if (change.type == FIRDataEventTypeChildRemoved) {
|
|
[removed addObject:change.childKey];
|
|
}
|
|
}];
|
|
if ([removed count] > 0 || [added count] > 0) {
|
|
[self.persistenceManager updateTrackedQueryKeysWithAddedKeys:added removedKeys:removed forQuery:view.query];
|
|
}
|
|
}
|
|
return result.events;
|
|
}
|
|
|
|
- (NSArray *) applyOperation:(id <FOperation>)operation writesCache:(FWriteTreeRef *)writesCache serverCache:(id <FNode>)optCompleteServerCache {
|
|
FQueryParams *queryParams = operation.source.queryParams;
|
|
if (queryParams != nil) {
|
|
FView *view = [self.views objectForKey:queryParams];
|
|
NSAssert(view != nil, @"SyncTree gave us an op for an invalid query.");
|
|
return [self applyOperation:operation toView:view writesCache:writesCache serverCache:optCompleteServerCache];
|
|
} else {
|
|
NSMutableArray *events = [[NSMutableArray alloc] init];
|
|
[self.views enumerateKeysAndObjectsUsingBlock:^(FQueryParams *key, FView *view, BOOL *stop) {
|
|
NSArray *eventsForView = [self applyOperation:operation toView:view writesCache:writesCache serverCache:optCompleteServerCache];
|
|
[events addObjectsFromArray:eventsForView];
|
|
}];
|
|
return events;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Add an event callback for the specified query
|
|
* Returns Array of FEvent events to raise.
|
|
*/
|
|
- (NSArray *) addEventRegistration:(id <FEventRegistration>)eventRegistration
|
|
forNonExistingViewForQuery:(FQuerySpec *)query
|
|
writesCache:(FWriteTreeRef *)writesCache
|
|
serverCache:(FCacheNode *)serverCache {
|
|
NSAssert(self.views[query.params] == nil, @"Found view for query: %@", query.params);
|
|
// TODO: make writesCache take flag for complete server node
|
|
id<FNode> eventCache = [writesCache calculateCompleteEventCacheWithCompleteServerCache:serverCache.isFullyInitialized ? serverCache.node : nil];
|
|
BOOL eventCacheComplete;
|
|
if (eventCache != nil) {
|
|
eventCacheComplete = YES;
|
|
} else {
|
|
eventCache = [writesCache calculateCompleteEventChildrenWithCompleteServerChildren:serverCache.node];
|
|
eventCacheComplete = NO;
|
|
}
|
|
|
|
FIndexedNode *indexed = [FIndexedNode indexedNodeWithNode:eventCache index:query.index];
|
|
FCacheNode *eventCacheNode = [[FCacheNode alloc] initWithIndexedNode:indexed
|
|
isFullyInitialized:eventCacheComplete
|
|
isFiltered:NO];
|
|
FViewCache *viewCache = [[FViewCache alloc] initWithEventCache:eventCacheNode serverCache:serverCache];
|
|
FView *view = [[FView alloc] initWithQuery:query initialViewCache:viewCache];
|
|
// If this is a non-default query we need to tell persistence our current view of the data
|
|
if (!query.loadsAllData) {
|
|
NSMutableSet *allKeys = [NSMutableSet set];
|
|
[view.eventCache enumerateChildrenUsingBlock:^(NSString *key, id<FNode> node, BOOL *stop) {
|
|
[allKeys addObject:key];
|
|
}];
|
|
[self.persistenceManager setTrackedQueryKeys:allKeys forQuery:query];
|
|
}
|
|
self.views[query.params] = view;
|
|
return [self addEventRegistration:eventRegistration forExistingViewForQuery:query];
|
|
}
|
|
|
|
- (NSArray *)addEventRegistration:(id<FEventRegistration>)eventRegistration
|
|
forExistingViewForQuery:(FQuerySpec *)query {
|
|
FView *view = self.views[query.params];
|
|
NSAssert(view != nil, @"No view for query: %@", query);
|
|
[view addEventRegistration:eventRegistration];
|
|
return [view initialEvents:eventRegistration];
|
|
}
|
|
|
|
/**
|
|
* Remove event callback(s). Return cancelEvents if a cancelError is specified.
|
|
*
|
|
* If query is the default query, we'll check all views for the specified eventRegistration.
|
|
* If eventRegistration is nil, we'll remove all callbacks for the specified view(s).
|
|
*
|
|
* @return FTupleRemovedQueriesEvents removed queries and any cancel events
|
|
*/
|
|
- (FTupleRemovedQueriesEvents *) removeEventRegistration:(id <FEventRegistration>)eventRegistration
|
|
forQuery:(FQuerySpec *)query
|
|
cancelError:(NSError *)cancelError {
|
|
NSMutableArray *removedQueries = [[NSMutableArray alloc] init];
|
|
__block NSMutableArray *cancelEvents = [[NSMutableArray alloc] init];
|
|
BOOL hadCompleteView = [self hasCompleteView];
|
|
if ([query isDefault]) {
|
|
// When you do [ref removeObserverWithHandle:], we search all views for the registration to remove.
|
|
[self.views enumerateKeysAndObjectsUsingBlock:^(FQueryParams *viewQueryParams, FView *view, BOOL *stop) {
|
|
[cancelEvents addObjectsFromArray:[view removeEventRegistration:eventRegistration cancelError:cancelError]];
|
|
if ([view isEmpty]) {
|
|
[self.views removeObjectForKey:viewQueryParams];
|
|
|
|
// We'll deal with complete views later
|
|
if (![view.query loadsAllData]) {
|
|
[removedQueries addObject:view.query];
|
|
}
|
|
}
|
|
}];
|
|
} else {
|
|
// remove the callback from the specific view
|
|
FView *view = [self.views objectForKey:query.params];
|
|
if (view != nil) {
|
|
[cancelEvents addObjectsFromArray:[view removeEventRegistration:eventRegistration cancelError:cancelError]];
|
|
|
|
if ([view isEmpty]) {
|
|
[self.views removeObjectForKey:query.params];
|
|
|
|
// We'll deal with complete views later
|
|
if (![view.query loadsAllData]) {
|
|
[removedQueries addObject:view.query];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (hadCompleteView && ![self hasCompleteView]) {
|
|
// We removed our last complete view
|
|
[removedQueries addObject:[FQuerySpec defaultQueryAtPath:query.path]];
|
|
}
|
|
|
|
return [[FTupleRemovedQueriesEvents alloc] initWithRemovedQueries:removedQueries cancelEvents:cancelEvents];
|
|
}
|
|
|
|
- (NSArray *) queryViews {
|
|
__block NSMutableArray *filteredViews = [[NSMutableArray alloc] init];
|
|
|
|
[self.views enumerateKeysAndObjectsUsingBlock:^(FQueryParams *key, FView *view, BOOL *stop) {
|
|
if (![view.query loadsAllData]) {
|
|
[filteredViews addObject:view];
|
|
}
|
|
}];
|
|
|
|
return filteredViews;
|
|
}
|
|
|
|
- (id <FNode>) completeServerCacheAtPath:(FPath *)path {
|
|
__block id<FNode> serverCache = nil;
|
|
[self.views enumerateKeysAndObjectsUsingBlock:^(FQueryParams *key, FView *view, BOOL *stop) {
|
|
serverCache = [view completeServerCacheFor:path];
|
|
*stop = (serverCache != nil);
|
|
}];
|
|
return serverCache;
|
|
}
|
|
|
|
- (FView *) viewForQuery:(FQuerySpec *)query {
|
|
return [self.views objectForKey:query.params];
|
|
}
|
|
|
|
- (BOOL) viewExistsForQuery:(FQuerySpec *)query {
|
|
return [self viewForQuery:query] != nil;
|
|
}
|
|
|
|
- (BOOL) hasCompleteView {
|
|
return [self completeView] != nil;
|
|
}
|
|
|
|
- (FView *) completeView {
|
|
__block FView *completeView = nil;
|
|
|
|
[self.views enumerateKeysAndObjectsUsingBlock:^(FQueryParams *key, FView *view, BOOL *stop) {
|
|
if ([view.query loadsAllData]) {
|
|
completeView = view;
|
|
*stop = YES;
|
|
}
|
|
}];
|
|
|
|
return completeView;
|
|
}
|
|
|
|
|
|
@end
|