1 files changed, 65 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/openvpn_status_log/collect.go b/src/go/collectors/go.d.plugin/modules/openvpn_status_log/collect.go
new file mode 100644
index 000000000..f6a442fd5
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/openvpn_status_log/collect.go
@@ -0,0 +1,65 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package openvpn_status_log
+
+import (
+	"time"
+)
+
+func (o *OpenVPNStatusLog) collect() (map[string]int64, error) {
+	clients, err := parse(o.LogPath)
+	if err != nil {
+		return nil, err
+	}
+
+	mx := make(map[string]int64)
+
+	collectTotalStats(mx, clients)
+
+	if o.perUserMatcher != nil && numOfClients(clients) > 0 {
+		o.collectUsers(mx, clients)
+	}
+
+	return mx, nil
+}
+
+func collectTotalStats(mx map[string]int64, clients []clientInfo) {
+	var in, out int64
+	for _, c := range clients {
+		in += c.bytesReceived
+		out += c.bytesSent
+	}
+	mx["clients"] = numOfClients(clients)
+	mx["bytes_in"] = in
+	mx["bytes_out"] = out
+}
+
+func (o *OpenVPNStatusLog) collectUsers(mx map[string]int64, clients []clientInfo) {
+	now := time.Now().Unix()
+
+	for _, user := range clients {
+		name := user.commonName
+		if !o.perUserMatcher.MatchString(name) {
+			continue
+		}
+		if !o.collectedUsers[name] {
+			o.collectedUsers[name] = true
+			if err := o.addUserCharts(name); err != nil {
+				o.Warning(err)
+			}
+		}
+		mx[name+"_bytes_in"] = user.bytesReceived
+		mx[name+"_bytes_out"] = user.bytesSent
+		mx[name+"_connection_time"] = now - user.connectedSince
+	}
+}
+
+func numOfClients(clients []clientInfo) int64 {
+	var num int64
+	for _, v := range clients {
+		if v.commonName != "" && v.commonName != "UNDEF" {
+			num++
+		}
+	}
+	return num
+}