NSObject (QMUI) 增加方法用于遍历某个实例的所有方法

MoLice · MoLice · commit d13f7634b4bf · 2017-03-24T10:50:45.000+08:00
diff --git a/QMUIKit/UIKitExtensions/NSObject+QMUI.h b/QMUIKit/UIKitExtensions/NSObject+QMUI.h
@@ -38,6 +38,18 @@
  */
 - (id)qmui_performSelectorToSuperclass:(SEL)aSelector withObject:(id)object;
 
+/**
+ 使用 block 遍历当前实例的所有方法，父类的方法不包含在内
+ */
+- (void)qmui_enumrateInstanceMethodsUsingBlock:(void (^)(SEL selector))block;
+
+/**
+ 使用 block 遍历指定的某个类的实例方法，该类的父类方法不包含在内
+ *  @param aClass   要遍历的某个类
+ *  @param block    遍历时使用的 block，参数为某一个方法
+ */
++ (void)qmui_enumrateInstanceMethodsOfClass:(Class)aClass usingBlock:(void (^)(SEL selector))block;
+
 /**
  遍历某个 protocol 里的所有方法
 
diff --git a/QMUIKit/UIKitExtensions/NSObject+QMUI.m b/QMUIKit/UIKitExtensions/NSObject+QMUI.m
@@ -8,6 +8,7 @@
 
 #import "NSObject+QMUI.h"
 #import <objc/message.h>
+#import <objc/runtime.h>
 
 @implementation NSObject (QMUI)
 
@@ -48,6 +49,23 @@ - (id)qmui_performSelectorToSuperclass:(SEL)aSelector withObject:(id)object {
     return (*objc_superAllocTyped)(&mySuper, aSelector, object);
 }
 
+- (void)qmui_enumrateInstanceMethodsUsingBlock:(void (^)(SEL))block {
+    [NSObject qmui_enumrateInstanceMethodsOfClass:self.class usingBlock:block];
+}
+
++ (void)qmui_enumrateInstanceMethodsOfClass:(Class)aClass usingBlock:(void (^)(SEL selector))block {
+    unsigned int methodCount = 0;
+    Method *methods = class_copyMethodList(aClass, &methodCount);
+    
+    for (unsigned int i = 0; i < methodCount; i++) {
+        Method method = methods[i];
+        SEL selector = method_getName(method);
+        if (block) block(selector);
+    }
+    
+    free(methods);
+}
+
 + (void)qmui_enumerateProtocolMethods:(Protocol *)protocol usingBlock:(void (^)(SEL))block {
     unsigned int methodCount = 0;
     struct objc_method_description *methods = protocol_copyMethodDescriptionList(protocol, NO, YES, &methodCount);
