2021-03-25 14:46:21 +01:00
|
|
|
package collectors
|
|
|
|
|
|
|
|
import (
|
2022-01-25 15:37:43 +01:00
|
|
|
"encoding/json"
|
2022-01-25 17:20:20 +01:00
|
|
|
"fmt"
|
2021-03-25 17:47:08 +01:00
|
|
|
"time"
|
2022-01-25 17:20:20 +01:00
|
|
|
|
|
|
|
lp "github.com/ClusterCockpit/cc-metric-collector/internal/ccMetric"
|
2021-03-25 14:46:21 +01:00
|
|
|
)
|
|
|
|
|
2022-01-25 15:37:43 +01:00
|
|
|
type MetricCollector interface {
|
2022-05-13 14:10:39 +02:00
|
|
|
Name() string // Name of the metric collector
|
|
|
|
Init(config json.RawMessage) error // Initialize metric collector
|
|
|
|
Initialized() bool // Is metric collector initialized?
|
|
|
|
Parallel() bool
|
2022-02-28 12:16:48 +01:00
|
|
|
Read(duration time.Duration, output chan lp.CCMetric) // Read metrics from metric collector
|
|
|
|
Close() // Close / finish metric collector
|
2021-03-25 14:46:21 +01:00
|
|
|
}
|
|
|
|
|
2022-01-25 15:37:43 +01:00
|
|
|
type metricCollector struct {
|
2022-05-13 14:10:39 +02:00
|
|
|
name string // name of the metric
|
|
|
|
init bool // is metric collector initialized?
|
|
|
|
parallel bool // can the metric collector be executed in parallel with others
|
|
|
|
meta map[string]string // static meta data tags
|
2021-03-25 14:46:21 +01:00
|
|
|
}
|
|
|
|
|
2022-02-28 12:16:48 +01:00
|
|
|
// Name returns the name of the metric collector
|
2022-01-25 15:37:43 +01:00
|
|
|
func (c *metricCollector) Name() string {
|
2021-03-25 14:46:21 +01:00
|
|
|
return c.name
|
|
|
|
}
|
|
|
|
|
2022-05-13 14:10:39 +02:00
|
|
|
// Name returns the name of the metric collector
|
|
|
|
func (c *metricCollector) Parallel() bool {
|
|
|
|
return c.parallel
|
|
|
|
}
|
|
|
|
|
2022-02-28 12:16:48 +01:00
|
|
|
// Setup is for future use
|
2022-01-25 15:37:43 +01:00
|
|
|
func (c *metricCollector) setup() error {
|
2021-03-25 17:47:08 +01:00
|
|
|
return nil
|
2021-03-25 14:46:21 +01:00
|
|
|
}
|
|
|
|
|
2022-02-28 12:16:48 +01:00
|
|
|
// Initialized indicates whether the metric collector has been initialized
|
2022-01-25 15:37:43 +01:00
|
|
|
func (c *metricCollector) Initialized() bool {
|
2022-01-25 17:20:20 +01:00
|
|
|
return c.init
|
2021-11-26 19:01:31 +01:00
|
|
|
}
|
|
|
|
|
2022-01-25 17:20:20 +01:00
|
|
|
// intArrayContains scans an array of ints if the value str is present in the array
|
|
|
|
// If the specified value is found, the corresponding array index is returned.
|
|
|
|
// The bool value is used to signal success or failure
|
2021-03-25 14:46:21 +01:00
|
|
|
func intArrayContains(array []int, str int) (int, bool) {
|
2021-03-25 17:47:08 +01:00
|
|
|
for i, a := range array {
|
|
|
|
if a == str {
|
|
|
|
return i, true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return -1, false
|
2021-03-25 14:46:21 +01:00
|
|
|
}
|
|
|
|
|
2022-01-25 17:20:20 +01:00
|
|
|
// stringArrayContains scans an array of strings if the value str is present in the array
|
|
|
|
// If the specified value is found, the corresponding array index is returned.
|
|
|
|
// The bool value is used to signal success or failure
|
2021-11-25 14:04:03 +01:00
|
|
|
func stringArrayContains(array []string, str string) (int, bool) {
|
2021-11-25 15:11:39 +01:00
|
|
|
for i, a := range array {
|
2021-11-25 14:04:03 +01:00
|
|
|
if a == str {
|
|
|
|
return i, true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return -1, false
|
|
|
|
}
|
|
|
|
|
2022-01-25 17:20:20 +01:00
|
|
|
// RemoveFromStringList removes the string r from the array of strings s
|
|
|
|
// If r is not contained in the array an error is returned
|
2021-11-25 14:04:03 +01:00
|
|
|
func RemoveFromStringList(s []string, r string) ([]string, error) {
|
2022-02-28 12:16:48 +01:00
|
|
|
for i := range s {
|
|
|
|
if r == s[i] {
|
2021-11-25 15:11:39 +01:00
|
|
|
return append(s[:i], s[i+1:]...), nil
|
|
|
|
}
|
|
|
|
}
|
2022-03-15 16:38:20 +01:00
|
|
|
return s, fmt.Errorf("no such string in list")
|
2021-11-25 14:04:03 +01:00
|
|
|
}
|