mirror of
https://github.com/ClusterCockpit/cc-metric-collector.git
synced 2026-02-17 08:21:46 +01:00
Use cc-lib lp.FromBytes, do not use influxdb client directly
This commit is contained in:
@@ -8,11 +8,9 @@
|
|||||||
package collectors
|
package collectors
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"slices"
|
"slices"
|
||||||
@@ -21,9 +19,6 @@ import (
|
|||||||
|
|
||||||
cclog "github.com/ClusterCockpit/cc-lib/v2/ccLogger"
|
cclog "github.com/ClusterCockpit/cc-lib/v2/ccLogger"
|
||||||
lp "github.com/ClusterCockpit/cc-lib/v2/ccMessage"
|
lp "github.com/ClusterCockpit/cc-lib/v2/ccMessage"
|
||||||
|
|
||||||
receivers "github.com/ClusterCockpit/cc-lib/v2/receivers"
|
|
||||||
lp2 "github.com/influxdata/line-protocol/v2/lineprotocol"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
const CUSTOMCMDPATH = `/home/unrz139/Work/cc-metric-collector/collectors/custom`
|
const CUSTOMCMDPATH = `/home/unrz139/Work/cc-metric-collector/collectors/custom`
|
||||||
@@ -100,92 +95,58 @@ func (m *CustomCmdCollector) Read(interval time.Duration, output chan lp.CCMessa
|
|||||||
// Execute configured commands
|
// Execute configured commands
|
||||||
for _, cmdFields := range m.cmdFieldsSlice {
|
for _, cmdFields := range m.cmdFieldsSlice {
|
||||||
command := exec.Command(cmdFields[0], cmdFields[1:]...)
|
command := exec.Command(cmdFields[0], cmdFields[1:]...)
|
||||||
stdout, err := command.StdoutPipe()
|
stdout, err := command.Output()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
cclog.ComponentError(
|
cclog.ComponentError(
|
||||||
m.name,
|
m.name,
|
||||||
fmt.Sprintf("Read(): Failed to create stdout pipe for command \"%s\": %v", command.String(), err),
|
fmt.Sprintf("Read(): Failed to read command output for command \"%s\": %v", command.String(), err),
|
||||||
)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
errBuf := new(bytes.Buffer)
|
|
||||||
command.Stderr = errBuf
|
|
||||||
|
|
||||||
// Start command
|
|
||||||
if err := command.Start(); err != nil {
|
|
||||||
cclog.ComponentError(
|
|
||||||
m.name,
|
|
||||||
fmt.Sprintf("Read(): Failed to start command \"%s\": %v", command.String(), err),
|
|
||||||
)
|
)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
// Read and decode influxDB line-protocol from command output
|
// Read and decode influxDB line-protocol from command output
|
||||||
d := lp2.NewDecoder(stdout)
|
metrics, err := lp.FromBytes(stdout)
|
||||||
for d.Next() {
|
if err != nil {
|
||||||
metric, err := receivers.DecodeInfluxMessage(d)
|
cclog.ComponentError(
|
||||||
if err != nil {
|
m.name,
|
||||||
cclog.ComponentError(
|
fmt.Sprintf("Read(): Failed to decode influx Message: %v", err),
|
||||||
m.name,
|
)
|
||||||
fmt.Sprintf("Read(): Failed to decode influx Message: %v", err),
|
continue
|
||||||
)
|
}
|
||||||
continue
|
for _, metric := range metrics {
|
||||||
}
|
|
||||||
if slices.Contains(m.config.ExcludeMetrics, metric.Name()) {
|
if slices.Contains(m.config.ExcludeMetrics, metric.Name()) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
output <- metric
|
output <- metric
|
||||||
}
|
}
|
||||||
|
|
||||||
// Wait for command end
|
|
||||||
if err := command.Wait(); err != nil {
|
|
||||||
errMsg, _ := io.ReadAll(errBuf)
|
|
||||||
cclog.ComponentError(
|
|
||||||
m.name,
|
|
||||||
fmt.Sprintf("Read(): Failed to wait for the end of command \"%s\": %v\n", command.String(), err),
|
|
||||||
)
|
|
||||||
cclog.ComponentError(
|
|
||||||
m.name,
|
|
||||||
fmt.Sprintf("Read(): command stderr: \"%s\"\n", strings.TrimSpace(string(errMsg))))
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Read configured files
|
// Read configured files
|
||||||
for _, filename := range m.files {
|
for _, filename := range m.files {
|
||||||
file, err := os.Open(filename)
|
input, err := os.ReadFile(filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
cclog.ComponentError(
|
cclog.ComponentError(
|
||||||
m.name,
|
m.name,
|
||||||
fmt.Sprintf("Read(): Failed to open file \"%s\": %v\n", filename, err),
|
fmt.Sprintf("Read(): Failed to read file \"%s\": %v\n", filename, err),
|
||||||
)
|
)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
// Read and decode influxDB line-protocol from file
|
// Read and decode influxDB line-protocol from file
|
||||||
d := lp2.NewDecoder(file)
|
metrics, err := lp.FromBytes(input)
|
||||||
for d.Next() {
|
if err != nil {
|
||||||
metric, err := receivers.DecodeInfluxMessage(d)
|
cclog.ComponentError(
|
||||||
if err != nil {
|
m.name,
|
||||||
cclog.ComponentError(
|
fmt.Sprintf("Read(): Failed to decode influx Message: %v", err),
|
||||||
m.name,
|
)
|
||||||
fmt.Sprintf("Read(): Failed to decode influx Message: %v", err),
|
continue
|
||||||
)
|
}
|
||||||
continue
|
for _, metric := range metrics {
|
||||||
}
|
|
||||||
if slices.Contains(m.config.ExcludeMetrics, metric.Name()) {
|
if slices.Contains(m.config.ExcludeMetrics, metric.Name()) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
output <- metric
|
output <- metric
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := file.Close(); err != nil {
|
|
||||||
cclog.ComponentError(
|
|
||||||
m.name,
|
|
||||||
fmt.Sprintf("Read(): Failed to close file \"%s\": %v\n", filename, err),
|
|
||||||
)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user