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
+}