go设计模式——单例模式

news2024/11/28 4:39:20

概念

单例是一种创建型设计模式，它确保一个类在整个程序运行期间只有一个实例，并提供一个全局访问点来使用该实例。虽然单例模式在某些情况下非常有用，例如管理全局配置、日志记录或资源共享，但它也带来了与全局变量相似的问题。具体来说，单例模式可能导致代码的模块化和可测试性下降，因为它在整个系统中引入了隐式依赖，使得代码难以解耦和扩展。因此，尽管单例模式可以简化某些设计，但在使用时需要谨慎，以避免对代码的灵活性和可维护性产生不良影响。

demo1

单例模式grpc的连接

package client

import (
	"log"
	"sync"
	"google.golang.org/grpc"
	"your/proto/package/path/example" // 更新为你的 proto 包路径
)

type GRPCClient struct {
	conn   *grpc.ClientConn
	client example.ExampleServiceClient
	mu     sync.Mutex
}

var instance *GRPCClient
var once sync.Once

// GetGRPCClient 返回 ExampleServiceClient 的单例实例
func GetGRPCClient() *GRPCClient {
	once.Do(func() {
		instance = &GRPCClient{}
	})

	// 获取锁来保护连接状态检查
	instance.mu.Lock()
	defer instance.mu.Unlock()

	// 如果连接为空或已关闭，则重新建立连接
	if instance.conn == nil || instance.conn.GetState() == grpc.Shutdown {
		conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())
		if err != nil {
			log.Fatalf("Failed to connect to gRPC server: %v", err)
		}
		instance.conn = conn
		instance.client = example.NewExampleServiceClient(conn)
	}

	return instance
}

// GetClient 返回 gRPC 客户端
func (g *GRPCClient) GetClient() example.ExampleServiceClient {
	return g.client
}

// CloseConnection 关闭 gRPC 连接
func (g *GRPCClient) CloseConnection() error {
	g.mu.Lock()
	defer g.mu.Unlock()

	if g.conn != nil {
		err := g.conn.Close()
		g.conn = nil // 重置连接，确保下一次调用时可以重新连接
		return err
	}
	return nil
}

demo1

你想要在程序中创建一个“月亮”对象，并确保在整个程序运行期间，只有一个这样的“月亮”对象存在。这个对象应该是私有的，不可以被其他地方直接访问或修改。就是我们的desc是不允许外部修改的

package singleton

import "sync"

// sync.Once 确保月亮单例实例只会被创建一次
var once sync.Once

// moon 是一个不可导出的结构体
type moon struct {
	description string
}

func (m *moon) String() string {
	return m.description
}

// theMoon 是 moon 类型的单例实例
var theMoon *moon

// TheMoon 返回月亮单例实例
func TheMoon() *moon {
	// 使用 sync.Once 确保 theMoon 只会被初始化一次
	once.Do(func() {
		theMoon = &moon{
			description: "美丽的月亮，照亮夜空。",
		}
	})
	return theMoon
}