33 lines
690 B
Go
33 lines
690 B
Go
|
package cache
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/didi/nightingale/v5/models"
|
||
|
|
)
|
||
|
|
|
||
|
|
type CollectRuleOfIdentMap struct {
|
||
|
|
sync.RWMutex
|
||
|
|
Data map[string][]*models.CollectRule
|
||
|
|
}
|
||
|
|
|
||
|
|
var CollectRulesOfIdent = &CollectRuleOfIdentMap{Data: make(map[string][]*models.CollectRule)}
|
||
|
|
|
||
|
|
func (c *CollectRuleOfIdentMap) GetBy(ident string) []*models.CollectRule {
|
||
|
|
c.RLock()
|
||
|
|
defer c.RUnlock()
|
||
|
|
return c.Data[ident]
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *CollectRuleOfIdentMap) Set(node string, collectRules []*models.CollectRule) {
|
||
|
|
c.Lock()
|
||
|
|
defer c.Unlock()
|
||
|
|
c.Data[node] = collectRules
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *CollectRuleOfIdentMap) SetAll(collectRulesMap map[string][]*models.CollectRule) {
|
||
|
|
c.Lock()
|
||
|
|
defer c.Unlock()
|
||
|
|
c.Data = collectRulesMap
|
||
|
|
}
|