众所周知,一般的情况下我们是没办法在category里加属性的.
如果想加,需要用到Associated.
@interface NSObject (XYFlyweightTransmit)
@property (nonatomic, strong) id uxy_flyweightData;
@end
@implementation NSObject (UXYFlyweightTransmit)
- (id)uxy_flyweightData
{
return objc_getAssociatedObject(self, NSObject_key_flyweightData);
}
- (void)setUxy_flyweightData:(id)flyweightData
{
objc_setAssociatedObject(self, NSObject_key_flyweightData, flyweightData, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
@end;
如果每次需要写这么多代码才可以实现, 确实很繁琐.
这事可以按照下面的代码封装一下
@interface NSObject (XY_associated)
- (id)uxy_getAssociatedObjectForKey:(const char *)key;
- (id)uxy_retainAssociatedObject:(id)obj forKey:(const char *)key;
- @end
@implementation NSObject (XY_associated)
- (id)uxy_getAssociatedObjectForKey:(const char *)key
{
const char * propName = key;
id currValue = objc_getAssociatedObject( self, propName );
return currValue;
}
- (id)uxy_retainAssociatedObject:(id)obj forKey:(const char *)key;
{
const char * propName = key;
id oldValue = objc_getAssociatedObject( self, propName );
objc_setAssociatedObject( self, propName, obj, OBJC_ASSOCIATION_RETAIN_NONATOMIC );
return oldValue;
}
这样,我们只需要按照如下方式添加一个属性
- (UIView *)overlay
{
return [self uxy_getAssociatedObjectForKey:"xy.navigationBar.overlay"];
}
- (void)setOverlay:(UIView *)overlay
{
[self uxy_retainAssociatedObject:overlay forKey:"xy.navigationBar.overlay"];
}
上面的代码看起来还是有不少重复的地方.可以尝试用一个宏来封装一下.
但是当我们真正写宏的时候发现get方法好写, set方法无从下手.因为后面是跟的一个大写的字母.
不漂亮的解决方法:
* 约定属性用大写字母开头
* 约定下划线开头
* 写宏的时候吧setName名字传进去
不过,用的的当然不是上述的方法.
先来看下代码表现起来是怎样的
@interface UINavigationBar (XY)
uxy_property_strong(id, test2);
@end
@implementation UINavigationBar (XY)
uxy_def_property_strong(id, test2)
@end
{
self.test2 = @"a";
id c = self.test2;
NSLog(@"%@", c);
}
实现思路:
在申明属性的时候用setter来修改属性的set方法,在前面加 __ 避开大小写.
具体实现代码如下:
#define uxy_property_strong( __type, __name)
@property (nonatomic, strong, setter=set__##__name:, getter=__##__name) __type __name;
#define uxy_def_property_strong( __type, __name)
- (__type)__##__name
{ return [self uxy_getAssociatedObjectForKey:#__name]; }
- (void)set__##__name:(id)__##__name
{ [self uxy_retainAssociatedObject:__##__name forKey:#__name]; }
我们再用元宏metamacro_concat给宏升级一下,让他支持基本类型, 然后取消property前面的@符号, 就有了加属性宏的最终版本uxy_property_basicDataType( __type, __name)
演示代码:
@interface Associated (test)
@uxy_property_basicDataType(int, age);
@uxy_property_basicDataType(NSTimeInterval, time);
@end
@implementation Associated (test)
uxy_def_property_basicDataType(int, age)
uxy_def_property_basicDataType(NSTimeInterval, time)
@end
// 测试代码
UXY_DESCRIBE( test1 )
{
Associated *associated = [[Associated alloc] init];
associated.age = 10;
associated.time = 100.5f;
UXY_EXPECTED( associated.age == 10 );
UXY_EXPECTED( associated.time == 100.5f );
}
具体实现代码如下:
/**
* Returns A and B concatenated after full macro expansion.
*/
#define metamacro_concat(A, B)
metamacro_concat_(A, B)
#define metamacro_concat_(A, B) A ## B
#define uxy_property_basicDataType( __type, __name)
property (nonatomic, assign, setter=set__##__name:, getter=__##__name) __type __name;
#define uxy_def_property_basicDataType( __type, __name)
- (__type)__##__name
{
NSNumber *number = [self uxy_getAssociatedObjectForKey:#__name];
return metamacro_concat(metamacro_concat(__uxy_, __type), _value)( number );
}
- (void)set__##__name:(__type)__##__name
{
id value = @(__##__name);
[self uxy_setAssignAssociatedObject:value forKey:#__name];
}
#define __uxy_int_value( __nubmer ) [__nubmer intValue]
#define __uxy_BOOL_value( __nubmer ) [__nubmer boolValue]
#define __uxy_NSTimeInterval_value( __nubmer ) [__nubmer doubleValue]