netdata/netdata

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