给数组加上map方法

gitliubo · Dec 4, 2019 · eaa2238 · eaa2238
1 parent 4a5d7b3
commit eaa2238
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/QMUIKit/UIKitExtensions/NSArray+QMUI.h b/QMUIKit/UIKitExtensions/NSArray+QMUI.h
@@ -39,4 +39,10 @@
  *  过滤数组元素，将 block 返回 YES 的 item 重新组装成一个数组返回
  */
 - (NSArray<ObjectType> *)qmui_filterWithBlock:(BOOL (^)(ObjectType item))block;
+
+/**
+*  转换数组元素，将每个 item 都经过 block 转换成一遍 返回转换后的新数组
+*/
+- (NSArray *)qmui_mapWithBlock:(id (^)(ObjectType item))block;
+
 @end
diff --git a/QMUIKit/UIKitExtensions/NSArray+QMUI.m b/QMUIKit/UIKitExtensions/NSArray+QMUI.m
@@ -84,4 +84,16 @@ - (NSArray *)qmui_filterWithBlock:(BOOL (^)(id))block {
     return [result copy];
 }
 
+- (NSArray *)qmui_mapWithBlock:(id (^)(id item))block {
+    if (!block) {
+        return self;
+    }
+
+    NSMutableArray *result = [[NSMutableArray alloc] initWithCapacity:self.count];
+    for (NSInteger i = 0; i < self.count; i++) {
+        [result addObject:block(self[i])];
+    }
+    return [result copy];
+}
+
 @end