Go语言中如何优雅实现单例模式

单例模式的基本定义

单例模式(Singleton Pattern)是一种设计模式,旨在保证一个类只有一个实例,并且提供全局访问点。单例模式通常用于需要限制某个对象的实例数量为一个的场景,比如数据库连接池、日志管理器、配置管理器等。


Go语言单例模式的实现

1. 线程安全的懒汉式单例

懒汉式的实现会延迟实例的创建,直到第一次调用时才会实例化对象。为了保证并发情况下的安全性,我们需要使用 sync.Once 来确保实例只会创建一次。

go 复制代码
package main

import (
	"fmt"
	"sync"
)

var wg sync.WaitGroup

// Singleton 类型
type Singleton struct {
}

var instance *Singleton
var once sync.Once

// GetInstance 提供全局唯一的实例
func GetInstance() *Singleton {

	once.Do(func() {
		instance = &Singleton{}
	})
	return instance
}

func main() {
	// 获取单例实例
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func(index int) {
			defer wg.Done()
			s1 := GetInstance()
			fmt.Printf("index %d, memery address: %p\n", index, s1)
		}(i)

	}
	wg.Wait()
}

结果

diff 复制代码
index 0, memery address: 0x56c480
index 5, memery address: 0x56c480
index 4, memery address: 0x56c480
index 2, memery address: 0x56c480
index 7, memery address: 0x56c480
index 9, memery address: 0x56c480
index 6, memery address: 0x56c480
index 8, memery address: 0x56c480
index 3, memery address: 0x56c480
index 1, memery address: 0x56c480

解析:

  • sync.Once:Go标准库提供的一个同步原语,确保某个函数只会被调用一次。它在并发情况下保证了线程安全。
  • once.Do:此方法确保传入的函数只执行一次,适用于懒加载单例实例。

2. 双重检查锁定(DCL)

双重检查锁定是一种优化方式,它通过在两次检查实例时,减少了加锁的开销,提高了性能。

go 复制代码
package main

import (
	"fmt"
	"sync"
)

var wg sync.WaitGroup

// Singleton 类型
type Singleton struct {
}

var instance *Singleton
var lock sync.Mutex

func GetInstance() *Singleton {
	if instance == nil {
		lock.Lock()
		defer lock.Unlock()
		if instance == nil {
			instance = &Singleton{}
		}
	}
	return instance
}

func main() {
	// 获取单例实例
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func(index int) {
			defer wg.Done()
			s1 := GetInstance()
			fmt.Printf("index %d, memery address: %p\n", index, s1)
		}(i)

	}
	wg.Wait()
}

解析:

  • 双重检查 :第一次检查实例是否为 nil,如果是,则加锁。然后再次检查实例是否为 nil,如果是则创建实例。
  • 锁的优化:只有在实例尚未创建时才会加锁,避免了每次获取实例时都需要加锁的性能损耗。

3. 原子操作法

Go 语言的 sync/atomic 包提供了原子操作,我们可以利用它来确保单例的线程安全。

go 复制代码
package main

import (
	"fmt"
	"sync"
	"sync/atomic"
	"unsafe"
)

var wg sync.WaitGroup

type Singleton struct {
}

var instance unsafe.Pointer

func GetInstance() *Singleton {
	// 使用原子操作获取实例
	if atomic.LoadPointer(&instance) == nil {
		newInstance := &Singleton{}
		atomic.StorePointer(&instance, unsafe.Pointer(newInstance))
	}
	return (*Singleton)(atomic.LoadPointer(&instance))
}

func main() {
	// 获取单例实例
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func(index int) {
			defer wg.Done()
			s1 := GetInstance()
			fmt.Printf("index %d, memery address: %p\n", index, s1)
		}(i)

	}
	wg.Wait()
}

解析:

  • unsafe.Pointer :在Go中,unsafe.Pointer 可以用来绕过类型系统,直接处理内存地址。通过原子操作确保实例赋值的安全性。
  • atomic.LoadPointeratomic.StorePointer:原子加载和存储指针,确保操作的线程安全性。

总结

在Go语言中实现单例模式有多种方式,最常见的是使用 sync.Once、双重检查锁定(DCL)和原子操作法。每种方法有其优缺点,选择合适的方式可以帮助你在保证线程安全的前提下优化性能。

感谢阅读!如果你觉得这篇文章对你有帮助,请分享给你的朋友们,让更多的人一起学习Go语言!

相关推荐
Asthenia04121 分钟前
ArrayList与CopyOnWriteArrayList源码深度解析及面试拷打
后端
Asthenia04121 分钟前
深入解析String、StringBuilder、StringBuffer与final修饰对象的问题
后端
Asthenia04125 分钟前
Java数据类型的四类八种与拆装箱底层原理
后端
郭萌69633 分钟前
Docker 安装陀螺匠教程
后端
u01037310642 分钟前
Django REST Framework (DRF)
后端·python·django
牛马喜喜1 小时前
sequelize的进阶使用(助力成为优秀全栈)
后端·node.js
勇敢牛牛_1 小时前
【Rust基础】使用Rocket构建基于SSE的流式回复
开发语言·后端·rust
Anarkh_Lee1 小时前
解决 Spring Boot 多数据源环境下事务管理器冲突问题(非Neo4j请求标记了 @Transactional 尝试启动Neo4j的事务管理器)
spring boot·后端·spring
Serverless社区1 小时前
从零开始开发 MCP Server
go
Yharim1 小时前
微服务项目相同nacos地址导致的问题
后端·微服务