mirror of
https://github.com/ClusterCockpit/cc-metric-collector.git
synced 2024-12-26 15:29:04 +01:00
be20f956c2
* InfiniBandCollector: Scale raw readings from octets to bytes * Fix clock frequency coming from LikwidCollector and update docs * Build DEB package for Ubuntu 20.04 for releases * Fix memstat collector with numa_stats option * Remove useless prints from MemstatCollector * Replace ioutils with os and io (#87) * Use lower case for error strings in RocmSmiCollector * move maybe-usable-by-other-cc-components to pkg. Fix all files to use the new paths (#88) * Add collector for monitoring the execution of cc-metric-collector itself (#81) * Add collector to monitor execution of cc-metric-collector itself * Register SelfCollector * Fix import paths for moved packages
74 lines
1.4 KiB
Go
74 lines
1.4 KiB
Go
package sinks
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
// "time"
|
|
lp "github.com/ClusterCockpit/cc-metric-collector/pkg/ccMetric"
|
|
)
|
|
|
|
type StdoutSink struct {
|
|
sink // meta_as_tags, name
|
|
output *os.File
|
|
config struct {
|
|
defaultSinkConfig
|
|
Output string `json:"output_file,omitempty"`
|
|
}
|
|
}
|
|
|
|
func (s *StdoutSink) Write(m lp.CCMetric) error {
|
|
fmt.Fprint(
|
|
s.output,
|
|
m.ToLineProtocol(s.meta_as_tags),
|
|
)
|
|
return nil
|
|
}
|
|
|
|
func (s *StdoutSink) Flush() error {
|
|
s.output.Sync()
|
|
return nil
|
|
}
|
|
|
|
func (s *StdoutSink) Close() {
|
|
if s.output != os.Stdout && s.output != os.Stderr {
|
|
s.output.Close()
|
|
}
|
|
}
|
|
|
|
func NewStdoutSink(name string, config json.RawMessage) (Sink, error) {
|
|
s := new(StdoutSink)
|
|
s.name = fmt.Sprintf("StdoutSink(%s)", name)
|
|
if len(config) > 0 {
|
|
err := json.Unmarshal(config, &s.config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
s.output = os.Stdout
|
|
if len(s.config.Output) > 0 {
|
|
switch strings.ToLower(s.config.Output) {
|
|
case "stdout":
|
|
s.output = os.Stdout
|
|
case "stderr":
|
|
s.output = os.Stderr
|
|
default:
|
|
f, err := os.OpenFile(s.config.Output, os.O_CREATE|os.O_WRONLY, os.FileMode(0600))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
s.output = f
|
|
}
|
|
}
|
|
// Create lookup map to use meta infos as tags in the output metric
|
|
s.meta_as_tags = make(map[string]bool)
|
|
for _, k := range s.config.MetaAsTags {
|
|
s.meta_as_tags[k] = true
|
|
}
|
|
|
|
return s, nil
|
|
}
|