開發(fā)中, 單例模式經(jīng)常會用到, 代碼幾乎都是一致的.
+(instancetype)shared {
static SomeClass* sharedInstance;
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
sharedInstance = [SomeClass new];
});
return sharedInstance;
}
聰明的同學可能會加入到snippet 中, 快速的構建一個單例.
不過有沒有什么更好的方式呢?
當然有. 用黑魔法.
先看看最終效果
singleton(SomeClass, shared)
@interface SomeClass: NSObject
+(instancetype)shared;
@end
@implementation SomeClass
-(NSString *)name {
return @"some class -- name";
}
@end;
int main(int argc, const char * argv[]) {
@autoreleasepool {
NSLog([[SomeClass shared] name]);
}
return 0;
}
不需要寫單列的實現(xiàn), 只需要寫一個單例的函數(shù)的聲明, 甚至這個函數(shù)聲明也不需要, 不過沒有的話, 調用起來不方便, 例如
singleton(SomeAnotherClass, shared)
@interface SomeAnotherClass: NSObject
@end
@implementation SomeAnotherClass
-(NSString *)name {
return @"some other class -- name";
}
@end;
int main(int argc, const char * argv[]) {
@autoreleasepool {
NSLog([[SomeAnotherClass performSelector:NSSelectorFromString(@"shared")] name]);
}
return 0;
}
你可能已經(jīng)注意到了, 有一個奇怪的東西singleton(SomeClass, shared).
對的, 和你猜想的一樣, 這個宏, 對的, 是一個宏, 第一個參數(shù)是類名, 第二個參數(shù)是單例的方法.
具體實現(xiàn)是這樣的
#define __toString(x) #x
#define singleton(class, method) __attribute__((objc_runtime_name(__toString(SGSINGLETON_##class##_##method))))
看不懂沒關系, 這相當于在類上面寫了這樣一個東西
__attribute__((objc_runtime_name("SGSINGLETON_SomeClass_shared")))
@interface SomeClass: NSObject
...
作用很簡單, 運行的時候, 會將類名改為 SGSINGLETON_SomeClass_shared
, 你可以參考這篇文章看看具體細節(jié)
http://blog.sunnyxx.com/2016/05/14/clang-attributes/.
在這里的作用是給類加上一個標記, 然后我們在運行時掃描所有加載的類, 找到這個標記的類后, 加上一個方法, 這個方法就是我們單例的實現(xiàn). 是不是很簡單.
接下來看看具體的步驟
__attribute__((constructor)) void addSingleton(){
// 獲取所有的類
int classCount = objc_getClassList(NULL, 0);
Class *classList = new Class[classCount];
objc_getClassList(classList, classCount);
Class singtonClassList[256];
int singtonClassListIndex = 0;
for(int i =0; i< classCount; i++) {
NSString *className = NSStringFromClass(classList[i]);
// 找到標記過的類
if([className hasPrefix:@"SGSINGLETON_"]) {
singtonClassList[singtonClassListIndex++] = classList[i];
}
}
for(int i = 0; i< singtonClassListIndex; i++) {
Class cls = singtonClassList[i];
NSString *clsName = NSStringFromClass(cls);
// 獲取元類
Class metaClass = objc_getMetaClass(clsName.UTF8String);
// 獲取方法名
NSString *sharedMethodName = [clsName componentsSeparatedByString:@"_"].lastObject;
SEL sel = NSSelectorFromString(sharedMethodName);
// 添加方法
BOOL result = class_addMethod(metaClass, sel, IMP(makeSharedInstance), "@@:");
if(!result) {
NSLog(@"f**k");
}
}
delete [] classList;
}
接下來是單例的實現(xiàn)方法, 這個就更加簡單了.
id makeSharedInstance(id obj, SEL sel) {
Class cls = obj;
// 保存所有單例的字典
static NSMutableDictionary *instanceMap;
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
instanceMap = [NSMutableDictionary new];
});
NSString *className = NSStringFromClass(obj);
// 多線程鎖
static NSLock *lock = [NSLock new];
[lock lock];
// 獲取單例
id instance = [instanceMap objectForKey:className];
if(! instance) {
// 沒有初始化則進行初始化
instance = [[cls alloc] init];
instanceMap[className] = instance;
NSLog(@"make instance for %@", className);
}
[lock unlock];
// 返回對應單例
return instance;
}
創(chuàng)意來自 Sunny 的博客文章--Clang Attributes 黑魔法小記, 覺得很有意思, 勉強實現(xiàn)了一下, 代碼寫的很垃圾, 歡迎各位大俠留言改進.
