Objective-C 中的动态方法创建

Question

在The Pragmatic Programmer一书中，作者建议所有方法输入都应该经过验证。这样可以尽早发现方法中的问题并轻松追踪其来源。

在我的 Mac 应用程序中，我通过创建一个 Assert 类来完成此任务。这个类有几个类方法。这些方法确定是否满足某些先决条件，如果不满足，则抛出异常。典型的断言可能看起来像这样：

-(void) setWidth: (int) theWidth {
    [Assert integer: width isGreaterThanInteger: 0];
    width = theWidth;
}

这非常有效，并且显着减少了我花在寻找错误上的时间。然而，我最近注意到一些断言方法作为谓词非常有用。例如，我的 integer:isGreaterThanInteger:andLessThanInteger: 和我的 stringIsNotEmpty: 方法同样有用。为此，我创建了第二个类 Predicate，其中填充了几个更有用的谓词方法。因此，我从断言方法中取出逻辑，并将其移至 Predicate 中，然后重写我的 Assert 方法，如下所示：

if ![Predicate predicateMethod]
    throw exception

这已变成维护噩梦。如果我在 Predicate 中更改方法名称，我还必须在 Assert 中更改它以保持一致。如果我更新 Assert 方法的文档，那么我必须对 Predicate 方法执行相同的操作。

理想情况下，我希望重建 Assert 类，以便在对其调用任何方法时，它会拦截选择器。然后可以检查 Predicate 类以查看它是否响应选择器，如果响应，则使用传递到 Predicate 的相同参数在 Predicate 上调用该方法。 断言方法。如果 Predicate 方法返回 false，则会引发异常。

Objective-C 有没有办法做到这一点？

谢谢。

Answer 1

您可以使用 -forwardingTargetForSelector: 简单地将方法转发给另一个方法对象，但如果您想要高级行为（例如检查返回值以查看其是否为 false），您可能需要使用 -forwardInspiration：。 （但是，请注意，文档说这比前一种选择“昂贵得多”。）

Answer 2

如果您使用纯 Objective-C，您应该在此处查看“转发”讨论。它基本上描述了如何准确地执行您想要的操作，包括示例代码。

如果您使用 Cocoa，那么您可能必须使用 forwardInitation： 代替。

Answer 3

我最终覆盖了 resolveClassMethod:。虽然覆盖 forwardInspiration 可能有效（我必须找出某种方法来为类对象覆盖它），但 resolveClassMethod: 似乎是更简单、更有效的方法。这是我的最终实现的样子：

#import "Assert.h"
#import "Predicate.h"
#include <objc/objc-runtime.h>

void handlePredicateSelector(id self, SEL _cmd, ...);

@implementation Assert

+(void) failWithMessage: (NSString *) message
{
    NSLog(@"%@", message);
    [NSException raise:@"ASSERTION FAILURE" format:message];
}

+(void) fail
{
    [Assert failWithMessage:@"An unconditional failure has been detected."];
}

+(BOOL) resolveClassMethod: (SEL) selector
{
    if ([(id) [Predicate class] respondsToSelector:selector])
    {
        /*
         The meta class fix was taken from here: http://iphonedevelopment.blogspot.com/2008/08/dynamically-adding-class-objects.html
         */

        //get the method properties from the Predicate class
        Class predicateMetaClass = objc_getMetaClass([[Predicate className] UTF8String]);
        Method predicateMethod = class_getClassMethod(predicateMetaClass, selector);
        const char *encoding = method_getTypeEncoding(predicateMethod);

        Class selfMetaClass = objc_getMetaClass([[self className] UTF8String]);
        class_addMethod(selfMetaClass, selector, (IMP) handlePredicateSelector, "B@:?");

        return YES;
    }
    return [super resolveClassMethod:selector];
}

@end

void handlePredicateSelector(id self, SEL _cmd, ...)
{
    //get the number of arguments minus the self and _cmd arguments
    NSMethodSignature *predicateMethodSignature = [(id) [Predicate class] methodSignatureForSelector:_cmd];
    NSUInteger numberOfArguments = [predicateMethodSignature numberOfArguments] - 2;

    NSInvocation *predicateInvocation = [NSInvocation invocationWithMethodSignature:predicateMethodSignature];
    [predicateInvocation setTarget:[Predicate class]];
    [predicateInvocation setSelector:_cmd];

    va_list ap;
    va_start(ap, _cmd);

    for (int i = 0; i < numberOfArguments; i++)
    {
        void *arg = va_arg(ap, void *);
        [predicateInvocation setArgument:&arg atIndex:i+2];
    }

    va_end(ap);

    BOOL returnValue;
    [predicateInvocation invoke];
    [predicateInvocation getReturnValue:&returnValue];

    //determine if the assertion is true
    if (!returnValue)
    {
        [Assert failWithMessage:[NSString stringWithFormat: @"The following assertion failed: %@", NSStringFromSelector(_cmd)]];
    }
}

我唯一无法真正弄清楚的是如何从方法签名中获取类型编码。它似乎没有影响方法的输出，但如果可以的话我想修复它。