add ascend or descend support

davidfuzju · Jul 31, 2015 · a1f1abf · a1f1abf
1 parent 3dfa303
commit a1f1abf
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/Pod/Classes/SMBFetchedResults.h b/Pod/Classes/SMBFetchedResults.h
@@ -25,14 +25,19 @@
 
 @property (readonly, nonatomic, assign) NSStringCompareOptions options;
 
+@property (readonly, nonatomic, assign) BOOL ascend;
+
 @property (readonly, nonatomic, assign) BOOL moving;
 
 @property (readonly, nonatomic, assign) BOOL sorted;
 
 - (void)sortedResultWithOrderedSet:(NSMutableOrderedSet *)orderedSet;
 
 - (instancetype)initWithMutableData:(NSMutableOrderedSet *)mutableData;
-- (instancetype)initWithMutableData:(NSMutableOrderedSet *)mutableData sortKeyPaths:(NSString *)sortKeyPaths sortOptions:(NSStringCompareOptions)options;
+- (instancetype)initWithMutableData:(NSMutableOrderedSet *)mutableData
+                       sortKeyPaths:(NSString *)sortKeyPaths
+                             ascend:(BOOL)ascend
+                        sortOptions:(NSStringCompareOptions)options;
 
 - (void)appendObject:(id <SMBFetchedResultsProtocol>)object;
 - (void)appendObjectsFromArray:(NSArray *)otherArray;

diff --git a/Pod/Classes/SMBFetchedResults.m b/Pod/Classes/SMBFetchedResults.m
@@ -43,14 +43,17 @@ @implementation SMBFetchedResults
 #pragma mark - life cycle
 
 - (instancetype)init {
-    return [self initWithMutableData:[NSMutableOrderedSet orderedSet] sortKeyPaths:nil sortOptions:0];
+    return [self initWithMutableData:[NSMutableOrderedSet orderedSet] sortKeyPaths:nil ascend:YES sortOptions:0];
 }
 
 - (instancetype)initWithMutableData:(NSMutableOrderedSet *)mutableData {
-    return [self initWithMutableData:mutableData sortKeyPaths:nil sortOptions:0];
+    return [self initWithMutableData:mutableData sortKeyPaths:nil ascend:YES sortOptions:0];
 }
 
-- (instancetype)initWithMutableData:(NSMutableOrderedSet *)mutableData sortKeyPaths:(NSString *)sortKeyPaths sortOptions:(NSStringCompareOptions)options {
+- (instancetype)initWithMutableData:(NSMutableOrderedSet *)mutableData
+                       sortKeyPaths:(NSString *)sortKeyPaths
+                             ascend:(BOOL)ascend
+                        sortOptions:(NSStringCompareOptions)options {
     self = [super init];
     if (self) {
         _data = mutableData;
@@ -125,7 +128,7 @@ - (void)sortedResultWithOrderedSet:(NSMutableOrderedSet *)orderedSet {
                                                    }
 
                                                    if (result != NSOrderedSame)
-                                                       return result;
+                                                       return self.ascend? result: -result;
                                                }
                                                return NSOrderedSame;
                                            }];