mirror of
https://github.com/Walter-Sparrow/lunar-tear.git
synced 2026-07-02 05:43:41 +03:00
Ilya Groshev
102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
package service
|
|
|
|
import (
|
|
"log"
|
|
"net"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type revisionTracker struct {
|
|
mu sync.RWMutex
|
|
activeByClient map[string]string
|
|
lastRevision string
|
|
}
|
|
|
|
type assetResolution struct {
|
|
ActiveRevision string
|
|
ListRevision string
|
|
ListSize int64
|
|
Candidates []assetCandidate
|
|
}
|
|
|
|
type assetResolver struct {
|
|
baseDir string
|
|
}
|
|
|
|
func newRevisionTracker() *revisionTracker {
|
|
return &revisionTracker{
|
|
activeByClient: make(map[string]string),
|
|
}
|
|
}
|
|
|
|
func newAssetResolver(baseDir string) *assetResolver {
|
|
return &assetResolver{baseDir: baseDir}
|
|
}
|
|
|
|
func normalizeClientAddr(remoteAddr string) string {
|
|
host, _, err := net.SplitHostPort(remoteAddr)
|
|
if err == nil && host != "" {
|
|
return host
|
|
}
|
|
return remoteAddr
|
|
}
|
|
|
|
func (t *revisionTracker) Remember(clientAddr, revision string) {
|
|
if revision == "" {
|
|
return
|
|
}
|
|
client := normalizeClientAddr(clientAddr)
|
|
t.mu.Lock()
|
|
if client != "" {
|
|
t.activeByClient[client] = revision
|
|
}
|
|
t.lastRevision = revision
|
|
t.mu.Unlock()
|
|
log.Printf("[Octo] Active list revision for client=%s set to %s", client, revision)
|
|
}
|
|
|
|
func (t *revisionTracker) Active(clientAddr string) string {
|
|
client := normalizeClientAddr(clientAddr)
|
|
t.mu.RLock()
|
|
revision := t.activeByClient[client]
|
|
if revision == "" {
|
|
revision = t.lastRevision
|
|
}
|
|
t.mu.RUnlock()
|
|
if revision == "" {
|
|
return "0"
|
|
}
|
|
return revision
|
|
}
|
|
|
|
func (r *assetResolver) Resolve(objectId, assetType, activeRevision, platform string) (assetResolution, bool) {
|
|
start := time.Now()
|
|
resolution := assetResolution{ActiveRevision: activeRevision}
|
|
revision := activeRevision
|
|
|
|
candidates, listSize, ok := objectIdToFilePathCandidates(r.baseDir, revision, platform, assetType, objectId)
|
|
if ok && len(candidates) > 0 {
|
|
resolution.ListRevision = revision
|
|
resolution.ListSize = listSize
|
|
resolution.Candidates = candidates
|
|
if elapsed := time.Since(start); elapsed > 100*time.Millisecond {
|
|
log.Printf("[HTTP] Asset resolve slow: object_id=%s type=%s platform=%s active_revision=%s list_revision=%s elapsed=%s", objectId, assetType, platform, activeRevision, revision, elapsed)
|
|
}
|
|
return resolution, true
|
|
}
|
|
|
|
if elapsed := time.Since(start); elapsed > 100*time.Millisecond {
|
|
log.Printf("[HTTP] Asset resolve miss: object_id=%s type=%s platform=%s active_revision=%s elapsed=%s", objectId, assetType, platform, activeRevision, elapsed)
|
|
}
|
|
return resolution, false
|
|
}
|
|
|
|
func (r *assetResolver) Prewarm(activeRevision, platform string) {
|
|
if activeRevision == "" {
|
|
return
|
|
}
|
|
_, _ = loadListBinIndex(r.baseDir, activeRevision, platform)
|
|
_ = loadInfoIndex(r.baseDir, activeRevision, platform)
|
|
}
|