wip for discovery via netbox

Author: i3149

go.mod

@@ -44,6 +44,7 @@ require (
 	github.com/liamg/furious v0.0.0-20191231090757-c295c872d6c1
 	github.com/linkedin/goavro/v2 v2.10.1
 	github.com/montanaflynn/stats v0.7.0
+	github.com/netbox-community/go-netbox/v4 v4.2.2-3
 	github.com/netsampler/goflow2/v2 v2.1.3
 	github.com/oschwald/geoip2-golang v1.9.0
 	github.com/pkg/errors v0.9.1

go.sum

@@ -717,6 +717,8 @@ github.com/montanaflynn/stats v0.7.0/go.mod h1:etXPPgVO6n31NxCd9KQUMvCM+ve0ruNzt
 github.com/mostlygeek/arp v0.0.0-20170424181311-541a2129847a h1:AfneHvfmYgUIcgdUrrDFklLdEzQAvG9AKRTe1x1mx/0=
 github.com/mostlygeek/arp v0.0.0-20170424181311-541a2129847a/go.mod h1:jZxafo9CAqaKFQE4zitrg5QNlA6CXUsjwXPlIppF3tk=
 github.com/ncw/swift v1.0.52/go.mod h1:23YIA4yWVnGwv2dQlN4bB7egfYX6YLn0Yo/S6zZO/ZM=
+github.com/netbox-community/go-netbox/v4 v4.2.2-3 h1:NrAkuI41ExiYa6p1o7F0NvxcS2qxsVJfymR5hAZglDs=
+github.com/netbox-community/go-netbox/v4 v4.2.2-3/go.mod h1:X7jbSuzejM0U5uy/ajXidRFEAshl6eKqrcinhMy4aAI=
 github.com/netsampler/goflow2/v2 v2.1.3 h1:glfeG2hIzFlAmEz236nZIAWi848AB+p1pJnuQqiyLkI=
 github.com/netsampler/goflow2/v2 v2.1.3/go.mod h1:94ZaxfHuUwG6KviCxMWuZIvz0UGu657StE/g83hlS8A=
 github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e/go.mod h1:zD1mROLANZcx1PVRCS0qkT7pwLkGfwJo4zjcN/Tysno=

pkg/inputs/snmp/disco.go

@@ -87,6 +87,14 @@ func Discover(ctx context.Context, log logger.ContextL, pollDuration time.Durati
 	}
 	defer mdb.Close()
 
+	if conf.Disco.NetboxAPIHost != "" {
+		err = getDevicesFromNetbox(ctx, conf, log)
+		if err != nil {
+			return nil, err
+		}
+		return nil, nil
+	}
+
 	ignoreMap := map[string]bool{}
 	for _, ip := range conf.Disco.IgnoreList {
 		ignoreMap[ip] = true

pkg/inputs/snmp/netbox.go

@@ -0,0 +1,32 @@
+package snmp
+
+/**
+Use netbox API to pull down list of devices to target. Get all devices matching a given tag.
+*/
+import (
+	"context"
+
+	"github.com/kentik/ktranslate/pkg/eggs/logger"
+	"github.com/kentik/ktranslate/pkg/kt"
+
+	"github.com/netbox-community/go-netbox/v4"
+)
+
+func getDevicesFromNetbox(ctx context.Context, conf *kt.SnmpConfig, log logger.ContextL) error {
+	c := netbox.NewAPIClientFor(conf.Disco.NetboxAPIHost, conf.Disco.NetboxAPIToken)
+
+	res, _, err := c.DcimAPI.
+		DcimDevicesList(ctx).
+		Status([]string{"active"}).
+		Limit(10).
+		Execute()
+
+	if err != nil {
+		return err
+	}
+
+	log.Infof("%v", res.Results)
+
+	return nil
+
+}

pkg/kt/snmp.go

@@ -258,6 +258,8 @@ type SnmpDiscoConfig struct {
 	CidrOrig           string          `yaml:"-"`
 	IgnoreOrig         string          `yaml:"-"`
 	NoUseBulkWalkAll   bool            `yaml:"no_use_bulkwalkall"`
+	NetboxAPIHost      string          `yaml:"netbox_host"`
+	NetboxAPIToken     string          `yaml:"netbox_token"`
 }
 
 type ProviderMap struct {