mirror of
https://github.com/ClusterCockpit/cc-backend
synced 2025-11-01 00:15:05 +01:00
127 lines
3.3 KiB
Go
127 lines
3.3 KiB
Go
package config
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
cclog "github.com/ClusterCockpit/cc-lib/ccLogger"
|
|
)
|
|
|
|
var InternalCCMSFlag bool = false
|
|
|
|
type MetricStoreConfig struct {
|
|
Checkpoints struct {
|
|
FileFormat string `json:"file-format"`
|
|
Interval string `json:"interval"`
|
|
RootDir string `json:"directory"`
|
|
Restore string `json:"restore"`
|
|
} `json:"checkpoints"`
|
|
Debug struct {
|
|
DumpToFile string `json:"dump-to-file"`
|
|
EnableGops bool `json:"gops"`
|
|
} `json:"debug"`
|
|
RetentionInMemory string `json:"retention-in-memory"`
|
|
Archive struct {
|
|
Interval string `json:"interval"`
|
|
RootDir string `json:"directory"`
|
|
DeleteInstead bool `json:"delete-instead"`
|
|
} `json:"archive"`
|
|
Nats []*NatsConfig `json:"nats"`
|
|
}
|
|
|
|
type NatsConfig struct {
|
|
// Address of the nats server
|
|
Address string `json:"address"`
|
|
|
|
// Username/Password, optional
|
|
Username string `json:"username"`
|
|
Password string `json:"password"`
|
|
|
|
// Creds file path
|
|
Credsfilepath string `json:"creds-file-path"`
|
|
|
|
Subscriptions []struct {
|
|
// Channel name
|
|
SubscribeTo string `json:"subscribe-to"`
|
|
|
|
// Allow lines without a cluster tag, use this as default, optional
|
|
ClusterTag string `json:"cluster-tag"`
|
|
} `json:"subscriptions"`
|
|
}
|
|
|
|
var MetricStoreKeys MetricStoreConfig
|
|
|
|
// AggregationStrategy for aggregation over multiple values at different cpus/sockets/..., not time!
|
|
type AggregationStrategy int
|
|
|
|
const (
|
|
NoAggregation AggregationStrategy = iota
|
|
SumAggregation
|
|
AvgAggregation
|
|
)
|
|
|
|
func AssignAggregationStratergy(str string) (AggregationStrategy, error) {
|
|
switch str {
|
|
case "":
|
|
return NoAggregation, nil
|
|
case "sum":
|
|
return SumAggregation, nil
|
|
case "avg":
|
|
return AvgAggregation, nil
|
|
default:
|
|
return NoAggregation, fmt.Errorf("[METRICSTORE]> unknown aggregation strategy: %s", str)
|
|
}
|
|
}
|
|
|
|
type MetricConfig struct {
|
|
// Interval in seconds at which measurements will arive.
|
|
Frequency int64
|
|
|
|
// Can be 'sum', 'avg' or null. Describes how to aggregate metrics from the same timestep over the hierarchy.
|
|
Aggregation AggregationStrategy
|
|
|
|
// Private, used internally...
|
|
Offset int
|
|
}
|
|
|
|
var Metrics map[string]MetricConfig
|
|
|
|
func InitMetricStore(msConfig json.RawMessage) {
|
|
// Validate(msConfigSchema, msConfig)
|
|
dec := json.NewDecoder(bytes.NewReader(msConfig))
|
|
// dec.DisallowUnknownFields()
|
|
if err := dec.Decode(&MetricStoreKeys); err != nil {
|
|
cclog.Abortf("[METRICSTORE]> Metric Store Config Init: Could not decode config file '%s'.\nError: %s\n", msConfig, err.Error())
|
|
}
|
|
}
|
|
|
|
func GetMetricFrequency(metricName string) (int64, error) {
|
|
if metric, ok := Metrics[metricName]; ok {
|
|
return metric.Frequency, nil
|
|
}
|
|
return 0, fmt.Errorf("[METRICSTORE]> metric %s not found", metricName)
|
|
}
|
|
|
|
// AddMetric adds logic to add metrics. Redundant metrics should be updated with max frequency.
|
|
// use metric.Name to check if the metric already exists.
|
|
// if not, add it to the Metrics map.
|
|
func AddMetric(name string, metric MetricConfig) error {
|
|
if Metrics == nil {
|
|
Metrics = make(map[string]MetricConfig, 0)
|
|
}
|
|
|
|
if existingMetric, ok := Metrics[name]; ok {
|
|
if existingMetric.Frequency != metric.Frequency {
|
|
if existingMetric.Frequency < metric.Frequency {
|
|
existingMetric.Frequency = metric.Frequency
|
|
Metrics[name] = existingMetric
|
|
}
|
|
}
|
|
} else {
|
|
Metrics[name] = metric
|
|
}
|
|
|
|
return nil
|
|
}
|