NSArray 的 -valueForKey 问题：当其项为 NSDictionary 时

Question

我有一个包含 NSDictionary 项目的数组，我想将这些项目转换为其他对象，我的第一个想法是 valueForKey:，所以我添加了一个类别方法 toMyObject for NSDictionary，并调用：

[array valueForKey:@"toMyObject"]

但它并没有按预期工作，它只是返回数组NSNull。

如果我不想枚举数组，有什么想法可以解决这个问题吗？

Answer 1

回答我自己。字典的 valueForKey: 会覆盖默认行为，如果字典没有 key，则会返回 nil，不会像 Apple 那样像 NSObject 那样调用访问器方法文件说：

如果 key 不是以“@”开头，则调用 objectForKey:。如果钥匙有
以“@”开头，去掉“@”并调用 [super valueForKey:]
其余的密钥。

由于 NSDictionary 是一个集群类，因此不建议子类化来覆盖该行为。相反，我使用这样的 swiss 方法：

@implementation NSDictionary (MyAddition)

static void swizzle(Class c, SEL orig, SEL new)
{
  Method origMethod = class_getInstanceMethod(c, orig);
  Method newMethod = class_getInstanceMethod(c, new);
  if(class_addMethod(c, orig, method_getImplementation(newMethod), method_getTypeEncoding(newMethod)))
    class_replaceMethod(c, new, method_getImplementation(origMethod), method_getTypeEncoding(origMethod));
  else
    method_exchangeImplementations(origMethod, newMethod);
}

+ (void)initialize
{
  if (self == [NSDictionary class]){
    swizzle([NSDictionary class],
            @selector(valueForKey:),
            @selector(myValueForKey:));
  }
}

- (id)toMyObject
{
  return toMyObject;
}

...

- (id)myValueForKey:(NSString *)key
{
  // for collection operators
  if ([key compare:@"@" options:0 range:NSMakeRange(0, 1)] == NSOrderedSame)
    return [super valueForKey:key];

  if ([key isEqualToString:@"toMyObject"])
    return [self toMyObject];

  return [self myValueForKey:key];
}

现在 NSArray 可以安全地调用 valueForKey:@"toMyObject"。

Answer 2

另一种无需调整的实现：

@implementation NSObject (MLWValueForKey)

- (id)mlw_valueForKey:(NSString *)key {
    if ([key hasPrefix:@"@"]) {
        return [self valueForKey:key];
    }
    NSAssert(![key containsString:@":"], @"Key should be selector without arguments");
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Warc-performSelector-leaks"
    return [self performSelector:NSSelectorFromString(key)];
#pragma clang diagnostic pop
}

@end

@implementation NSArray (MLWValueForKey)

- (id)mlw_valueForKey:(NSString *)key {
    if ([key hasPrefix:@"@"]) {
        return [self valueForKey:key];
    }
    NSMutableArray *array = [NSMutableArray arrayWithCapacity:self.count];
    for (id object in self) {
        [array addObject:[object mlw_valueForKey:key]];
    }
    return array;
}

@end