动手实现 KVC 键值编码

#Original (English)

Source: https://www.mikeash.com/pyblog/friday-qa-2013-02-08-lets-build-key-value-coding.html

Last time, I showed how to build the basic functionality of NSObject. I left out key-value coding, because the implementation of valueForKey: and setValue:forKey: is complex enough to need its own article. This is that article.

BasicsKey-value coding (KVC) is an API that allows string-based access to object properties. NSObject implements the methods to look up accessor methods or instance variables based on the key name, and fetch or set the value using those.

There are two basic methods that form the basis of KVC.

The valueForKey: method searches for a getter method with the same name as the key. If found, it calls the method and returns its return value. If none is found, it searches for an instance variable with the same name as the key. Failing those, it looks for an instance variable with the same name as the key, but prefixed with an underscore. If an instance variable is found, it returns the value it currently holds.

The setValue:forKey: method performs the same search, except that it searches for a setter method rather than a getter. It then either calls the setter or sets the instance variable directly.

An interesting feature of both of these methods is that they work with primitive values by automatically boxing and unboxing them into instances of NSNumber or NSValue. You can use valueForKey: to invoke a method that returns int, and the result will be an NSNumber object containing the return value. Likewise, you can use setValue:forKey: to invoke a method that takes int, pass it an NSNumber, and it will automatically extract the integer value.

KVC also has the concept of key paths, which are sequences of keys put together with periods, like:

1
    foo.bar.baz

There are corresponding methods to work with key paths: valueForKeyPath: and setValue:forKeyPath:. These simply call the more primitive methods recursively.

There are a bunch of other KVC features for managing collections, but these are less interesting and I’m going to skip over them here.

CodeToday’s code is available on GitHub as part of the MAObject project:

https://github.com/mikeash/MAObject

Let’s get to it.

valueForKey

1
    - (id)valueForKey: (NSString *)key
