Golang 设计模式之单例模式

最新推荐文章于 2024-09-15 16:54:15 发布

UUID_UI

最新推荐文章于 2024-09-15 16:54:15 发布

阅读量459

点赞数 5

文章标签：后端

本文链接：https://blog.csdn.net/a746277441/article/details/141931807

版权

简介

单例模式（Singleton Pattern） 是一种创建型设计模式，其目的是确保一个类只有一个实例，并提供一个全局访问点来获取该实例。单例模式在应用程序中是很常见的，特别是当你需要一个全局的唯一实例来协调系统中的一些共享资源时。

应用场景

单例模式适用于以下场景：

全局共享资源：当你需要一个全局唯一的实例来协调系统中的共享资源，如数据库连接池、配置管理器等。
限制实例化：当你需要确保某个类在整个应用程序中只有一个实例时，如应用程序的日志记录器。
节省资源：当对象创建开销较大或资源有限时，通过单例模式可以避免重复创建对象，从而节省资源。

例如，配置管理器、日志记录器和线程池都是常见的单例模式应用场景。

示例代码

下面的示例展示了如何在 Go 语言中实现单例模式。我们将创建一个 Logger 结构体，它负责记录日志，并确保只有一个 Logger 实例存在。

代码实现

package main

import (
    "fmt"
    "sync"
)

// Logger 是单例对象
type Logger struct {
    LogLevel string
}

var (
    instance *Logger
    once     sync.Once
)

// GetInstance 返回 Logger 的唯一实例
func GetInstance() *Logger {
    once.Do(func() {
        instance = &Logger{LogLevel: "INFO"}
    })
    return instance
}

// Log 方法用于记录日志
func (l *Logger) Log(message string) {
    fmt.Printf("[%s] %s\n", l.LogLevel, message)
}

func main() {
    // 获取 Logger 实例并记录日志
    logger1 := GetInstance()
    logger1.Log("Application started")

    // 获取另一个 Logger 实例并记录日志
    logger2 := GetInstance()
    logger2.Log("Application finished")

    // 验证两个 Logger 实例是否相同
    if logger1 == logger2 {
        fmt.Println("logger1 和 logger2 是同一个实例")
    }
}

代码解释

Logger 结构体：表示日志记录器对象，包含一个 LogLevel 字段。
instance 和 once：instance 是 Logger 的唯一实例，once 是 sync.Once 类型，用于确保初始化代码只执行一次。
GetInstance 函数：通过 once.Do 方法保证 Logger 实例只被创建一次，并返回该实例。
Log 方法：记录日志消息。
main 函数：演示如何获取单例实例并记录日志，同时验证两个 Logger 实例是否相同。

单元测试

以下是测试 Logger 单例模式实现的测试代码，确保 GetInstance 方法返回相同的实例。

package main

import (
    "testing"
)

func TestSingleton(t *testing.T) {
    logger1 := GetInstance()
    logger2 := GetInstance()

    if logger1 != logger2 {
        t.Errorf("Expected both instances to be the same, but got different instances")
    }
}

func TestLoggerLog(t *testing.T) {
    logger := GetInstance()
    message := "Test message"

    // Capture the output
    original := fmt.Sprintf("%s", message)
    logger.Log(message)
    output := fmt.Sprintf("%s", message)

    if output != original {
        t.Errorf("Expected output to be %s, but got %s", original, output)
    }
}