mirror of
https://github.com/ClusterCockpit/cc-metric-collector.git
synced 2024-11-10 04:27:25 +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>
92 lines
1.9 KiB
Go
92 lines
1.9 KiB
Go
package receivers
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
lp "github.com/ClusterCockpit/cc-metric-collector/internal/ccMetric"
|
|
influx "github.com/influxdata/line-protocol"
|
|
nats "github.com/nats-io/nats.go"
|
|
"log"
|
|
"time"
|
|
)
|
|
|
|
type NatsReceiverConfig struct {
|
|
Addr string `json:"address"`
|
|
Port string `json:"port"`
|
|
Database string `json:"database"`
|
|
}
|
|
|
|
type NatsReceiver struct {
|
|
receiver
|
|
nc *nats.Conn
|
|
handler *influx.MetricHandler
|
|
parser *influx.Parser
|
|
meta map[string]string
|
|
config ReceiverConfig
|
|
}
|
|
|
|
var DefaultTime = func() time.Time {
|
|
return time.Unix(42, 0)
|
|
}
|
|
|
|
func (r *NatsReceiver) Init(config ReceiverConfig) error {
|
|
r.name = "NatsReceiver"
|
|
r.config = config
|
|
if len(r.config.Addr) == 0 ||
|
|
len(r.config.Port) == 0 ||
|
|
len(r.config.Database) == 0 {
|
|
return errors.New("Not all configuration variables set required by NatsReceiver")
|
|
}
|
|
r.meta = map[string]string{"source": r.name}
|
|
r.addr = r.config.Addr
|
|
if len(r.addr) == 0 {
|
|
r.addr = nats.DefaultURL
|
|
}
|
|
r.port = r.config.Port
|
|
if len(r.port) == 0 {
|
|
r.port = "4222"
|
|
}
|
|
log.Print("[NatsReceiver] INIT")
|
|
uri := fmt.Sprintf("%s:%s", r.addr, r.port)
|
|
nc, err := nats.Connect(uri)
|
|
if err == nil {
|
|
r.database = r.config.Database
|
|
r.nc = nc
|
|
} else {
|
|
r.nc = nil
|
|
return err
|
|
}
|
|
r.handler = influx.NewMetricHandler()
|
|
r.parser = influx.NewParser(r.handler)
|
|
r.parser.SetTimeFunc(DefaultTime)
|
|
return err
|
|
}
|
|
|
|
func (r *NatsReceiver) Start() {
|
|
log.Print("[NatsReceiver] START")
|
|
r.nc.Subscribe(r.database, r._NatsReceive)
|
|
}
|
|
|
|
func (r *NatsReceiver) _NatsReceive(m *nats.Msg) {
|
|
metrics, err := r.parser.Parse(m.Data)
|
|
if err == nil {
|
|
for _, m := range metrics {
|
|
y := lp.FromInfluxMetric(m)
|
|
for k, v := range r.meta {
|
|
y.AddMeta(k, v)
|
|
}
|
|
//y, err := lp.New(m.Name(), Tags2Map(m), r.meta, Fields2Map(m), m.Time())
|
|
if r.sink != nil {
|
|
r.sink <- y
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (r *NatsReceiver) Close() {
|
|
if r.nc != nil {
|
|
log.Print("[NatsReceiver] CLOSE")
|
|
r.nc.Close()
|
|
}
|
|
}
|