blob: 0b1413f5dca29bb705955e60ff76f866c186c693 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
/*
* 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 "FIRDocumentChange.h"
#import "Firestore/Source/API/FIRDocumentSnapshot+Internal.h"
#import "Firestore/Source/Core/FSTQuery.h"
#import "Firestore/Source/Core/FSTViewSnapshot.h"
#import "Firestore/Source/Model/FSTDocument.h"
#import "Firestore/Source/Model/FSTDocumentSet.h"
#include "Firestore/core/src/firebase/firestore/util/hard_assert.h"
NS_ASSUME_NONNULL_BEGIN
@interface FIRDocumentChange ()
- (instancetype)initWithType:(FIRDocumentChangeType)type
document:(FIRDocumentSnapshot *)document
oldIndex:(NSUInteger)oldIndex
newIndex:(NSUInteger)newIndex NS_DESIGNATED_INITIALIZER;
@end
@implementation FIRDocumentChange (Internal)
+ (FIRDocumentChangeType)documentChangeTypeForChange:(FSTDocumentViewChange *)change {
if (change.type == FSTDocumentViewChangeTypeAdded) {
return FIRDocumentChangeTypeAdded;
} else if (change.type == FSTDocumentViewChangeTypeModified ||
change.type == FSTDocumentViewChangeTypeMetadata) {
return FIRDocumentChangeTypeModified;
} else if (change.type == FSTDocumentViewChangeTypeRemoved) {
return FIRDocumentChangeTypeRemoved;
} else {
HARD_FAIL("Unknown FSTDocumentViewChange: %s", change.type);
}
}
+ (NSArray<FIRDocumentChange *> *)documentChangesForSnapshot:(FSTViewSnapshot *)snapshot
includeMetadataChanges:(BOOL)includeMetadataChanges
firestore:(FIRFirestore *)firestore {
if (snapshot.oldDocuments.isEmpty) {
// Special case the first snapshot because index calculation is easy and fast. Also all changes
// on the first snapshot are adds so there are also no metadata-only changes to filter out.
FSTDocument *_Nullable lastDocument = nil;
NSUInteger index = 0;
NSMutableArray<FIRDocumentChange *> *changes = [NSMutableArray array];
for (FSTDocumentViewChange *change in snapshot.documentChanges) {
FIRQueryDocumentSnapshot *document =
[FIRQueryDocumentSnapshot snapshotWithFirestore:firestore
documentKey:change.document.key
document:change.document
fromCache:snapshot.isFromCache];
HARD_ASSERT(change.type == FSTDocumentViewChangeTypeAdded,
"Invalid event type for first snapshot");
HARD_ASSERT(!lastDocument || snapshot.query.comparator(lastDocument, change.document) ==
NSOrderedAscending,
"Got added events in wrong order");
[changes addObject:[[FIRDocumentChange alloc] initWithType:FIRDocumentChangeTypeAdded
document:document
oldIndex:NSNotFound
newIndex:index++]];
}
return changes;
} else {
// A DocumentSet that is updated incrementally as changes are applied to use to lookup the index
// of a document.
FSTDocumentSet *indexTracker = snapshot.oldDocuments;
NSMutableArray<FIRDocumentChange *> *changes = [NSMutableArray array];
for (FSTDocumentViewChange *change in snapshot.documentChanges) {
if (!includeMetadataChanges && change.type == FSTDocumentViewChangeTypeMetadata) {
continue;
}
FIRQueryDocumentSnapshot *document =
[FIRQueryDocumentSnapshot snapshotWithFirestore:firestore
documentKey:change.document.key
document:change.document
fromCache:snapshot.isFromCache];
NSUInteger oldIndex = NSNotFound;
NSUInteger newIndex = NSNotFound;
if (change.type != FSTDocumentViewChangeTypeAdded) {
oldIndex = [indexTracker indexOfKey:change.document.key];
HARD_ASSERT(oldIndex != NSNotFound, "Index for document not found");
indexTracker = [indexTracker documentSetByRemovingKey:change.document.key];
}
if (change.type != FSTDocumentViewChangeTypeRemoved) {
indexTracker = [indexTracker documentSetByAddingDocument:change.document];
newIndex = [indexTracker indexOfKey:change.document.key];
}
[FIRDocumentChange documentChangeTypeForChange:change];
FIRDocumentChangeType type = [FIRDocumentChange documentChangeTypeForChange:change];
[changes addObject:[[FIRDocumentChange alloc] initWithType:type
document:document
oldIndex:oldIndex
newIndex:newIndex]];
}
return changes;
}
}
@end
@implementation FIRDocumentChange
- (instancetype)initWithType:(FIRDocumentChangeType)type
document:(FIRQueryDocumentSnapshot *)document
oldIndex:(NSUInteger)oldIndex
newIndex:(NSUInteger)newIndex {
if (self = [super init]) {
_type = type;
_document = document;
_oldIndex = oldIndex;
_newIndex = newIndex;
}
return self;
}
- (BOOL)isEqual:(nullable id)other {
if (other == self) return YES;
if (![other isKindOfClass:[FIRDocumentChange class]]) return NO;
FIRDocumentChange *change = (FIRDocumentChange *)other;
return self.type == change.type && [self.document isEqual:change.document] &&
self.oldIndex == change.oldIndex && self.newIndex == change.newIndex;
}
- (NSUInteger)hash {
NSUInteger result = (NSUInteger)self.type;
result = result * 31u + [self.document hash];
result = result * 31u + (NSUInteger)self.oldIndex;
result = result * 31u + (NSUInteger)self.newIndex;
return result;
}
@end
NS_ASSUME_NONNULL_END
|
