1414 * limitations under the License.
1515 */
1616
17- #import " FIRDocumentChange.h"
17+ #import " Firestore/Source/API/ FIRDocumentChange+Internal .h"
1818
1919#import " Firestore/Source/API/FIRDocumentSnapshot+Internal.h"
20- #import " Firestore/Source/API/FIRFirestore+Internal.h"
21- #import " Firestore/Source/Core/FSTQuery.h"
22- #import " Firestore/Source/Model/FSTDocument.h"
2320
24- #include " Firestore/core/src/firebase/firestore/core/view_snapshot.h"
25- #include " Firestore/core/src/firebase/firestore/model/document_set.h"
21+ #include " Firestore/core/src/firebase/firestore/api/document_change.h"
2622#include " Firestore/core/src/firebase/firestore/util/hard_assert.h"
2723
28- using firebase::firestore::api::Firestore;
29- using firebase::firestore::core::DocumentViewChange;
30- using firebase::firestore::core::ViewSnapshot;
31- using firebase::firestore::model::DocumentSet;
24+ using firebase::firestore::api::DocumentChange;
3225
3326NS_ASSUME_NONNULL_BEGIN
3427
35- @interface FIRDocumentChange ()
36-
37- - (instancetype )initWithType : (FIRDocumentChangeType)type
38- document : (FIRDocumentSnapshot *)document
39- oldIndex : (NSUInteger )oldIndex
40- newIndex : (NSUInteger )newIndex NS_DESIGNATED_INITIALIZER;
41-
42- @end
43-
44- @implementation FIRDocumentChange (Internal)
45-
46- + (FIRDocumentChangeType)documentChangeTypeForChange : (const DocumentViewChange &)change {
47- switch (change.type ()) {
48- case DocumentViewChange::Type::kAdded :
49- return FIRDocumentChangeTypeAdded;
50- case DocumentViewChange::Type::kModified :
51- case DocumentViewChange::Type::kMetadata :
52- return FIRDocumentChangeTypeModified;
53- case DocumentViewChange::Type::kRemoved :
54- return FIRDocumentChangeTypeRemoved;
55- }
56-
57- HARD_FAIL (" Unknown DocumentViewChange::Type: %s" type ());
28+ @implementation FIRDocumentChange {
29+ DocumentChange _documentChange;
5830}
5931
60- + (NSArray <FIRDocumentChange *> *)documentChangesForSnapshot : (const ViewSnapshot &)snapshot
61- includeMetadataChanges : (bool )includeMetadataChanges
62- firestore : (Firestore *)firestore {
63- if (snapshot.old_documents ().empty ()) {
64- // Special case the first snapshot because index calculation is easy and fast. Also all changes
65- // on the first snapshot are adds so there are also no metadata-only changes to filter out.
66- FSTDocument *_Nullable lastDocument = nil ;
67- NSUInteger index = 0 ;
68- NSMutableArray <FIRDocumentChange *> *changes = [NSMutableArray array ];
69- for (const DocumentViewChange &change : snapshot.document_changes ()) {
70- FIRQueryDocumentSnapshot *document = [[FIRQueryDocumentSnapshot alloc ]
71- initWithFirestore: firestore
72- documentKey: change.document ().key
73- document: change.document ()
74- fromCache: snapshot.from_cache ()
75- hasPendingWrites: snapshot.mutated_keys ().contains (change.document ().key)];
76- HARD_ASSERT (change.type () == DocumentViewChange::Type::kAdded ,
77- " Invalid event type for first snapshot"
78- HARD_ASSERT (!lastDocument || snapshot.query ().comparator (lastDocument, change.document ()) ==
79- NSOrderedAscending,
80- " Got added events in wrong order"
81- [changes addObject: [[FIRDocumentChange alloc ] initWithType: FIRDocumentChangeTypeAdded
82- document: document
83- oldIndex: NSNotFound
84- newIndex: index++]];
85- }
86- return changes;
87- } else {
88- // A DocumentSet that is updated incrementally as changes are applied to use to lookup the index
89- // of a document.
90- DocumentSet indexTracker = snapshot.old_documents ();
91- NSMutableArray <FIRDocumentChange *> *changes = [NSMutableArray array ];
92- for (const DocumentViewChange &change : snapshot.document_changes ()) {
93- if (!includeMetadataChanges && change.type () == DocumentViewChange::Type::kMetadata ) {
94- continue ;
95- }
96-
97- FIRQueryDocumentSnapshot *document = [[FIRQueryDocumentSnapshot alloc ]
98- initWithFirestore: firestore
99- documentKey: change.document ().key
100- document: change.document ()
101- fromCache: snapshot.from_cache ()
102- hasPendingWrites: snapshot.mutated_keys ().contains (change.document ().key)];
103-
104- size_t oldIndex = DocumentSet::npos;
105- size_t newIndex = DocumentSet::npos;
106- if (change.type () != DocumentViewChange::Type::kAdded ) {
107- oldIndex = indexTracker.IndexOf (change.document ().key );
108- HARD_ASSERT (oldIndex != DocumentSet::npos, " Index for document not found"
109- indexTracker = indexTracker.erase (change.document ().key );
110- }
111- if (change.type () != DocumentViewChange::Type::kRemoved ) {
112- indexTracker = indexTracker.insert (change.document ());
113- newIndex = indexTracker.IndexOf (change.document ().key );
114- }
115- [FIRDocumentChange documentChangeTypeForChange: change];
116- FIRDocumentChangeType type = [FIRDocumentChange documentChangeTypeForChange: change];
117- [changes addObject: [[FIRDocumentChange alloc ] initWithType: type
118- document: document
119- oldIndex: oldIndex
120- newIndex: newIndex]];
121- }
122- return changes;
123- }
124- }
125-
126- @end
127-
128- @implementation FIRDocumentChange
129-
130- - (instancetype )initWithType : (FIRDocumentChangeType)type
131- document : (FIRQueryDocumentSnapshot *)document
132- oldIndex : (NSUInteger )oldIndex
133- newIndex : (NSUInteger )newIndex {
32+ - (instancetype )initWithDocumentChange : (DocumentChange &&)documentChange {
13433 if (self = [super init ]) {
135- _type = type;
136- _document = document;
137- _oldIndex = oldIndex;
138- _newIndex = newIndex;
34+ _documentChange = std::move (documentChange);
13935 }
14036 return self;
14137}
@@ -145,16 +41,36 @@ - (BOOL)isEqual:(nullable id)other {
14541 if (![other isKindOfClass: [FIRDocumentChange class ]]) return NO ;
14642
14743 FIRDocumentChange *change = (FIRDocumentChange *)other;
148- return self.type == change.type && [self .document isEqual: change.document] &&
149- self.oldIndex == change.oldIndex && self.newIndex == change.newIndex ;
44+ return _documentChange == change->_documentChange ;
15045}
15146
15247- (NSUInteger )hash {
153- NSUInteger result = (NSUInteger )self.type ;
154- result = result * 31u + [self .document hash ];
155- result = result * 31u + (NSUInteger )self.oldIndex ;
156- result = result * 31u + (NSUInteger )self.newIndex ;
157- return result;
48+ return _documentChange.Hash ();
49+ }
50+
51+ - (FIRDocumentChangeType)type {
52+ switch (_documentChange.type ()) {
53+ case DocumentChange::Type::Added:
54+ return FIRDocumentChangeTypeAdded;
55+ case DocumentChange::Type::Modified:
56+ return FIRDocumentChangeTypeModified;
57+ case DocumentChange::Type::Removed:
58+ return FIRDocumentChangeTypeRemoved;
59+ }
60+
61+ HARD_FAIL (" Unknown DocumentChange::Type: %s" type ());
62+ }
63+
64+ - (FIRQueryDocumentSnapshot *)document {
65+ return [[FIRQueryDocumentSnapshot alloc ] initWithSnapshot: _documentChange.document ()];
66+ }
67+
68+ - (NSUInteger )oldIndex {
69+ return _documentChange.old_index ();
70+ }
71+
72+ - (NSUInteger )newIndex {
73+ return _documentChange.new_index ();
15874}
15975
16076@end
0 commit comments