为Objective-C的NSArray实现Swift的Sequence的filter方法

0x00 背景：

在使用swift的squence的时候，经常会给使用filter方法来过滤一些自己想要的东西。想到Objective-C中也有类似的方式，但是是使用了NSPredicate来转化的，所以就引出了此文，看看是否能像swift的filter一样来实现（但使用NSPredicate实际可以更普遍的处理更多类型的问题，此文只是处理比较简单的一些情况）。

 

0x01 问题

我们在使用NSArray的时候，有时会有这样的需求。

 

• 我要列表里面所有字符数大于7的字符串
• 过滤掉user列表里age < 18的

假设给定字符串列表如下：

NSArray *strings = @[@"abcdefg", @"abcdefg123", @"abcde", @"whosyourdaddy", @"givemeafavor",@"adobe",@"od"];

我们要过滤出其中字符数大于7的字符串，一般我们会这样处理：

循环过滤

    NSMutableArray *result = [NSMutableArray new];
    [strings enumerateObjectsUsingBlock:^(NSString *obj, NSUInteger idx, BOOL * _Nonnull stop) {
        if (obj.length > 7) {
            [result addObject:obj];
        }
    }];
    NSLog(@"filter strings is %@", result);

NSPredicate

    NSLog(@"test predicate array is %@", [strings filteredArrayUsingPredicate:[NSPredicate predicateWithFormat:@"length > 7"]]);

那有没有什么方式，像swift里的filter一样来实现呢？

let testArray = strings.filter { (element) -> Bool in
    return element.count > 7 
}
print(testArray)

 

0x02 过程

看swift的方法，入参是接收了一个有返回值为Bool类型的block，大概猜测一下内部的实现可能是内部也是针对strings做了一个循环，将符合条件 element.count > 7的字符串取出来，组合成一个列表再返回。不过鉴于swift源码已开放，最好的办法，也算是不费神的做法，我们直接去查看一下swift中filter方法的源码。

打开链接 https://github.com/apple/swift/blob/master/stdlib/public/core/Sequence.swift

搜索 "filter", 找到它真实的实现方法，在 863 行左右：

public func _filter(
    _ isIncluded: (Element) throws -> Bool
  ) rethrows -> [Element] {

    var result = ContiguousArray<Element>()

    var iterator = self.makeIterator()

    while let element = iterator.next() {
      if try isIncluded(element) {
        result.append(element)
      }
    }

    return Array(result)
  }

isIncluded即是传入的block，看它的实现是使用了本身的一个迭代器来实现循环将 满足isIncluded的就将元素加入新列表，结束之后返回新生成的结果列表result。

 

0x03 实现

那我们就按照这个逻辑来处理好了

新建一个NSArray的类别，实现方法：

- (NSArray *)sn_filter:(BOOL (^)(NSObject **obj))filterBlock {
    NSMutableArray *result = [NSMutableArray array];
    NSEnumerator *iterator = self.objectEnumerator;
    
    NSObject *object = iterator.nextObject;
    if (filterBlock(&object)) {
        [result addObject:object];
    }
    
    return result.copy;
}

obj给两星号， 是为了能和swift方法一样，作为一个可修改的入参两边使用

现在这样调用一下：

    NSLog(@"test result is %@", [strings sn_filter:^BOOL(NSObject *__autoreleasing *obj) {
        NSString *item = *obj;
        return item.length > 7;
    }]);

输出为：

 

2018-04-09 10:30:55.357225+0800 Test[34117:1209673] filter strings is (

    abcdefg123,

    whosyourdaddy,

    givemeafavor

)