golang函数缓存的热键处理策略揭秘

import "sync"

// 定义热键缓存
var hotKeyCache sync.Map

// 定义函数缓存
var funcCache sync.Map

// 添加函数到热键缓存
func AddToHotKeyCache(key string, fn interface{}) {
    hotKeyCache.Store(key, true)
}

// 从缓存中获取函数
func GetFromCache(key string) (interface{}, bool) {
    // 检查热键缓存
    if _, ok := hotKeyCache.Load(key); ok {
        return funcCache.Load(key)
    }

    // 从函数缓存中获取
    return funcCache.Load(key)
}

// 增加函数调用次数
func IncrementCallCount(key string) {
    // 检查热键缓存
    if _, ok := hotKeyCache.Load(key); ok {
        return
    }

    // 如果函数调用次数达到某个阈值，将其添加到热键缓存
    if callCount := funcCache.Store(key, callCount + 1); callCount >= 10 {
        AddToHotKeyCache(key,  funcCache.Load(key))
    }
}

func fib(n int) int {
    if n == 0 || n == 1 {
        return 1
    }
    return fib(n-1) + fib(n-2)
}

// 将 fib 函数添加到缓存
funcCache.Store("fib", fib)

// 为 fib 函数增加调用次数
func IncrementFibCallCount() {
    IncrementCallCount("fib")
}