Go 中使用单例模式

[kevinyan815]

单例模式，分两种加载模式，下面介绍下 Go 里面怎么使用单例模式。

饿汉模式

适合程序初始化时就确定怎么加载实例的情况

package singleton

// Singleton 饿汉式单例
type Singleton struct{}

var singleton *Singleton

func init() {
    singleton = &Singleton{}
}

// GetInstance 获取实例
func GetInstance() *Singleton {
    return singleton
}

懒汉模式

延迟加载，适合根据传入参数实现定制化的初始化操作。

package singleton

import (
    "sync"
    "sync/atomic"
)

type singleton struct {}

var (
    initialized uint32
    instance *singleton
    mu sync.Mutex
)

func GetInstance() *singleton {

    if atomic.LoadUint32(&initialized) == 1 { // 原子操作
        return instance
    }

    mu.Lock()
    defer mu.Unlock()

    if initialized == 0 {
        instance = &singleton{}
        atomic.StoreUint32(&initialized, 1)
    }

    return instance
}

上面通过了atomic 和 mutex 进行双重检查，保证线程安全，不过可以利用 sync.Once 实现更简洁的版本，这一版最好也记住，有可能面试的时候就不让你用sync.Once实现，sync.Once 内部也是维护了个标志为，用atomic 判断它是否已经执行过函数。

Java 里的线程安全的单例也是这么通过volatile+synchronized 双锁检查实现的。

简洁版的懒汉模式单例

package singleton

type singleton struct {}

var instance *singleton
var once sync.Once

func GetInstance() *singleton {
    once.Do(func() {
        instance = &singleton{}
    })
    return instance
}

这里直接上的代码，没有什么废话，如果想看故事版的，可以访问最简单的单例模式，Go版本的实现你写对了吗？