91 lines
3.0 KiB
Go
91 lines
3.0 KiB
Go
// cloudcheck.go — 每輪固定要問雲端的兩件事（版本、雲端 AI 通了沒），限一分鐘問一次
|
||
// （`inkstone/arcrun-rag#121`，2026-09-13 真機量測抓到的副作用）。
|
||
//
|
||
// 病：這兩發原本「每輪各打一次」。以前一輪要跑十幾秒到幾分鐘（逐檔去撞壞掉的雲端），
|
||
// 所以每小時大約幾百發。路由退避生效之後，壞掉的那條路不再被打，一輪變成五、六秒就跑完
|
||
// ⇒ **這兩發變成每 5 秒一次**。實測 geek6688 真雲端 3 分鐘：
|
||
// 退避版 80 發（其中 /health 36、空探測 36），修前只有 34 發——
|
||
// 壞掉那條路的請求變少了，雲端收到的總請求卻變多。
|
||
//
|
||
// 解：兩件事都是「狀態」，不是「工作」——一分鐘內再問一次得到的答案幾乎一定一樣。
|
||
// ⇒ 同一台知識庫一分鐘內只真的問一次，其餘沿用上一次的答案。
|
||
//
|
||
// 使用者按「立刻同步」的那一輪照問（他要的就是最新狀況）。
|
||
package collector
|
||
|
||
import (
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
// cloudCheckInterval＝同一台知識庫多久才真的重問一次版本／雲端 AI 狀態。
|
||
var cloudCheckInterval = 60 * time.Second
|
||
|
||
type cachedVersion struct {
|
||
at time.Time
|
||
ver string
|
||
ok bool
|
||
}
|
||
|
||
type cachedAIState struct {
|
||
at time.Time
|
||
state CloudAIState
|
||
}
|
||
|
||
var (
|
||
cloudCheckMu sync.Mutex
|
||
cloudVersionSeen = map[string]cachedVersion{}
|
||
cloudAISeen = map[string]cachedAIState{}
|
||
)
|
||
|
||
// resetCloudChecks 清空快取（測試用）。
|
||
func resetCloudChecks() {
|
||
cloudCheckMu.Lock()
|
||
defer cloudCheckMu.Unlock()
|
||
cloudVersionSeen = map[string]cachedVersion{}
|
||
cloudAISeen = map[string]cachedAIState{}
|
||
}
|
||
|
||
func cloudCheckKey(cypherURL string) string {
|
||
return strings.TrimSuffix(strings.TrimSpace(cypherURL), "/")
|
||
}
|
||
|
||
// cloudVersionThrottled＝fetchCloudVersion，但同一台一分鐘內只真的問一次。force＝立刻同步。
|
||
func cloudVersionThrottled(cypherURL string, force bool) (string, bool) {
|
||
key := cloudCheckKey(cypherURL)
|
||
now := directNow()
|
||
cloudCheckMu.Lock()
|
||
c, hit := cloudVersionSeen[key]
|
||
cloudCheckMu.Unlock()
|
||
if hit && !force && now.Sub(c.at) < cloudCheckInterval && !now.Before(c.at) {
|
||
return c.ver, c.ok
|
||
}
|
||
ver, ok := fetchCloudVersion(cypherURL)
|
||
cloudCheckMu.Lock()
|
||
cloudVersionSeen[key] = cachedVersion{at: now, ver: ver, ok: ok}
|
||
cloudCheckMu.Unlock()
|
||
return ver, ok
|
||
}
|
||
|
||
// cachedAI 回上一次（一分鐘內）問到的雲端 AI 狀態；沒有就 ok=false。
|
||
func cachedAI(cypherURL string, force bool) (CloudAIState, bool) {
|
||
if force {
|
||
return CloudAIState{}, false
|
||
}
|
||
now := directNow()
|
||
cloudCheckMu.Lock()
|
||
defer cloudCheckMu.Unlock()
|
||
c, hit := cloudAISeen[cloudCheckKey(cypherURL)]
|
||
if !hit || now.Sub(c.at) >= cloudCheckInterval || now.Before(c.at) {
|
||
return CloudAIState{}, false
|
||
}
|
||
return c.state, true
|
||
}
|
||
|
||
func rememberAI(cypherURL string, st CloudAIState) {
|
||
cloudCheckMu.Lock()
|
||
defer cloudCheckMu.Unlock()
|
||
cloudAISeen[cloudCheckKey(cypherURL)] = cachedAIState{at: directNow(), state: st}
|
||
}
|