Carlos Florêncio
41 lines
858 B
Go
41 lines
858 B
Go
package hostutility
|
|
|
|
import (
|
|
"sort"
|
|
"sync"
|
|
)
|
|
|
|
// cache is a thread-safe in-memory map of agent capabilities keyed by agent type.
|
|
type cache struct {
|
|
mu sync.RWMutex
|
|
byType map[string]AgentCapabilities
|
|
}
|
|
|
|
func newCache() *cache {
|
|
return &cache{byType: make(map[string]AgentCapabilities)}
|
|
}
|
|
|
|
func (c *cache) set(caps AgentCapabilities) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
c.byType[caps.AgentType] = caps
|
|
}
|
|
|
|
func (c *cache) get(agentType string) (AgentCapabilities, bool) {
|
|
c.mu.RLock()
|
|
defer c.mu.RUnlock()
|
|
caps, ok := c.byType[agentType]
|
|
return caps, ok
|
|
}
|
|
|
|
func (c *cache) all() []AgentCapabilities {
|
|
c.mu.RLock()
|
|
defer c.mu.RUnlock()
|
|
out := make([]AgentCapabilities, 0, len(c.byType))
|
|
for _, v := range c.byType {
|
|
out = append(out, v)
|
|
}
|
|
sort.Slice(out, func(i, j int) bool { return out[i].AgentType < out[j].AgentType })
|
|
return out
|
|
}
|