renjue
9dfa7aa7c5
CI / docker (push) Successful in 4m28s
Co-authored-by: Cursor <cursoragent@cursor.com>
69 lines
1.3 KiB
Go
69 lines
1.3 KiB
Go
package cache
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type DashboardSnapshot struct {
|
|
UploadRate int64 `json:"upload_rate"`
|
|
DownloadRate int64 `json:"download_rate"`
|
|
Connections int `json:"connections"`
|
|
TotalNodes int `json:"total_nodes"`
|
|
AliveNodes int `json:"alive_nodes"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
type Cache struct {
|
|
mu sync.RWMutex
|
|
lastYAML []byte
|
|
dashboard DashboardSnapshot
|
|
proxyLatency map[string]int
|
|
}
|
|
|
|
func New() *Cache {
|
|
return &Cache{
|
|
proxyLatency: make(map[string]int),
|
|
}
|
|
}
|
|
|
|
func (c *Cache) SetLastYAML(yaml []byte) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
c.lastYAML = append([]byte(nil), yaml...)
|
|
}
|
|
|
|
func (c *Cache) GetLastYAML() []byte {
|
|
c.mu.RLock()
|
|
defer c.mu.RUnlock()
|
|
if c.lastYAML == nil {
|
|
return nil
|
|
}
|
|
return append([]byte(nil), c.lastYAML...)
|
|
}
|
|
|
|
func (c *Cache) SetDashboard(s DashboardSnapshot) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
c.dashboard = s
|
|
}
|
|
|
|
func (c *Cache) GetDashboard() DashboardSnapshot {
|
|
c.mu.RLock()
|
|
defer c.mu.RUnlock()
|
|
return c.dashboard
|
|
}
|
|
|
|
func (c *Cache) SetProxyLatency(name string, ms int) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
c.proxyLatency[name] = ms
|
|
}
|
|
|
|
func (c *Cache) GetProxyLatency(name string) (int, bool) {
|
|
c.mu.RLock()
|
|
defer c.mu.RUnlock()
|
|
v, ok := c.proxyLatency[name]
|
|
return v, ok
|
|
}
|