Swift百万线程攻破单例（Singleton)模式

一、不安全的单例实现

在上一篇文章我们给出了单例的设计模式，直接给出了线程安全的实现方法。单例的实现有多种方法，如下面：

[javascript] view plaincopy

class SwiftSingleton {

class var shared: SwiftSingleton {

if !Inner.instance {

Inner.instance = SwiftSingleton()

}

return Inner.instance!

}

struct Inner {

static var instance: SwiftSingleton?

}

}

这段代码的实现，在shared中进行条件判断，如果Inner.instance.为空就生成一个实例，这段代码很简单看出当线程同时访问SwiftSingleton.shared方法时，会有如下问题出现，线程A判断Inner.instance为空，进入if语句后立即切换到线程B执行，线程B也进行判断，由于线程A只是进入了if语句，这行代码

[javascript] view plaincopy

Inner.instance = SwiftSingleton()

并没有执行，这时Inner.instance还是为空，纯种B也进行了if语句，这种情况下就会创建多个实例，没有保证实例的唯一性。上面的理论分析基本上任何一篇文章都会讲的，也不能理解，关键问题，如何测试上面的理论是否正确呢？

二、线程抢占原理

其实要实现上面的例子不是很难，创建N个线程，让他同时访问SwiftSingleton.shared的方法，然后将所返回值保存最后比较引用。原理很正确，但是创建线程的过程也是极为耗时的，现在的电脑执行速度又非常快，模拟具有不稳定性。如何才能最大的程序测试上面的安全性呢？这里我们可以考虑一个现实的问题，假设找1000人通过一段100米的赛道，我们想要更多的人同时去冲刺终点，越多越好。如果你找一个人，告诉他去跑100米，然后再找一下，这种肯定同时到达终点的几率很底。怎么办才能让更多的人在同一时刻到达终点呢？问题很简单，让这1000人有一个同一起跑点，让他们都准备好了，随着一声令下，一起奔跑。回到技术问题，我们想要更多的线程访问SwiftSingleton.shared方法，只要先准备好所有的线程，然后发一个信号，让他们同时去访问这个方法就可以了。



实现代码如下：

[javascript] view plaincopy

class SwiftSingletonTest: XCTestCase {

let condition = NSCondition()

let mainCondition = NSCondition()

let singleton: NSMutableArray = NSMutableArray()

let threadNumbers = 1000

var count = 0

func testSingletonThreadSafe() {

for index in 0...threadNumbers {

NSThread.detachNewThreadSelector("startNewThread", toTarget: self, withObject: nil)

}

condition.broadcast()

mainCondition.lock()

mainCondition.wait()

mainCondition.unlock()

checkOnlyOne()

}

func startNewThread() {

condition.lock()

condition.wait()

condition.unlock()

let temp = SwiftSingleton.shared

count++

singleton.addObject(temp)

if count >= threadNumbers {

mainCondition.signal()

}

}

func checkOnlyOne () {

let one = singleton[0] as SwiftSingleton

for temp : AnyObject in singleton {

let newTemp = temp as SwiftSingleton

if(newTemp !== one) {

XCTFail("singleton error!");

break;

}

}

}

}

这段代码主要使用了NSCondition进行同步，其中NSCondition分为两组，condition主要负责除主线程外的线程，在for语句中会创建并启动N（threadNumbers）个线程，每个线程启动后都会去执行startNewThread方法，执行到语句

[javascript] view plaincopy

condition.wait()

会挂起当前线程，当所有线程都创建并启动完时，主线程会执行

[javascript] view plaincopy

condition.broadcast()

来通知挂起的N个线程继承执行，此时主线程调了

[javascript] view plaincopy

mainCondition.wait()

主线和进入持起状态，此处将主线程挂起是为了在所有线程执行完，依次检查取得引用的唯一性。

[javascript] view plaincopy

if count >= threadNumbers {

mainCondition.signal()

}

当所有线程执行完时，通知主线程开始检查引用 ，执行结果如下：



从上面执行结果可以看出，这种单例并不能保证唯一性。上面用到了NSMutableArray类，网上说是线程不安全的，这里用的Swift语言，这么多线程一起操作暂没有发现异常......

三、其它实现测试结果

1、最简单实现

[javascript] view plaincopy

class SwiftSingleton {

class var shared: SwiftSingleton {

return Inner.instance

}

struct Inner {

static let instance: SwiftSingleton = SwiftSingleton()

}

}

解释：上述代表也实现了延迟加载技术

[javascript] view plaincopy

static let instance: SwiftSingleton = SwiftSingleton()

首次访问Inner.instance时才会创建SwiftSingleton,此处的延迟加载由Swift语言原生提供

测试结果：通过

2、使用GCD技术实现的单例模式

[javascript] view plaincopy

class SwiftSingleton {

class var shared: SwiftSingleton {

dispatch_once(&Inner.token) {

Inner.instance = SwiftSingleton()

}

return Inner.instance!

}

struct Inner {

static var instance: SwiftSingleton?

static var token: dispatch_once_t = 0

}

}

测试结果：通过

四、测试说明

1、Mac OS线程总量有限制，你可以创建线程，但是最大线程启动数为2048（我的电脑是这样，不清楚是否跟硬件有关）
2、如果遇到测试无响应时，可以尝试重启电脑