mirror of
https://github.com/ccfos/nightingale.git
synced 2026-03-02 22:19:10 +00:00
86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
package dscache
|
||
|
||
import (
|
||
"sync"
|
||
|
||
"github.com/ccfos/nightingale/v6/datasource"
|
||
"github.com/toolkits/pkg/logger"
|
||
)
|
||
|
||
type Cache struct {
|
||
datas map[string]map[int64]datasource.Datasource
|
||
mutex *sync.RWMutex
|
||
}
|
||
|
||
var DsCache = Cache{
|
||
datas: make(map[string]map[int64]datasource.Datasource),
|
||
mutex: new(sync.RWMutex),
|
||
}
|
||
|
||
func (cs *Cache) Put(cate string, dsId int64, ds datasource.Datasource) {
|
||
cs.mutex.Lock()
|
||
if _, found := cs.datas[cate]; !found {
|
||
cs.datas[cate] = make(map[int64]datasource.Datasource)
|
||
}
|
||
|
||
if _, found := cs.datas[cate][dsId]; found {
|
||
if cs.datas[cate][dsId].Equal(ds) {
|
||
cs.mutex.Unlock()
|
||
return
|
||
}
|
||
}
|
||
cs.mutex.Unlock()
|
||
|
||
// InitClient() 在用户配置错误或远端不可用时, 会非常耗时, mutex被长期持有, 导致Get()会超时
|
||
err := ds.InitClient()
|
||
if err != nil {
|
||
logger.Errorf("init plugin:%s %d %+v client fail: %v", cate, dsId, ds, err)
|
||
return
|
||
}
|
||
|
||
logger.Debugf("init plugin:%s %d %+v client success", cate, dsId, ds)
|
||
cs.mutex.Lock()
|
||
cs.datas[cate][dsId] = ds
|
||
cs.mutex.Unlock()
|
||
}
|
||
|
||
func (cs *Cache) Get(cate string, dsId int64) (datasource.Datasource, bool) {
|
||
cs.mutex.RLock()
|
||
defer cs.mutex.RUnlock()
|
||
if _, found := cs.datas[cate]; !found {
|
||
return nil, false
|
||
}
|
||
|
||
if _, found := cs.datas[cate][dsId]; !found {
|
||
return nil, false
|
||
}
|
||
|
||
return cs.datas[cate][dsId], true
|
||
}
|
||
|
||
func (cs *Cache) Delete(cate string, dsId int64) {
|
||
cs.mutex.Lock()
|
||
defer cs.mutex.Unlock()
|
||
if _, found := cs.datas[cate]; !found {
|
||
return
|
||
}
|
||
delete(cs.datas[cate], dsId)
|
||
|
||
logger.Debugf("delete plugin:%s %d from cache", cate, dsId)
|
||
}
|
||
|
||
// GetAllIds 返回缓存中所有数据源的 ID，按类型分组
|
||
func (cs *Cache) GetAllIds() map[string][]int64 {
|
||
cs.mutex.RLock()
|
||
defer cs.mutex.RUnlock()
|
||
result := make(map[string][]int64)
|
||
for cate, dsMap := range cs.datas {
|
||
ids := make([]int64, 0, len(dsMap))
|
||
for dsId := range dsMap {
|
||
ids = append(ids, dsId)
|
||
}
|
||
result[cate] = ids
|
||
}
|
||
return result
|
||
}
|