mirror of
https://github.com/ClusterCockpit/cc-metric-collector.git
synced 2024-12-27 07:39:05 +01:00
200af84c54
* Use channels, add a metric router, split up configuration and use extended version of Influx line protocol internally * Use central timer for collectors and router. Add expressions to router * Add expression to router config * Update entry points * Start with README * Update README for CCMetric * Formatting * Update README.md * Add README for MultiChanTicker * Add README for MultiChanTicker * Update README.md * Add README to metric router * Update main README * Remove SinkEntity type * Update README for sinks * Update go files * Update README for receivers * Update collectors README * Update collectors README * Use seperate page per collector * Fix for tempstat page * Add docs for customcmd collector * Add docs for ipmistat collector * Add docs for topprocs collector * Update customCmdMetric.md * Use seconds when calculating LIKWID metrics * Add IB metrics ib_recv_pkts and ib_xmit_pkts * Drop domain part of host name * Updated to latest stable version of likwid * Define source code dependencies in Makefile * Add GPFS / IBM Spectrum Scale collector * Add vet and staticcheck make targets * Add vet and staticcheck make targets * Avoid go vet warning: struct field tag `json:"..., omitempty"` not compatible with reflect.StructTag.Get: suspicious space in struct tag value struct field tag `json:"...", omitempty` not compatible with reflect.StructTag.Get: key:"value" pairs not separated by spaces * Add sample collector to README.md * Add CPU frequency collector * Avoid staticcheck warning: redundant return statement * Avoid staticcheck warning: unnecessary assignment to the blank identifier * Simplified code * Add CPUFreqCollectorCpuinfo a metric collector to measure the current frequency of the CPUs as obtained from /proc/cpuinfo Only measure on the first hyperthread * Add collector for NFS clients * Move publication of metrics into Flush() for NatsSink * Update GitHub actions * Refactoring * Avoid vet warning: Println arg list ends with redundant newline * Avoid vet warning struct field commands has json tag but is not exported * Avoid vet warning: return copies lock value. * Corrected typo * Refactoring * Add go sources in internal/... * Bad separator in Makefile * Fix Infiniband collector Co-authored-by: Holger Obermaier <40787752+ho-ob@users.noreply.github.com>
88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package sinks
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"errors"
|
|
"fmt"
|
|
lp "github.com/ClusterCockpit/cc-metric-collector/internal/ccMetric"
|
|
influxdb2 "github.com/influxdata/influxdb-client-go/v2"
|
|
influxdb2Api "github.com/influxdata/influxdb-client-go/v2/api"
|
|
"log"
|
|
)
|
|
|
|
type InfluxSink struct {
|
|
sink
|
|
client influxdb2.Client
|
|
writeApi influxdb2Api.WriteAPIBlocking
|
|
retPolicy string
|
|
}
|
|
|
|
func (s *InfluxSink) connect() error {
|
|
var auth string
|
|
var uri string
|
|
if s.ssl {
|
|
uri = fmt.Sprintf("https://%s:%s", s.host, s.port)
|
|
} else {
|
|
uri = fmt.Sprintf("http://%s:%s", s.host, s.port)
|
|
}
|
|
if len(s.user) == 0 {
|
|
auth = s.password
|
|
} else {
|
|
auth = fmt.Sprintf("%s:%s", s.user, s.password)
|
|
}
|
|
log.Print("Using URI ", uri, " Org ", s.organization, " Bucket ", s.database)
|
|
s.client = influxdb2.NewClientWithOptions(uri, auth,
|
|
influxdb2.DefaultOptions().SetTLSConfig(&tls.Config{InsecureSkipVerify: true}))
|
|
s.writeApi = s.client.WriteAPIBlocking(s.organization, s.database)
|
|
return nil
|
|
}
|
|
|
|
func (s *InfluxSink) Init(config sinkConfig) error {
|
|
s.name = "InfluxSink"
|
|
if len(config.Host) == 0 ||
|
|
len(config.Port) == 0 ||
|
|
len(config.Database) == 0 ||
|
|
len(config.Organization) == 0 ||
|
|
len(config.Password) == 0 {
|
|
return errors.New("Not all configuration variables set required by InfluxSink")
|
|
}
|
|
s.host = config.Host
|
|
s.port = config.Port
|
|
s.database = config.Database
|
|
s.organization = config.Organization
|
|
s.user = config.User
|
|
s.password = config.Password
|
|
s.ssl = config.SSL
|
|
s.meta_as_tags = config.MetaAsTags
|
|
return s.connect()
|
|
}
|
|
|
|
func (s *InfluxSink) Write(point lp.CCMetric) error {
|
|
tags := map[string]string{}
|
|
fields := map[string]interface{}{}
|
|
for _, t := range point.TagList() {
|
|
tags[t.Key] = t.Value
|
|
}
|
|
if s.meta_as_tags {
|
|
for _, m := range point.MetaList() {
|
|
tags[m.Key] = m.Value
|
|
}
|
|
}
|
|
for _, f := range point.FieldList() {
|
|
fields[f.Key] = f.Value
|
|
}
|
|
p := influxdb2.NewPoint(point.Name(), tags, fields, point.Time())
|
|
err := s.writeApi.WritePoint(context.Background(), p)
|
|
return err
|
|
}
|
|
|
|
func (s *InfluxSink) Flush() error {
|
|
return nil
|
|
}
|
|
|
|
func (s *InfluxSink) Close() {
|
|
log.Print("Closing InfluxDB connection")
|
|
s.client.Close()
|
|
}
|