forked from ccfos/nightingale
102 lines
2.1 KiB
Go
102 lines
2.1 KiB
Go
package prom
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/ccfos/nightingale/v6/models"
|
|
"github.com/ccfos/nightingale/v6/pkg/ctx"
|
|
"github.com/ccfos/nightingale/v6/pkg/prom"
|
|
)
|
|
|
|
type PromClientMap struct {
|
|
sync.RWMutex
|
|
ctx *ctx.Context
|
|
ReaderClients map[int64]prom.API
|
|
WriterClients map[int64]prom.WriterType
|
|
}
|
|
|
|
func (pc *PromClientMap) Set(datasourceId int64, r prom.API, w prom.WriterType) {
|
|
if r == nil {
|
|
return
|
|
}
|
|
pc.Lock()
|
|
defer pc.Unlock()
|
|
pc.ReaderClients[datasourceId] = r
|
|
pc.WriterClients[datasourceId] = w
|
|
}
|
|
|
|
func (pc *PromClientMap) GetDatasourceIds() []int64 {
|
|
pc.RLock()
|
|
defer pc.RUnlock()
|
|
var datasourceIds []int64
|
|
for k := range pc.ReaderClients {
|
|
datasourceIds = append(datasourceIds, k)
|
|
}
|
|
|
|
return datasourceIds
|
|
}
|
|
|
|
func (pc *PromClientMap) GetCli(datasourceId int64) prom.API {
|
|
pc.RLock()
|
|
defer pc.RUnlock()
|
|
c := pc.ReaderClients[datasourceId]
|
|
return c
|
|
}
|
|
|
|
func (pc *PromClientMap) GetWriterCli(datasourceId int64) prom.WriterType {
|
|
pc.RLock()
|
|
defer pc.RUnlock()
|
|
c := pc.WriterClients[datasourceId]
|
|
return c
|
|
}
|
|
|
|
func (pc *PromClientMap) IsNil(datasourceId int64) bool {
|
|
pc.RLock()
|
|
defer pc.RUnlock()
|
|
|
|
c, exists := pc.ReaderClients[datasourceId]
|
|
if !exists {
|
|
return true
|
|
}
|
|
|
|
return c == nil
|
|
}
|
|
|
|
// Hit 根据当前有效的 datasourceId 和规则的 datasourceId 配置计算有效的cluster列表
|
|
func (pc *PromClientMap) Hit(datasourceIds []int64) []int64 {
|
|
pc.RLock()
|
|
defer pc.RUnlock()
|
|
dsIds := make([]int64, 0, len(pc.ReaderClients))
|
|
if len(datasourceIds) == 1 && datasourceIds[0] == models.DatasourceIdAll {
|
|
for c := range pc.ReaderClients {
|
|
dsIds = append(dsIds, c)
|
|
}
|
|
return dsIds
|
|
}
|
|
|
|
for dsId := range pc.ReaderClients {
|
|
for _, id := range datasourceIds {
|
|
if id == dsId {
|
|
dsIds = append(dsIds, id)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
return dsIds
|
|
}
|
|
|
|
func (pc *PromClientMap) Reset() {
|
|
pc.Lock()
|
|
defer pc.Unlock()
|
|
|
|
pc.ReaderClients = make(map[int64]prom.API)
|
|
pc.WriterClients = make(map[int64]prom.WriterType)
|
|
}
|
|
|
|
func (pc *PromClientMap) Del(datasourceId int64) {
|
|
pc.Lock()
|
|
defer pc.Unlock()
|
|
delete(pc.ReaderClients, datasourceId)
|
|
delete(pc.WriterClients, datasourceId)
|
|
}
|