聊聊 Go 中的单例模式

// 饿汉式单例模式
package main

type singleton struct {
    count int
}

// 饿汉式单例，程序启动即初始化
var Instance = new(singleton)

// Add 方法用于累加并返回计数值
func (s *singleton) Add() int {
    s.count++
    return s.count
}

// 懒汉式单例模式
package main

import (
	"sync"
)

type singleton struct {
	count int
}

var (
	instance *singleton
	mutex    sync.Mutex
)

// New 实例化一个对象
// 这里采用了【双重检查】
// 假设 goroutine X 和 Y 几乎同时调用 New 函数
// 当它们同时进入此函数时，instance 变量值是 nil 因此 goroutine X 和 Y 会同时到达【位置1】
// 假设 goroutine X 会先到达【位置2】，并进入 mutex.Lock() 到达【位置3】，这时，由于 mutex.Lock() 的同步限制
// goroutine Y 无法到达【位置3】只能在【位置2】等候
// goroutine X 执行 instance = new(singleton) 语句，使得 instance 变量得到一个值，此时 goroutine Y 还是只能在【位置2】等候
// goroutine X 释放锁，返回 instance 变量，退出 New 函数
// goroutine Y 进入 mutex.Lock() 到达【位置3】，进而到达【位置4】。由于此时 instance 变量已经不是 nil，因此 goroutine Y 释放锁
// 可见，锁仅用来避免多个 goroutine 同时实例化 singleton
func New() *singleton {
	if instance == nil { // 【位置1】
		// 这里可能有多于一个 goroutine 同时到达 【位置2】
		mutex.Lock()
		// 这里每个时刻只会有一个 goroutine 到达  【位置3】
		if instance == nil { // 【位置4】
			instance = new(singleton)
		}
		mutex.Unlock()
	}

	return instance
}

func (s *singleton) Add() int {
	s.count++
	return s.count
}