diff options
Diffstat (limited to 'Firestore/Source/Core/FSTQuery.mm')
| -rw-r--r-- | Firestore/Source/Core/FSTQuery.mm | 61 |
1 files changed, 46 insertions, 15 deletions
diff --git a/Firestore/Source/Core/FSTQuery.mm b/Firestore/Source/Core/FSTQuery.mm index abec474..fd6a999 100644 --- a/Firestore/Source/Core/FSTQuery.mm +++ b/Firestore/Source/Core/FSTQuery.mm @@ -23,6 +23,8 @@ #import "Firestore/Source/API/FIRFirestore+Internal.h" #import "Firestore/Source/Model/FSTDocument.h" #import "Firestore/Source/Model/FSTFieldValue.h" +#import "Firestore/Source/Util/FSTClasses.h" +#import "Firestore/Source/Util/FSTUsageValidation.h" #include "Firestore/core/src/firebase/firestore/model/document_key.h" #include "Firestore/core/src/firebase/firestore/model/field_path.h" @@ -66,6 +68,43 @@ NSString *FSTStringFromQueryRelationOperator(FSTRelationFilterOperator filterOpe } } +@implementation FSTFilter + ++ (instancetype)filterWithField:(const FieldPath &)field + filterOperator:(FSTRelationFilterOperator)op + value:(FSTFieldValue *)value { + if ([value isEqual:[FSTNullValue nullValue]]) { + if (op != FSTRelationFilterOperatorEqual) { + FSTThrowInvalidUsage(@"InvalidQueryException", + @"Invalid Query. You can only perform equality comparisons on nil / " + "NSNull."); + } + return [[FSTNullFilter alloc] initWithField:field]; + } else if ([value isEqual:[FSTDoubleValue nanValue]]) { + if (op != FSTRelationFilterOperatorEqual) { + FSTThrowInvalidUsage(@"InvalidQueryException", + @"Invalid Query. You can only perform equality comparisons on NaN."); + } + return [[FSTNanFilter alloc] initWithField:field]; + } else { + return [[FSTRelationFilter alloc] initWithField:field filterOperator:op value:value]; + } +} + +- (const FieldPath &)field { + @throw FSTAbstractMethodException(); // NOLINT +} + +- (BOOL)matchesDocument:(FSTDocument *)document { + @throw FSTAbstractMethodException(); // NOLINT +} + +- (NSString *)canonicalID { + @throw FSTAbstractMethodException(); // NOLINT +} + +@end + #pragma mark - FSTRelationFilter @interface FSTRelationFilter () { @@ -99,14 +138,6 @@ NSString *FSTStringFromQueryRelationOperator(FSTRelationFilterOperator filterOpe #pragma mark - Constructor methods -+ (instancetype)filterWithField:(FieldPath)field - filterOperator:(FSTRelationFilterOperator)filterOperator - value:(FSTFieldValue *)value { - return [[FSTRelationFilter alloc] initWithField:std::move(field) - filterOperator:filterOperator - value:value]; -} - - (instancetype)initWithField:(FieldPath)field filterOperator:(FSTRelationFilterOperator)filterOperator value:(FSTFieldValue *)value { @@ -513,7 +544,7 @@ NSString *FSTStringFromQueryRelationOperator(FSTRelationFilterOperator filterOpe * @param limit If not NSNotFound, only this many results will be returned. */ - (instancetype)initWithPath:(ResourcePath)path - filterBy:(NSArray<id<FSTFilter>> *)filters + filterBy:(NSArray<FSTFilter *> *)filters orderBy:(NSArray<FSTSortOrder *> *)sortOrders limit:(NSInteger)limit startAt:(nullable FSTBound *)startAtBound @@ -541,7 +572,7 @@ NSString *FSTStringFromQueryRelationOperator(FSTRelationFilterOperator filterOpe } - (instancetype)initWithPath:(ResourcePath)path - filterBy:(NSArray<id<FSTFilter>> *)filters + filterBy:(NSArray<FSTFilter *> *)filters orderBy:(NSArray<FSTSortOrder *> *)sortOrders limit:(NSInteger)limit startAt:(nullable FSTBound *)startAtBound @@ -630,7 +661,7 @@ NSString *FSTStringFromQueryRelationOperator(FSTRelationFilterOperator filterOpe return self.memoizedSortOrders; } -- (instancetype)queryByAddingFilter:(id<FSTFilter>)filter { +- (instancetype)queryByAddingFilter:(FSTFilter *)filter { HARD_ASSERT(!DocumentKey::IsDocumentKey(_path), "No filtering allowed for document query"); const FieldPath *newInequalityField = nullptr; @@ -715,7 +746,7 @@ NSString *FSTStringFromQueryRelationOperator(FSTRelationFilterOperator filterOpe } - (const FieldPath *)inequalityFilterField { - for (id<FSTFilter> filter in self.filters) { + for (FSTFilter *filter in self.filters) { if ([filter isKindOfClass:[FSTRelationFilter class]] && ((FSTRelationFilter *)filter).isInequality) { return &filter.field; @@ -725,7 +756,7 @@ NSString *FSTStringFromQueryRelationOperator(FSTRelationFilterOperator filterOpe } - (BOOL)hasArrayContainsFilter { - for (id<FSTFilter> filter in self.filters) { + for (FSTFilter *filter in self.filters) { if ([filter isKindOfClass:[FSTRelationFilter class]] && ((FSTRelationFilter *)filter).filterOperator == FSTRelationFilterOperatorArrayContains) { return YES; @@ -758,7 +789,7 @@ NSString *FSTStringFromQueryRelationOperator(FSTRelationFilterOperator filterOpe // Add filters. [canonicalID appendString:@"|f:"]; - for (id<FSTFilter> predicate in self.filters) { + for (FSTFilter *predicate in self.filters) { [canonicalID appendFormat:@"%@", [predicate canonicalID]]; } @@ -822,7 +853,7 @@ NSString *FSTStringFromQueryRelationOperator(FSTRelationFilterOperator filterOpe /** Returns YES if the document matches all of the filters in the receiver. */ - (BOOL)filtersMatchDocument:(FSTDocument *)document { - for (id<FSTFilter> filter in self.filters) { + for (FSTFilter *filter in self.filters) { if (![filter matchesDocument:document]) { return NO; } |