1 files changed, 89 insertions, 0 deletions
diff --git a/src/go/plugin/go.d/modules/dovecot/collect.go b/src/go/plugin/go.d/modules/dovecot/collect.go
new file mode 100644
index 000000000..a93bfc811
--- /dev/null
+++ b/src/go/plugin/go.d/modules/dovecot/collect.go
@@ -0,0 +1,89 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package dovecot
+
+import (
+	"bufio"
+	"bytes"
+	"errors"
+	"fmt"
+	"strconv"
+	"strings"
+)
+
+// FIXME: drop using "old_stats" in favour of "stats" (https://doc.dovecot.org/configuration_manual/stats/openmetrics/).
+
+func (d *Dovecot) collect() (map[string]int64, error) {
+	if d.conn == nil {
+		conn, err := d.establishConn()
+		if err != nil {
+			return nil, err
+		}
+		d.conn = conn
+	}
+
+	stats, err := d.conn.queryExportGlobal()
+	if err != nil {
+		d.conn.disconnect()
+		d.conn = nil
+		return nil, err
+	}
+
+	mx := make(map[string]int64)
+
+	// https://doc.dovecot.org/configuration_manual/stats/old_statistics/#statistics-gathered
+	if err := d.collectExportGlobal(mx, stats); err != nil {
+		return nil, err
+	}
+
+	return mx, nil
+}
+
+func (d *Dovecot) collectExportGlobal(mx map[string]int64, resp []byte) error {
+	sc := bufio.NewScanner(bytes.NewReader(resp))
+
+	if !sc.Scan() {
+		return errors.New("failed to read fields line from export global response")
+	}
+	fieldsLine := strings.TrimSpace(sc.Text())
+
+	if !sc.Scan() {
+		return errors.New("failed to read values line from export global response")
+	}
+	valuesLine := strings.TrimSpace(sc.Text())
+
+	if fieldsLine == "" || valuesLine == "" {
+		return errors.New("empty fields line or values line from export global response")
+	}
+
+	fields := strings.Fields(fieldsLine)
+	values := strings.Fields(valuesLine)
+
+	if len(fields) != len(values) {
+		return fmt.Errorf("mismatched fields and values count: fields=%d, values=%d", len(fields), len(values))
+	}
+
+	for i, name := range fields {
+		val := values[i]
+
+		v, err := strconv.ParseInt(val, 10, 64)
+		if err != nil {
+			d.Debugf("failed to parse export value %s %s: %v", name, val, err)
+			continue
+		}
+
+		mx[name] = v
+	}
+
+	return nil
+}
+
+func (d *Dovecot) establishConn() (dovecotConn, error) {
+	conn := d.newConn(d.Config)
+
+	if err := conn.connect(); err != nil {
+		return nil, err
+	}
+
+	return conn, nil
+}