Objective-c單例模式的正確寫法--用dispatch 線程安全

單例模式在iOS開發中可能算是最常用的模式之一了，但是由於oc本身的語言特性，想要寫一個正確的單例模式相對來說比較麻煩，這里我就拋磚引玉來聊一聊iOS中單例模式的設計思路。關於單例模式更多的介紹請參考這篇文章。

單例顧名思義就是說一個類的實例只能有一個，在java、C++這類語言中，可以通過將構造函數私有化來避免對象的重復創建，但是objective-c卻不能夠這樣做，我們需要通過其他機制來達到這個目的。一般情況下，可能我們寫的單例模式是這樣的：

 

#import 

@interface Singleton : NSObject

+(instancetype) shareInstance ;

@end



#import "Singleton.h"

@implementation Singleton

static Singleton* _instance = nil;

+(instancetype) shareInstance
{
    static dispatch_once_t onceToken ;
    dispatch_once(&onceToken, ^{
        _instance = [[self alloc] init] ;
    }) ;
    
    return _instance ;
}

@end

具體使用:

 

 

#import 
#import "Singleton.h"

int main(int argc, const char * argv[]) {
    @autoreleasepool {
        
        Singleton* obj1 = [Singleton shareInstance] ;
        NSLog(@"obj1 = %@.", obj1) ;
        
        Singleton* obj2 = [Singleton shareInstance] ;
        NSLog(@"obj2 = %@.", obj2) ;
        
        //
        Singleton* obj3 = [[Singleton alloc] init] ;
        NSLog(@"obj3 = %@.", obj3) ;
    }
    return 0;
}

輸出結果為 :

 

 

2014-12-15 16:06:28.344 ObjcSingleton[8847:303] obj1 = .
2014-12-15 16:06:28.346 ObjcSingleton[8847:303] obj2 = .
2014-12-15 16:06:28.346 ObjcSingleton[8847:303] obj3 = .

可以看到，當我們調用shareInstance方法時獲取到的對象是相同的，但是當我們通過alloc和init來構造對象的時候，得到的對象卻是不一樣的。

 

那么問題就來了，我們通過不同的途徑得到不同的對象，顯然是不行的。我們必須要確保對象的唯一性，所以我們就需要封鎖用戶通過alloc和init以及copy來構造對象這條道路。

我們知道，創建對象的步驟分為申請內存(alloc)、初始化(init)這兩個步驟，我們要確保對象的唯一性，因此在第一步這個階段我們就要攔截它。當我們調用alloc方法時，oc內部會調用allocWithZone這個方法來申請內存，我們覆寫這個方法，然后在這個方法中調用shareInstance方法返回單例對象，這樣就可以達到我們的目的。拷貝對象也是同樣的原理，覆寫copyWithZone方法，然后在這個方法中調用shareInstance方法返回單例對象。看代碼吧：

 

#import "Singleton.h"

@implementation Singleton

static Singleton* _instance = nil;

+(instancetype) shareInstance
{
    static dispatch_once_t onceToken ;
    dispatch_once(&onceToken, ^{
        _instance = [[super allocWithZone:NULL] init] ;
    }) ;
    
    return _instance ;
}

+(id) allocWithZone:(struct _NSZone *)zone
{
    return [Singleton shareInstance] ;
}

-(id) copyWithZone:(struct _NSZone *)zone
{
    return [Singleton shareInstance] ;
}

@end

再看看效果如何:

 

 

main : 

#import 
#import "Singleton.h"

int main(int argc, const char * argv[]) {
    @autoreleasepool {
        
        Singleton* obj1 = [Singleton shareInstance] ;
        NSLog(@"obj1 = %@.", obj1) ;
        
        Singleton* obj2 = [Singleton shareInstance] ;
        NSLog(@"obj2 = %@.", obj2) ;
        
        //
        Singleton* obj3 = [[Singleton alloc] init] ;
        NSLog(@"obj3 = %@.", obj3) ;
        
        Singleton* obj4 = [[Singleton alloc] init] ;
        NSLog(@"obj4 = %@.", [obj4 copy]) ;
    }
    return 0;
}

輸出結果：

 

 

2014-12-15 16:11:24.734 ObjcSingleton[8979:303] obj1 = .
2014-12-15 16:11:24.735 ObjcSingleton[8979:303] obj2 = .
2014-12-15 16:11:24.736 ObjcSingleton[8979:303] obj3 = .
2014-12-15 16:11:24.736 ObjcSingleton[8979:303] obj4 = .

可以看到獲取到的對象都是一樣的了。

 

思路就是這樣，如果還有更嚴謹的寫法請留言告知。謝謝~