package main

import (
    "context"
    "fmt"
    "time"

    "golang.org/x/sync/singleflight"
)

var sfGroup singleflight.Group

// CacheGetOrSet 从缓存中获取指定 key 的值，
// 如果不存在则调用 fetch 获取并写入缓存。
func CacheGetOrSet(ctx context.Context, key string, ttl time.Duration, fetch func(context.Context) (any, error)) (any, error) {
    // 实际场景下，这里可以先查 Redis / 内存缓存
    // 这里为了示例直接调用 fetch
    return fetch(ctx)
}

// 模拟获取数据函数
func fetchData(ctx context.Context) ([]string, error) {
    // 模拟耗时操作
    select {
    case <-time.After(500 * time.Millisecond):
        return []string{"a", "b", "c"}, nil
    case <-ctx.Done():
        return nil, ctx.Err()
    }
}

func loadData(ctx context.Context) ([]string, error) {
    data, err, _ := sfGroup.Do("mydata", func() (interface{}, error) {
        return CacheGetOrSet(ctx, "mydata", time.Minute, func(c context.Context) (any, error) {
            fetchCtx, cancel := context.WithTimeout(c, time.Second)
            defer cancel()
            return fetchData(fetchCtx)
        })
    })
    if err != nil {
        return nil, fmt.Errorf("failed to load data: %w", err)
    }

    result, ok := data.([]string)
    if !ok {
        return nil, fmt.Errorf("unexpected type in cache")
    }

    return result, nil
}

func main() {
    ctx := context.Background()
    data, err := loadData(ctx)
    if err != nil {
        panic(err)
    }
    fmt.Println("Loaded:", data)
}