传入自定义选择器实现

Question

假设我有两个objective-c类，LBFoo和LBBar。

在LBFoo中，我有一个看起来像这样的方法：

- (void)doSomethingWithFoo:(NSNumber*)anArgument
{
  if(anArgument.intValue > 2)
    [LBBar doSomethingWithLBBar];
  else
    [LBBar doSomethingElseWithLBBar];
}

相反，我想做的是将一个未提前声明的实现传递给LBBar。（如动态覆盖LBBar内现有的@选择器）

我知道存在IMP类型，是否可以将IMP传递给类以更改其选择器实现。

Answer 1

您可以在objective-c运行时中使用method_setImplementation(Method method, IMP imp)函数。

如果你想设置一个实例方法，它会像这个一样工作

method_setImplementation(class_getInstanceMethod([yourClass class], @selector(yourMethod)), yourIMP);

如果您想要一个类方法，只需使用class_getClassMethod而不是class_getInstanceMethod。论点应该是相同的。

注意，IMP只是一个void函数指针，前两个参数是id self和SEL _cmd

Answer 2

您当然可以使用运行时函数来做这样的事情，*但我建议这正是引入Blocks来解决的问题。它们允许您传递一块可执行代码——您的方法实际上可以接受Block作为参数并运行它

这是SSCCE:

#import <Foundation/Foundation.h>
typedef dispatch_block_t GenericBlock;
@interface Albatross : NSObject 
- (void)slapFace:(NSNumber *)n usingFish:(GenericBlock)block;
@end
@implementation Albatross
- (void)slapFace:(NSNumber *)n usingFish:(GenericBlock)block
{
    if( [n intValue] > 2 ){
        NSLog(@"Cabbage crates coming over the briny!");
    }
    else {
        block();    // Execute the block
    }
}
@end
int main(int argc, const char * argv[])
{
    @autoreleasepool {
        Albatross * p = [Albatross new];
        [p slapFace:[NSNumber numberWithInt:3] usingFish:^{
            NSLog(@"We'd like to see the dog kennels, please.");
        }];
         [p slapFace:[NSNumber numberWithInt:1] usingFish:^{
            NSLog(@"Lemon curry?");
        }];
    }
    return 0;
}

---

*请注意，使用method_setImplementation()将更改将来每次从任何地方调用该方法时使用的代码——这是一个持久的更改。