帶有 Go 泛型的單例模式

問題內容

我正在嘗試找出在 golang 中為泛型變數實作單例的最不糟糕的方法。使用普通的 sync.once 模式與全域變數是行不通的，因為通用類型資訊在那裡不可用（如下）。

這個範例是人為設計的，但實際上，維護單例的程式碼可以與定義 t 的客戶端程式碼分開（例如在庫中）。

假設此庫程式碼，其中 t 的具體值未知：

type cache[t any] struct{}

var (
    cacheonce sync.once
    cache     cache[any] // global singleton
)

func getorcreatecache[t any]() cache[t] {
    cacheonce.do(func() {
        typedcache := buildcache()
        cache = typedcache.(cache[any]) // invalid type assertion
    })
    return cache.(cache[t]) // invalid type assertion
}

並假設這個單獨的客戶端程式碼，其中 t 被定義為 string：

stringCache := getOrCreateCache[string]()

實現這目標的最佳方法是什麼？

正確答案

我最終使用atomic.pointer api 解決了這個問題，並將其整合到一個簡單的庫中，供其他有興趣的人使用：單線。使用單線態重新處理原始帖子如下所示：

範例庫程式碼：

type cache[t any] struct{}

var singleton = &singlet.singleton{}

func getorcreatecache[t any]() (cache[t], err) {
    return singlet.getordo(singleton, func() cache[t] {
        return cache[t]{}
    })
}

客戶端程式碼：

stringCache := getOrCreateCache[string]()

以及支援此功能的 singlet 程式庫程式碼：

var ErrTypeMismatch = errors.New("the requested type does not match the singleton type")

type Singleton struct {
    p   atomic.Pointer[any]
    mtx sync.Mutex
}

func GetOrDo[T any](singleton *Singleton, fn func() T) (result T, err error) {
    maybeResult := singleton.p.Load()
    if maybeResult == nil {
        // Lock to guard against applying fn twice
        singleton.mtx.Lock()
        defer singleton.mtx.Unlock()
        maybeResult = singleton.p.Load()

        // Double check
        if maybeResult == nil {
            result = fn()
            var resultAny any = result
            singleton.p.Store(&resultAny)
            return result, nil
        }
    }

    var ok bool
    result, ok = (*maybeResult).(T)
    if !ok {
        return *new(T), ErrTypeMismatch
    }
    return result, nil
}

我希望對遇到這種情況的其他人有所幫助。

以上是帶有 Go 泛型的單例模式的詳細內容。更多資訊請關注PHP中文網其他相關文章！