diff options
Diffstat (limited to '')
-rw-r--r-- | src/go/collectors/go.d.plugin/modules/pika/collect.go | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/pika/collect.go b/src/go/collectors/go.d.plugin/modules/pika/collect.go new file mode 100644 index 000000000..72a4961dd --- /dev/null +++ b/src/go/collectors/go.d.plugin/modules/pika/collect.go @@ -0,0 +1,71 @@ +// SPDX-License-Identifier: GPL-3.0-or-later + +package pika + +import ( + "bufio" + "context" + "errors" + "fmt" + "regexp" + "strings" + + "github.com/blang/semver/v4" +) + +const precision = 1000 // float values multiplier and dimensions divisor + +func (p *Pika) collect() (map[string]int64, error) { + info, err := p.pdb.Info(context.Background(), "all").Result() + if err != nil { + return nil, err + } + + if p.server == "" { + s, v, err := extractServerVersion(info) + if err != nil { + return nil, fmt.Errorf("can not extract server app and version: %v", err) + } + p.server, p.version = s, v + p.Debugf(`server="%s",version="%s"`, s, v) + } + + if p.server != "pika" { + return nil, fmt.Errorf("unsupported server app, want=pika, got=%s", p.server) + } + + ms := make(map[string]int64) + p.collectInfo(ms, info) + + return ms, nil +} + +// pika_version:3.4.0 +var reVersion = regexp.MustCompile(`([a-z]+)_version:(\d+\.\d+\.\d+)`) + +func extractServerVersion(info string) (string, *semver.Version, error) { + var versionLine string + for sc := bufio.NewScanner(strings.NewReader(info)); sc.Scan(); { + line := sc.Text() + if strings.Contains(line, "_version") { + versionLine = strings.TrimSpace(line) + break + } + } + if versionLine == "" { + return "", nil, errors.New("no version property") + } + + match := reVersion.FindStringSubmatch(versionLine) + if match == nil { + return "", nil, fmt.Errorf("can not parse version property '%s'", versionLine) + } + + server, version := match[1], match[2] + ver, err := semver.New(version) + if err != nil { + return "", nil, err + } + + return server, ver, nil +} |