package main

import (
    "log"
    "net/http"
    "sync"
)

type Config struct {
    APIKey   string
    LogLevel string
}

var (
    config *Config
    once   sync.Once
)

func loadConfig() {
    // 模擬從文件或環(huán)境變量加載配置
    config = &Config{
       APIKey:   "secret-key",
       LogLevel: "debug",
    }
    log.Println("Configuration loaded")
}

func GetConfig() *Config {
    once.Do(loadConfig) // 僅第一次訪問時(shí)會執(zhí)行l(wèi)oadConfig函數(shù)
    return config
}

func handler(w http.ResponseWriter, r *http.Request) {
    cfg := GetConfig()
    log.Printf("Request handled with API key: %s", cfg.APIKey)
    w.Write([]byte("OK"))
}

func main() {
    http.HandleFunc("/", handler)
    log.Fatal(http.ListenAndServe(":8888", nil))
}

package sync

import (
    "sync/atomic"
)

type Once struct {
    done atomic.Uint32 // 是否已執(zhí)行標(biāo)識位，0-未執(zhí)行 1-已執(zhí)行
    m    Mutex         // 互斥鎖，確保并發(fā)安全
}

func (o *Once) Do(f func()) {
    // 第一次執(zhí)行Do函數(shù)時(shí)，原子操作檢查o.done==0,執(zhí)行doSlow函數(shù)后，o.done==1
    // 第二次及之后執(zhí)行Do函數(shù),原子操作檢查o.done標(biāo)識位為1，Do函數(shù)不執(zhí)行任何功能，確保了f函數(shù)只在第一次被執(zhí)行
    if o.done.Load() == 0 {
       o.doSlow(f) // 調(diào)用doSlow函數(shù)執(zhí)行f方法。第一次執(zhí)行時(shí)，同一時(shí)間可能有多個(gè)goroutine嘗試同時(shí)執(zhí)行doSlow函數(shù)
    }
}

func (o *Once) doSlow(f func()) {
    o.m.Lock() // 加鎖保護(hù)，避免多個(gè)goroutine同時(shí)繞過之前的原子操作檢查，并發(fā)修改o.done的值
    defer o.m.Unlock()
    // 二次檢查o.done的值，同一時(shí)間并發(fā)執(zhí)行doSlow函數(shù)的goroutine,在第一個(gè)goroutine將o.done置為1并解除互斥鎖后,
    // 剩下的goroutine識別到自身的o.done已經(jīng)被設(shè)為1，無法繞過二次檢查
    if o.done.Load() == 0 { 
       defer o.done.Store(1) // 需要在f()函數(shù)執(zhí)行完成之后，原子性地將o.done設(shè)為1
       f() // 執(zhí)行f方法，一定只有一個(gè)goroutine會調(diào)用這個(gè)方法
    }
}