1 files changed, 55 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/snmp/collect.go b/src/go/collectors/go.d.plugin/modules/snmp/collect.go
new file mode 100644
index 000000000..9f0e78d7e
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/snmp/collect.go
@@ -0,0 +1,55 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package snmp
+
+import (
+	"github.com/gosnmp/gosnmp"
+)
+
+func (s *SNMP) collect() (map[string]int64, error) {
+	collected := make(map[string]int64)
+
+	if err := s.collectOIDs(collected); err != nil {
+		return nil, err
+	}
+
+	return collected, nil
+}
+
+func (s *SNMP) collectOIDs(collected map[string]int64) error {
+	for i, end := 0, 0; i < len(s.oids); i += s.Options.MaxOIDs {
+		if end = i + s.Options.MaxOIDs; end > len(s.oids) {
+			end = len(s.oids)
+		}
+
+		oids := s.oids[i:end]
+		resp, err := s.snmpClient.Get(oids)
+		if err != nil {
+			s.Errorf("cannot get SNMP data: %v", err)
+			return err
+		}
+
+		for i, oid := range oids {
+			if i >= len(resp.Variables) {
+				continue
+			}
+
+			switch v := resp.Variables[i]; v.Type {
+			case gosnmp.Boolean,
+				gosnmp.Counter32,
+				gosnmp.Counter64,
+				gosnmp.Gauge32,
+				gosnmp.TimeTicks,
+				gosnmp.Uinteger32,
+				gosnmp.OpaqueFloat,
+				gosnmp.OpaqueDouble,
+				gosnmp.Integer:
+				collected[oid] = gosnmp.ToBigInt(v.Value).Int64()
+			default:
+				s.Debugf("skipping OID '%s' (unsupported type '%s')", oid, v.Type)
+			}
+		}
+	}
+
+	return nil
+}