FIRQuery.mm 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633
  1. /*
  2. * Copyright 2017 Google
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. #import "FIRQuery.h"
  17. #import "FIRDocumentReference.h"
  18. #import "Firestore/Source/API/FIRDocumentReference+Internal.h"
  19. #import "Firestore/Source/API/FIRDocumentSnapshot+Internal.h"
  20. #import "Firestore/Source/API/FIRFieldPath+Internal.h"
  21. #import "Firestore/Source/API/FIRFirestore+Internal.h"
  22. #import "Firestore/Source/API/FIRListenerRegistration+Internal.h"
  23. #import "Firestore/Source/API/FIRQuery+Internal.h"
  24. #import "Firestore/Source/API/FIRQuerySnapshot+Internal.h"
  25. #import "Firestore/Source/API/FIRQuery_Init.h"
  26. #import "Firestore/Source/API/FIRSnapshotMetadata+Internal.h"
  27. #import "Firestore/Source/API/FSTUserDataConverter.h"
  28. #import "Firestore/Source/Core/FSTEventManager.h"
  29. #import "Firestore/Source/Core/FSTFirestoreClient.h"
  30. #import "Firestore/Source/Core/FSTQuery.h"
  31. #import "Firestore/Source/Model/FSTDocument.h"
  32. #import "Firestore/Source/Model/FSTDocumentKey.h"
  33. #import "Firestore/Source/Model/FSTFieldValue.h"
  34. #import "Firestore/Source/Model/FSTPath.h"
  35. #import "Firestore/Source/Util/FSTAssert.h"
  36. #import "Firestore/Source/Util/FSTAsyncQueryListener.h"
  37. #import "Firestore/Source/Util/FSTUsageValidation.h"
  38. NS_ASSUME_NONNULL_BEGIN
  39. @interface FIRQueryListenOptions ()
  40. - (instancetype)initWithIncludeQueryMetadataChanges:(BOOL)includeQueryMetadataChanges
  41. includeDocumentMetadataChanges:(BOOL)includeDocumentMetadataChanges
  42. NS_DESIGNATED_INITIALIZER;
  43. @end
  44. @implementation FIRQueryListenOptions
  45. + (instancetype)options {
  46. return [[FIRQueryListenOptions alloc] init];
  47. }
  48. - (instancetype)initWithIncludeQueryMetadataChanges:(BOOL)includeQueryMetadataChanges
  49. includeDocumentMetadataChanges:(BOOL)includeDocumentMetadataChanges {
  50. if (self = [super init]) {
  51. _includeQueryMetadataChanges = includeQueryMetadataChanges;
  52. _includeDocumentMetadataChanges = includeDocumentMetadataChanges;
  53. }
  54. return self;
  55. }
  56. - (instancetype)init {
  57. return [self initWithIncludeQueryMetadataChanges:NO includeDocumentMetadataChanges:NO];
  58. }
  59. - (instancetype)includeQueryMetadataChanges:(BOOL)includeQueryMetadataChanges {
  60. return [[FIRQueryListenOptions alloc]
  61. initWithIncludeQueryMetadataChanges:includeQueryMetadataChanges
  62. includeDocumentMetadataChanges:_includeDocumentMetadataChanges];
  63. }
  64. - (instancetype)includeDocumentMetadataChanges:(BOOL)includeDocumentMetadataChanges {
  65. return [[FIRQueryListenOptions alloc]
  66. initWithIncludeQueryMetadataChanges:_includeQueryMetadataChanges
  67. includeDocumentMetadataChanges:includeDocumentMetadataChanges];
  68. }
  69. @end
  70. @interface FIRQuery ()
  71. @property(nonatomic, strong, readonly) FSTQuery *query;
  72. @end
  73. @implementation FIRQuery (Internal)
  74. + (instancetype)referenceWithQuery:(FSTQuery *)query firestore:(FIRFirestore *)firestore {
  75. return [[FIRQuery alloc] initWithQuery:query firestore:firestore];
  76. }
  77. @end
  78. @implementation FIRQuery
  79. #pragma mark - Constructor Methods
  80. - (instancetype)initWithQuery:(FSTQuery *)query firestore:(FIRFirestore *)firestore {
  81. if (self = [super init]) {
  82. _query = query;
  83. _firestore = firestore;
  84. }
  85. return self;
  86. }
  87. #pragma mark - NSObject Methods
  88. - (BOOL)isEqual:(nullable id)other {
  89. if (other == self) return YES;
  90. if (![[other class] isEqual:[self class]]) return NO;
  91. return [self isEqualToQuery:other];
  92. }
  93. - (BOOL)isEqualToQuery:(nullable FIRQuery *)query {
  94. if (self == query) return YES;
  95. if (query == nil) return NO;
  96. return [self.firestore isEqual:query.firestore] && [self.query isEqual:query.query];
  97. }
  98. - (NSUInteger)hash {
  99. NSUInteger hash = [self.firestore hash];
  100. hash = hash * 31u + [self.query hash];
  101. return hash;
  102. }
  103. #pragma mark - Public Methods
  104. - (void)getDocumentsWithCompletion:(void (^)(FIRQuerySnapshot *_Nullable snapshot,
  105. NSError *_Nullable error))completion {
  106. FSTListenOptions *options = [[FSTListenOptions alloc] initWithIncludeQueryMetadataChanges:YES
  107. includeDocumentMetadataChanges:YES
  108. waitForSyncWhenOnline:YES];
  109. dispatch_semaphore_t registered = dispatch_semaphore_create(0);
  110. __block id<FIRListenerRegistration> listenerRegistration;
  111. FIRQuerySnapshotBlock listener = ^(FIRQuerySnapshot *snapshot, NSError *error) {
  112. if (error) {
  113. completion(nil, error);
  114. return;
  115. }
  116. // Remove query first before passing event to user to avoid user actions affecting the
  117. // now stale query.
  118. dispatch_semaphore_wait(registered, DISPATCH_TIME_FOREVER);
  119. [listenerRegistration remove];
  120. completion(snapshot, nil);
  121. };
  122. listenerRegistration = [self addSnapshotListenerInternalWithOptions:options listener:listener];
  123. dispatch_semaphore_signal(registered);
  124. }
  125. - (id<FIRListenerRegistration>)addSnapshotListener:(FIRQuerySnapshotBlock)listener {
  126. return [self addSnapshotListenerWithOptions:nil listener:listener];
  127. }
  128. - (id<FIRListenerRegistration>)addSnapshotListenerWithOptions:
  129. (nullable FIRQueryListenOptions *)options
  130. listener:(FIRQuerySnapshotBlock)listener {
  131. return [self addSnapshotListenerInternalWithOptions:[self internalOptions:options]
  132. listener:listener];
  133. }
  134. - (id<FIRListenerRegistration>)
  135. addSnapshotListenerInternalWithOptions:(FSTListenOptions *)internalOptions
  136. listener:(FIRQuerySnapshotBlock)listener {
  137. FIRFirestore *firestore = self.firestore;
  138. FSTQuery *query = self.query;
  139. FSTViewSnapshotHandler snapshotHandler = ^(FSTViewSnapshot *snapshot, NSError *error) {
  140. if (error) {
  141. listener(nil, error);
  142. return;
  143. }
  144. FIRSnapshotMetadata *metadata =
  145. [FIRSnapshotMetadata snapshotMetadataWithPendingWrites:snapshot.hasPendingWrites
  146. fromCache:snapshot.fromCache];
  147. listener([FIRQuerySnapshot snapshotWithFirestore:firestore
  148. originalQuery:query
  149. snapshot:snapshot
  150. metadata:metadata],
  151. nil);
  152. };
  153. FSTAsyncQueryListener *asyncListener =
  154. [[FSTAsyncQueryListener alloc] initWithDispatchQueue:self.firestore.client.userDispatchQueue
  155. snapshotHandler:snapshotHandler];
  156. FSTQueryListener *internalListener =
  157. [firestore.client listenToQuery:query
  158. options:internalOptions
  159. viewSnapshotHandler:[asyncListener asyncSnapshotHandler]];
  160. return [[FSTListenerRegistration alloc] initWithClient:self.firestore.client
  161. asyncListener:asyncListener
  162. internalListener:internalListener];
  163. }
  164. - (FIRQuery *)queryWhereField:(NSString *)field isEqualTo:(id)value {
  165. return [self queryWithFilterOperator:FSTRelationFilterOperatorEqual field:field value:value];
  166. }
  167. - (FIRQuery *)queryWhereFieldPath:(FIRFieldPath *)path isEqualTo:(id)value {
  168. return [self queryWithFilterOperator:FSTRelationFilterOperatorEqual
  169. path:path.internalValue
  170. value:value];
  171. }
  172. - (FIRQuery *)queryWhereField:(NSString *)field isLessThan:(id)value {
  173. return [self queryWithFilterOperator:FSTRelationFilterOperatorLessThan field:field value:value];
  174. }
  175. - (FIRQuery *)queryWhereFieldPath:(FIRFieldPath *)path isLessThan:(id)value {
  176. return [self queryWithFilterOperator:FSTRelationFilterOperatorLessThan
  177. path:path.internalValue
  178. value:value];
  179. }
  180. - (FIRQuery *)queryWhereField:(NSString *)field isLessThanOrEqualTo:(id)value {
  181. return [self queryWithFilterOperator:FSTRelationFilterOperatorLessThanOrEqual
  182. field:field
  183. value:value];
  184. }
  185. - (FIRQuery *)queryWhereFieldPath:(FIRFieldPath *)path isLessThanOrEqualTo:(id)value {
  186. return [self queryWithFilterOperator:FSTRelationFilterOperatorLessThanOrEqual
  187. path:path.internalValue
  188. value:value];
  189. }
  190. - (FIRQuery *)queryWhereField:(NSString *)field isGreaterThan:(id)value {
  191. return
  192. [self queryWithFilterOperator:FSTRelationFilterOperatorGreaterThan field:field value:value];
  193. }
  194. - (FIRQuery *)queryWhereFieldPath:(FIRFieldPath *)path isGreaterThan:(id)value {
  195. return [self queryWithFilterOperator:FSTRelationFilterOperatorGreaterThan
  196. path:path.internalValue
  197. value:value];
  198. }
  199. - (FIRQuery *)queryWhereField:(NSString *)field isGreaterThanOrEqualTo:(id)value {
  200. return [self queryWithFilterOperator:FSTRelationFilterOperatorGreaterThanOrEqual
  201. field:field
  202. value:value];
  203. }
  204. - (FIRQuery *)queryWhereFieldPath:(FIRFieldPath *)path isGreaterThanOrEqualTo:(id)value {
  205. return [self queryWithFilterOperator:FSTRelationFilterOperatorGreaterThanOrEqual
  206. path:path.internalValue
  207. value:value];
  208. }
  209. - (FIRQuery *)queryFilteredUsingComparisonPredicate:(NSPredicate *)predicate {
  210. NSComparisonPredicate *comparison = (NSComparisonPredicate *)predicate;
  211. if (comparison.comparisonPredicateModifier != NSDirectPredicateModifier) {
  212. FSTThrowInvalidArgument(@"Invalid query. Predicate cannot have an aggregate modifier.");
  213. }
  214. NSString *path;
  215. id value = nil;
  216. if ([comparison.leftExpression expressionType] == NSKeyPathExpressionType &&
  217. [comparison.rightExpression expressionType] == NSConstantValueExpressionType) {
  218. path = comparison.leftExpression.keyPath;
  219. value = comparison.rightExpression.constantValue;
  220. switch (comparison.predicateOperatorType) {
  221. case NSEqualToPredicateOperatorType:
  222. return [self queryWhereField:path isEqualTo:value];
  223. case NSLessThanPredicateOperatorType:
  224. return [self queryWhereField:path isLessThan:value];
  225. case NSLessThanOrEqualToPredicateOperatorType:
  226. return [self queryWhereField:path isLessThanOrEqualTo:value];
  227. case NSGreaterThanPredicateOperatorType:
  228. return [self queryWhereField:path isGreaterThan:value];
  229. case NSGreaterThanOrEqualToPredicateOperatorType:
  230. return [self queryWhereField:path isGreaterThanOrEqualTo:value];
  231. default:; // Fallback below to throw assertion.
  232. }
  233. } else if ([comparison.leftExpression expressionType] == NSConstantValueExpressionType &&
  234. [comparison.rightExpression expressionType] == NSKeyPathExpressionType) {
  235. path = comparison.rightExpression.keyPath;
  236. value = comparison.leftExpression.constantValue;
  237. switch (comparison.predicateOperatorType) {
  238. case NSEqualToPredicateOperatorType:
  239. return [self queryWhereField:path isEqualTo:value];
  240. case NSLessThanPredicateOperatorType:
  241. return [self queryWhereField:path isGreaterThan:value];
  242. case NSLessThanOrEqualToPredicateOperatorType:
  243. return [self queryWhereField:path isGreaterThanOrEqualTo:value];
  244. case NSGreaterThanPredicateOperatorType:
  245. return [self queryWhereField:path isLessThan:value];
  246. case NSGreaterThanOrEqualToPredicateOperatorType:
  247. return [self queryWhereField:path isLessThanOrEqualTo:value];
  248. default:; // Fallback below to throw assertion.
  249. }
  250. } else {
  251. FSTThrowInvalidArgument(
  252. @"Invalid query. Predicate comparisons must include a key path and a constant.");
  253. }
  254. // Fallback cases of unsupported comparison operator.
  255. switch (comparison.predicateOperatorType) {
  256. case NSCustomSelectorPredicateOperatorType:
  257. FSTThrowInvalidArgument(@"Invalid query. Custom predicate filters are not supported.");
  258. break;
  259. default:
  260. FSTThrowInvalidArgument(@"Invalid query. Operator type %lu is not supported.",
  261. (unsigned long)comparison.predicateOperatorType);
  262. }
  263. }
  264. - (FIRQuery *)queryFilteredUsingCompoundPredicate:(NSPredicate *)predicate {
  265. NSCompoundPredicate *compound = (NSCompoundPredicate *)predicate;
  266. if (compound.compoundPredicateType != NSAndPredicateType || compound.subpredicates.count == 0) {
  267. FSTThrowInvalidArgument(@"Invalid query. Only compound queries using AND are supported.");
  268. }
  269. FIRQuery *query = self;
  270. for (NSPredicate *pred in compound.subpredicates) {
  271. query = [query queryFilteredUsingPredicate:pred];
  272. }
  273. return query;
  274. }
  275. - (FIRQuery *)queryFilteredUsingPredicate:(NSPredicate *)predicate {
  276. if ([predicate isKindOfClass:[NSComparisonPredicate class]]) {
  277. return [self queryFilteredUsingComparisonPredicate:predicate];
  278. } else if ([predicate isKindOfClass:[NSCompoundPredicate class]]) {
  279. return [self queryFilteredUsingCompoundPredicate:predicate];
  280. } else if ([predicate isKindOfClass:[[NSPredicate
  281. predicateWithBlock:^BOOL(id obj, NSDictionary *bindings) {
  282. return true;
  283. }] class]]) {
  284. FSTThrowInvalidArgument(
  285. @"Invalid query. Block-based predicates are not "
  286. "supported. Please use predicateWithFormat to "
  287. "create predicates instead.");
  288. } else {
  289. FSTThrowInvalidArgument(
  290. @"Invalid query. Expect comparison or compound of "
  291. "comparison predicate. Please use "
  292. "predicateWithFormat to create predicates.");
  293. }
  294. }
  295. - (FIRQuery *)queryOrderedByField:(NSString *)field {
  296. return
  297. [self queryOrderedByFieldPath:[FIRFieldPath pathWithDotSeparatedString:field] descending:NO];
  298. }
  299. - (FIRQuery *)queryOrderedByFieldPath:(FIRFieldPath *)fieldPath {
  300. return [self queryOrderedByFieldPath:fieldPath descending:NO];
  301. }
  302. - (FIRQuery *)queryOrderedByField:(NSString *)field descending:(BOOL)descending {
  303. return [self queryOrderedByFieldPath:[FIRFieldPath pathWithDotSeparatedString:field]
  304. descending:descending];
  305. }
  306. - (FIRQuery *)queryOrderedByFieldPath:(FIRFieldPath *)fieldPath descending:(BOOL)descending {
  307. [self validateNewOrderByPath:fieldPath.internalValue];
  308. if (self.query.startAt) {
  309. FSTThrowInvalidUsage(
  310. @"InvalidQueryException",
  311. @"Invalid query. You must not specify a starting point before specifying the order by.");
  312. }
  313. if (self.query.endAt) {
  314. FSTThrowInvalidUsage(
  315. @"InvalidQueryException",
  316. @"Invalid query. You must not specify an ending point before specifying the order by.");
  317. }
  318. FSTSortOrder *sortOrder =
  319. [FSTSortOrder sortOrderWithFieldPath:fieldPath.internalValue ascending:!descending];
  320. return [FIRQuery referenceWithQuery:[self.query queryByAddingSortOrder:sortOrder]
  321. firestore:self.firestore];
  322. }
  323. - (FIRQuery *)queryLimitedTo:(NSInteger)limit {
  324. if (limit <= 0) {
  325. FSTThrowInvalidArgument(@"Invalid Query. Query limit (%ld) is invalid. Limit must be positive.",
  326. (long)limit);
  327. }
  328. return [FIRQuery referenceWithQuery:[self.query queryBySettingLimit:limit] firestore:_firestore];
  329. }
  330. - (FIRQuery *)queryStartingAtDocument:(FIRDocumentSnapshot *)snapshot {
  331. FSTBound *bound = [self boundFromSnapshot:snapshot isBefore:YES];
  332. return [FIRQuery referenceWithQuery:[self.query queryByAddingStartAt:bound]
  333. firestore:self.firestore];
  334. }
  335. - (FIRQuery *)queryStartingAtValues:(NSArray *)fieldValues {
  336. FSTBound *bound = [self boundFromFieldValues:fieldValues isBefore:YES];
  337. return [FIRQuery referenceWithQuery:[self.query queryByAddingStartAt:bound]
  338. firestore:self.firestore];
  339. }
  340. - (FIRQuery *)queryStartingAfterDocument:(FIRDocumentSnapshot *)snapshot {
  341. FSTBound *bound = [self boundFromSnapshot:snapshot isBefore:NO];
  342. return [FIRQuery referenceWithQuery:[self.query queryByAddingStartAt:bound]
  343. firestore:self.firestore];
  344. }
  345. - (FIRQuery *)queryStartingAfterValues:(NSArray *)fieldValues {
  346. FSTBound *bound = [self boundFromFieldValues:fieldValues isBefore:NO];
  347. return [FIRQuery referenceWithQuery:[self.query queryByAddingStartAt:bound]
  348. firestore:self.firestore];
  349. }
  350. - (FIRQuery *)queryEndingBeforeDocument:(FIRDocumentSnapshot *)snapshot {
  351. FSTBound *bound = [self boundFromSnapshot:snapshot isBefore:YES];
  352. return
  353. [FIRQuery referenceWithQuery:[self.query queryByAddingEndAt:bound] firestore:self.firestore];
  354. }
  355. - (FIRQuery *)queryEndingBeforeValues:(NSArray *)fieldValues {
  356. FSTBound *bound = [self boundFromFieldValues:fieldValues isBefore:YES];
  357. return
  358. [FIRQuery referenceWithQuery:[self.query queryByAddingEndAt:bound] firestore:self.firestore];
  359. }
  360. - (FIRQuery *)queryEndingAtDocument:(FIRDocumentSnapshot *)snapshot {
  361. FSTBound *bound = [self boundFromSnapshot:snapshot isBefore:NO];
  362. return
  363. [FIRQuery referenceWithQuery:[self.query queryByAddingEndAt:bound] firestore:self.firestore];
  364. }
  365. - (FIRQuery *)queryEndingAtValues:(NSArray *)fieldValues {
  366. FSTBound *bound = [self boundFromFieldValues:fieldValues isBefore:NO];
  367. return
  368. [FIRQuery referenceWithQuery:[self.query queryByAddingEndAt:bound] firestore:self.firestore];
  369. }
  370. #pragma mark - Private Methods
  371. /** Private helper for all of the queryWhereField: methods. */
  372. - (FIRQuery *)queryWithFilterOperator:(FSTRelationFilterOperator)filterOperator
  373. field:(NSString *)field
  374. value:(id)value {
  375. return [self queryWithFilterOperator:filterOperator
  376. path:[FIRFieldPath pathWithDotSeparatedString:field].internalValue
  377. value:value];
  378. }
  379. - (FIRQuery *)queryWithFilterOperator:(FSTRelationFilterOperator)filterOperator
  380. path:(FSTFieldPath *)fieldPath
  381. value:(id)value {
  382. FSTFieldValue *fieldValue;
  383. if ([fieldPath isKeyFieldPath]) {
  384. if ([value isKindOfClass:[NSString class]]) {
  385. NSString *documentKey = (NSString *)value;
  386. if ([documentKey containsString:@"/"]) {
  387. FSTThrowInvalidArgument(
  388. @"Invalid query. When querying by document ID you must provide "
  389. "a valid document ID, but '%@' contains a '/' character.",
  390. documentKey);
  391. } else if (documentKey.length == 0) {
  392. FSTThrowInvalidArgument(
  393. @"Invalid query. When querying by document ID you must provide "
  394. "a valid document ID, but it was an empty string.");
  395. }
  396. FSTResourcePath *path = [self.query.path pathByAppendingSegment:documentKey];
  397. fieldValue = [FSTReferenceValue referenceValue:[FSTDocumentKey keyWithPath:path]
  398. databaseID:self.firestore.databaseID];
  399. } else if ([value isKindOfClass:[FIRDocumentReference class]]) {
  400. FIRDocumentReference *ref = (FIRDocumentReference *)value;
  401. fieldValue = [FSTReferenceValue referenceValue:ref.key databaseID:self.firestore.databaseID];
  402. } else {
  403. FSTThrowInvalidArgument(
  404. @"Invalid query. When querying by document ID you must provide a "
  405. "valid string or DocumentReference, but it was of type: %@",
  406. NSStringFromClass([value class]));
  407. }
  408. } else {
  409. fieldValue = [self.firestore.dataConverter parsedQueryValue:value];
  410. }
  411. id<FSTFilter> filter;
  412. if ([fieldValue isEqual:[FSTNullValue nullValue]]) {
  413. if (filterOperator != FSTRelationFilterOperatorEqual) {
  414. FSTThrowInvalidUsage(@"InvalidQueryException",
  415. @"Invalid Query. You can only perform equality comparisons on nil / "
  416. "NSNull.");
  417. }
  418. filter = [[FSTNullFilter alloc] initWithField:fieldPath];
  419. } else if ([fieldValue isEqual:[FSTDoubleValue nanValue]]) {
  420. if (filterOperator != FSTRelationFilterOperatorEqual) {
  421. FSTThrowInvalidUsage(@"InvalidQueryException",
  422. @"Invalid Query. You can only perform equality comparisons on NaN.");
  423. }
  424. filter = [[FSTNanFilter alloc] initWithField:fieldPath];
  425. } else {
  426. filter = [FSTRelationFilter filterWithField:fieldPath
  427. filterOperator:filterOperator
  428. value:fieldValue];
  429. [self validateNewRelationFilter:filter];
  430. }
  431. return [FIRQuery referenceWithQuery:[self.query queryByAddingFilter:filter]
  432. firestore:self.firestore];
  433. }
  434. - (void)validateNewRelationFilter:(FSTRelationFilter *)filter {
  435. if ([filter isInequality]) {
  436. FSTFieldPath *existingField = [self.query inequalityFilterField];
  437. if (existingField && ![existingField isEqual:filter.field]) {
  438. FSTThrowInvalidUsage(
  439. @"InvalidQueryException",
  440. @"Invalid Query. All where filters with an inequality "
  441. "(lessThan, lessThanOrEqual, greaterThan, or greaterThanOrEqual) must be on the same "
  442. "field. But you have inequality filters on '%@' and '%@'",
  443. existingField, filter.field);
  444. }
  445. FSTFieldPath *firstOrderByField = [self.query firstSortOrderField];
  446. if (firstOrderByField) {
  447. [self validateOrderByField:firstOrderByField matchesInequalityField:filter.field];
  448. }
  449. }
  450. }
  451. - (void)validateNewOrderByPath:(FSTFieldPath *)fieldPath {
  452. if (![self.query firstSortOrderField]) {
  453. // This is the first order by. It must match any inequality.
  454. FSTFieldPath *inequalityField = [self.query inequalityFilterField];
  455. if (inequalityField) {
  456. [self validateOrderByField:fieldPath matchesInequalityField:inequalityField];
  457. }
  458. }
  459. }
  460. - (void)validateOrderByField:(FSTFieldPath *)orderByField
  461. matchesInequalityField:(FSTFieldPath *)inequalityField {
  462. if (!([orderByField isEqual:inequalityField])) {
  463. FSTThrowInvalidUsage(
  464. @"InvalidQueryException",
  465. @"Invalid query. You have a where filter with an "
  466. "inequality (lessThan, lessThanOrEqual, greaterThan, or greaterThanOrEqual) on field '%@' "
  467. "and so you must also use '%@' as your first queryOrderedBy field, but your first "
  468. "queryOrderedBy is currently on field '%@' instead.",
  469. inequalityField, inequalityField, orderByField);
  470. }
  471. }
  472. /**
  473. * Create a FSTBound from a query given the document.
  474. *
  475. * Note that the FSTBound will always include the key of the document and the position will be
  476. * unambiguous.
  477. *
  478. * Will throw if the document does not contain all fields of the order by of the query.
  479. */
  480. - (FSTBound *)boundFromSnapshot:(FIRDocumentSnapshot *)snapshot isBefore:(BOOL)isBefore {
  481. if (![snapshot exists]) {
  482. FSTThrowInvalidUsage(@"InvalidQueryException",
  483. @"Invalid query. You are trying to start or end a query using a document "
  484. @"that doesn't exist.");
  485. }
  486. FSTDocument *document = snapshot.internalDocument;
  487. NSMutableArray<FSTFieldValue *> *components = [NSMutableArray array];
  488. // Because people expect to continue/end a query at the exact document provided, we need to
  489. // use the implicit sort order rather than the explicit sort order, because it's guaranteed to
  490. // contain the document key. That way the position becomes unambiguous and the query
  491. // continues/ends exactly at the provided document. Without the key (by using the explicit sort
  492. // orders), multiple documents could match the position, yielding duplicate results.
  493. for (FSTSortOrder *sortOrder in self.query.sortOrders) {
  494. if ([sortOrder.field isEqual:[FSTFieldPath keyFieldPath]]) {
  495. [components addObject:[FSTReferenceValue referenceValue:document.key
  496. databaseID:self.firestore.databaseID]];
  497. } else {
  498. FSTFieldValue *value = [document fieldForPath:sortOrder.field];
  499. if (value != nil) {
  500. [components addObject:value];
  501. } else {
  502. FSTThrowInvalidUsage(@"InvalidQueryException",
  503. @"Invalid query. You are trying to start or end a query using a "
  504. "document for which the field '%@' (used as the order by) "
  505. "does not exist.",
  506. sortOrder.field.canonicalString);
  507. }
  508. }
  509. }
  510. return [FSTBound boundWithPosition:components isBefore:isBefore];
  511. }
  512. /** Converts a list of field values to an FSTBound. */
  513. - (FSTBound *)boundFromFieldValues:(NSArray<id> *)fieldValues isBefore:(BOOL)isBefore {
  514. // Use explicit sort order because it has to match the query the user made
  515. NSArray<FSTSortOrder *> *explicitSortOrders = self.query.explicitSortOrders;
  516. if (fieldValues.count > explicitSortOrders.count) {
  517. FSTThrowInvalidUsage(@"InvalidQueryException",
  518. @"Invalid query. You are trying to start or end a query using more values "
  519. @"than were specified in the order by.");
  520. }
  521. NSMutableArray<FSTFieldValue *> *components = [NSMutableArray array];
  522. [fieldValues enumerateObjectsUsingBlock:^(id rawValue, NSUInteger idx, BOOL *stop) {
  523. FSTSortOrder *sortOrder = explicitSortOrders[idx];
  524. if ([sortOrder.field isEqual:[FSTFieldPath keyFieldPath]]) {
  525. if (![rawValue isKindOfClass:[NSString class]]) {
  526. FSTThrowInvalidUsage(@"InvalidQueryException",
  527. @"Invalid query. Expected a string for the document ID.");
  528. }
  529. NSString *documentID = (NSString *)rawValue;
  530. if ([documentID containsString:@"/"]) {
  531. FSTThrowInvalidUsage(@"InvalidQueryException",
  532. @"Invalid query. Document ID '%@' contains a slash.", documentID);
  533. }
  534. FSTDocumentKey *key =
  535. [FSTDocumentKey keyWithPath:[self.query.path pathByAppendingSegment:documentID]];
  536. [components
  537. addObject:[FSTReferenceValue referenceValue:key databaseID:self.firestore.databaseID]];
  538. } else {
  539. FSTFieldValue *fieldValue = [self.firestore.dataConverter parsedQueryValue:rawValue];
  540. [components addObject:fieldValue];
  541. }
  542. }];
  543. return [FSTBound boundWithPosition:components isBefore:isBefore];
  544. }
  545. /** Converts the public API options object to the internal options object. */
  546. - (FSTListenOptions *)internalOptions:(nullable FIRQueryListenOptions *)options {
  547. return [[FSTListenOptions alloc]
  548. initWithIncludeQueryMetadataChanges:options.includeQueryMetadataChanges
  549. includeDocumentMetadataChanges:options.includeDocumentMetadataChanges
  550. waitForSyncWhenOnline:NO];
  551. }
  552. @end
  553. NS_ASSUME_NONNULL_END