2
    {
3
        SEL getterSEL = NSSelectorFromString(key);
4
        if([self respondsToSelector: getterSEL])
5
        {

If the object responds to that selector, it will use the accessor to get the value. Exactly how that’s done will depend on the accessor’s return type. To get ready, it fetches the return type and IMP for the method:

1
            NSMethodSignature *sig = [self methodSignatureForSelector: getterSEL];
2
            char type = [sig methodReturnType][0];
3
            IMP imp = [self methodForSelector: getterSEL];

If the return type is an object or a class, then the code is simple: cast the IMP to the right function pointer type, call it, and return what it returns:

1
            if(type == @encode(id)[0] || type == @encode(Class)[0])
2
            {
3
                return ((id (*)(id, SEL))imp)(self, getterSEL);
4
            }

Otherwise, the method returns a primitive, which is where things get interesting.

There is no convenient way to take a function pointer with an arbitrary type, call it, and box up the result. We have to do things the brute-force way, by enumerating all of the possibilities one by one and writing code to handle each possible type. I created a small macro to help with this:

1
            else
2
            {
3
                #define CASE(ctype, selectorpart) \
4
                    if(type == @encode(ctype)[0]) \
5
                        return [NSNumber numberWith ## selectorpart: ((ctype (*)(id, SEL))imp)(self, getterSEL)];

The idea is that each type gets a single line. You pass the type name as one parameter, and a selector part that fits in with [NSNumber numberWithType:] as the other parameter. The macro uses these to construct code that checks for the type and calls the IMP with the right function pointer type if it matches. With this macro, it’s just a matter of writing out every supported primitive type:

1
                CASE(char, Char);
2
                CASE(unsigned char, UnsignedChar);
3
                CASE(short, Short);
4
                CASE(unsigned short, UnsignedShort);
5
                CASE(int, Int);
6
                CASE(unsigned int, UnsignedInt);
7
                CASE(long, Long);
8
                CASE(unsigned long, UnsignedLong);
9
                CASE(long long, LongLong);
10
                CASE(unsigned long long, UnsignedLongLong);
11
                CASE(float, Float);
12
                CASE(double, Double);

Let’s not forget to undefine the CASE macro so we can reuse the name later:

1
                #undef CASE

If a matching case was found, then the method returned immediately. If the method is still running at this point, then the type isn’t known. Rather than try to handle this gracefully somehow, the method just throws an exception to complain:

1
                [NSException raise: NSInternalInconsistencyException format: @"Class %@ key %@ don't know how to interpret method return type from getter, signature is %@", [isa description], key, sig];
2
            }
3
        }

That was the code to handle the case where a getter method exists. If no getter exists, then KVC falls back to instance variables. First, it tries to get an instance variable with the same name as the key:

1
        Ivar ivar = class_getInstanceVariable(isa, [key UTF8String]);

If that fails, it tries again with a leading underscore:

1
        if(!ivar)
2
            ivar = class_getInstanceVariable(isa, [[@"_" stringByAppendingString: key] UTF8String]);

If either of those found an instance variable, it proceeds to actually fetching its value. In order to fetch the contents of the variable, we need to know where it’s stored. This is done by getting the variable’s offset, and adding it to the value of self:

1
        if(ivar)
2
        {
3
            ptrdiff_t offset = ivar_getOffset(ivar);
4
            char *ptr = (char *)self;
5
            ptr += offset;

self is cast to char * first, because the offset is in bytes, and operating on a char * ensures that the += operation does what we need.

We also need to know the type of the variable:

1
            const char *type = ivar_getTypeEncoding(ivar);

If the type is an object or class, then it just extracts the value directly and returns it:

1
            const char *type = ivar_getTypeEncoding(ivar);
2
            if(type[0] == @encode(id)[0] || type[0] == @encode(Class)[0])
3
            {
4
                return *(id *)ptr;
5
            }

Otherwise, it falls back to special cases again. This code uses a slightly different CASE macro. This one checks the type and then extracts the value from ptr if there’s a match:

1
            else
2
            {
3
                #define CASE(ctype, selectorpart) \
4
                    if(strcmp(type, @encode(ctype)) == 0) \
5
                        return [NSNumber numberWith ## selectorpart: *(ctype *)ptr];

Once again, there’s a long list of supported types:

1
                CASE(char, Char);
2
                CASE(unsigned char, UnsignedChar);
3
                CASE(short, Short);
4
                CASE(unsigned short, UnsignedShort);
5
                CASE(int, Int);
6
                CASE(unsigned int, UnsignedInt);
7
                CASE(long, Long);
8
                CASE(unsigned long, UnsignedLong);
9
                CASE(long long, LongLong);
10
                CASE(unsigned long long, UnsignedLongLong);
11
                CASE(float, Float);
12
                CASE(double, Double);

Followed by macro cleanup:

1
                #undef CASE

This code falls back to creating a generic NSValue with the contents of ptr if there’s no match. Because the data is already laid out in memory, it’s trivial to have a fallback here, rather than just throwing an exception like the getter code above does:

1
                return [NSValue valueWithBytes: ptr objCType: type];
2
            }
3
        }

Finally, if no getter or instance variable was found, the method throws an exception. The dummy return statement at the end is just to ensure that the compiler doesn’t complain about not returning a value:

1
        [NSException raise: NSInternalInconsistencyException format: @"Class %@ is not key-value compliant for key %@", [isa description], key];
2
        return nil;
3
    }

That takes care of valueForKey:.

setValue:forKey

The first thing it does is construct the name of the setter method to search for. valueForKey: can simply translate the key directly to a selector, but this method needs to do a bit of work. The setter method is generated by capitalizing the first letter of the key, then adding “set” to the beginning, and a colon at the end:

1
    - (void)setValue: (id)value forKey: (NSString *)key
2
    {
3
        NSString *setterName = [NSString stringWithFormat: @"set%@:", [key capitalizedString]];

It then turns that into a selector and checks to see if the object responds:

1
        SEL setterSEL = NSSelectorFromString(setterName);
2
        if([self respondsToSelector: setterSEL])
3
        {

If it does, it fetches the method’s argument type and IMP much like the getter code above:

1
            NSMethodSignature *sig = [self methodSignatureForSelector: setterSEL];
2
            char type = [sig getArgumentTypeAtIndex: 2][0];
3
            IMP imp = [self methodForSelector: setterSEL];

If the type is an object or class, it simply calls the setter, passing value, and returns:

1
            if(type == @encode(id)[0] || type == @encode(Class)[0])
2
            {
3
                ((void (*)(id, SEL, id))imp)(self, setterSEL, value);
4
                return;
5
            }

Otherwise, it’s once again time for a CASE macro. This one calls the IMP, passing [value typeValue] as the parameter, when a match is found:

1
            else
2
            {
3
                #define CASE(ctype, selectorpart) \
4
                    if(type == @encode(ctype)[0]) { \
5
                        ((void (*)(id, SEL, ctype))imp)(self, setterSEL, [value selectorpart ## Value]); \
6
                        return; \
7
                    }

Here is the big list of cases:

1
                CASE(char, char);
2
                CASE(unsigned char, unsignedChar);
3
                CASE(short, short);
4
                CASE(unsigned short, unsignedShort);
5
                CASE(int, int);
6
                CASE(unsigned int, unsignedInt);
7
                CASE(long, long);
8
                CASE(unsigned long, unsignedLong);
9
                CASE(long long, longLong);
10
                CASE(unsigned long long, unsignedLongLong);
11
                CASE(float, float);
12
                CASE(double, double);

Followed by macro cleanup:

1
                #undef CASE

Last, if the type is unknown, it throws an exception:

1
                [NSException raise: NSInternalInconsistencyException format: @"Class %@ key %@ set from incompatible object %@", [isa description], key, value];
2
            }
3
        }

If no setter method is found, then it searches for instance variables. No string manipulation is needed, since the instance variable’s name doesn’t change the way the setter’s name does. This code does the same check for instance variables with a leading underscore:

1
        Ivar ivar = class_getInstanceVariable(isa, [key UTF8String]);
2
        if(!ivar)
3
            ivar = class_getInstanceVariable(isa, [[@"_" stringByAppendingString: key] UTF8String]);

If the instance variable exists, it creates a pointer to it and gets its type just like valueForKey: does:

1
        if(ivar)
2
        {
3
            ptrdiff_t offset = ivar_getOffset(ivar);
4
            char *ptr = (char *)self;
5
            ptr += offset;
6

7
            const char *type = ivar_getTypeEncoding(ivar);

If the variable is an object or class pointer, the code can set it directly. Well, nearly directly. There’s a minor retain release dance to be done in order to ensure that memory management is correct:

1
            if(type[0] == @encode(id)[0] || type[0] == @encode(Class)[0])
2
            {
3
                value = [value retain];
4
                [*(id *)ptr release];
5
                *(id *)ptr = value;
6
                return;
7
            }

Otherwise, value is boxed, and the primitive value needs to be extracted. If value is an NSValue with the exact same type as the instance variable, the getValue: method can be used to simply copy the value over directly:

1
            else if(strcmp([value objCType], type) == 0)
2
            {
3
                [value getValue: ptr];
4
                return;
5
            }

If that doesn’t work, it’s time to fall back to the last long list of cases. This version of the CASE macro sets the value at ptr, appropriately cast, to [value typeValue]:

1
            else
2
            {
3
                #define CASE(ctype, selectorpart) \
4
                    if(strcmp(type, @encode(ctype)) == 0) { \
5
                        *(ctype *)ptr = [value selectorpart ## Value]; \
6
                        return; \
7
                    }

The traditional exhaustive enumeration of primitive types follows:

1
                CASE(char, char);
2
                CASE(unsigned char, unsignedChar);
3
                CASE(short, short);
4
                CASE(unsigned short, unsignedShort);
5
                CASE(int, int);
6
                CASE(unsigned int, unsignedInt);
7
                CASE(long, long);
8
                CASE(unsigned long, unsignedLong);
9
                CASE(long long, longLong);
10
                CASE(unsigned long long, unsignedLongLong);
11
                CASE(float, float);
12
                CASE(double, double);

Macro cleanup:

1
                #undef CASE

Finally, if none of the cases were hit, throw an exception:

1
                [NSException raise: NSInternalInconsistencyException format: @"Class %@ key %@ set from incompatible object %@", [isa description], key, value];
2
            }
3
        }

If neither setter method nor instance variable was found, throw an exception to complain:

1
        [NSException raise: NSInternalInconsistencyException format: @"Class %@ is not key-value compliant for key %@", [isa description], key];
2
    }

Key PathsTo round out the implementation of KVC, I’ll implement valueForKeyPath: and setValue:forKeyPath: as well.

The first thing that valueForKeyPath: does is look for a . in the key path. If it doesn’t exist, then it’s treated as a plain key and passed to valueForKey:

1
    - (id)valueForKeyPath: (NSString *)keyPath
2
    {
3
        NSRange range = [keyPath rangeOfString: @"."];
4
        if(range.location == NSNotFound)
5
            return [self valueForKey: keyPath];

Otherwise, the key is split into two pieces. The piece up to the . is the local key, and the following piece is the remainder of the key path:

1
        NSString *key = [keyPath substringToIndex: range.location];
2
        NSString *rest = [keyPath substringFromIndex: NSMaxRange(range)];

The key is passed to valueForKey:

1
        id next = [self valueForKey: key];

Then valueForKeyPath: is sent recursively to the next object:

1
        return [next valueForKeyPath: rest];
2
    }

Its implementation will decompose rest further until every . is consumed. The result is a chain of valueForKey: calls, returning the result of the very last call.

setValue:forKeyPath: works similarly. If there’s no . in the key path, call setValue:forKey: and return:

1
    - (void)setValue: (id)value forKeyPath: (NSString *)keyPath
2
    {
3
        NSRange range = [keyPath rangeOfString: @"."];
4
        if(range.location == NSNotFound)
5
        {
6
            [self setValue: value forKey: keyPath];
7
            return;
8
        }

Otherwise, extract the key and remainder:

1
        NSString *key = [keyPath substringToIndex: range.location];
2
        NSString *rest = [keyPath substringFromIndex: NSMaxRange(range)];

Grab the next object using valueForKey:

1
        id next = [self valueForKey: key];

Then recursively send setValue:forKeyPath: to next, passing rest as the key path:

1
        [next setValue: value forKeyPath: rest];
2
    }

The result is a chain of valueForKey: calls, culminating in a call to setValue:forKey: on the last object in the chain.

ConclusionYou can now see how key-value coding works on the inside. There isn’t anything particularly complicated. It’s largely just a long list of different things to try. Cocoa’s implementation is a bit smarter, and can leverage things like NSInvocation for more comprehensive coverage, but that’s the basic idea. A large part of NSInvocation is simply baked-in knowledge of all the different cases that need to be handled as well.

That’s it for today. May you code your keys and values in peace. Until next time, since Friday Q&A is driven by reader suggestions, please send in your ideas for topics!