move netbox feed to graph
diff --git a/main.go b/main.go
index ce79ca7..b97e171 100644
--- a/main.go
+++ b/main.go
@@ -1,13 +1,12 @@
package main
import (
+ "context"
"flag"
"io/ioutil"
- //"github.com/digitalocean/go-netbox/netbox"
- //"github.com/digitalocean/go-netbox/netbox/client"
- //"github.com/digitalocean/go-netbox/netbox/client/dcim"
-
+ "github.com/digitalocean/go-netbox/netbox"
+ "github.com/digitalocean/go-netbox/netbox/client"
"github.com/golang/glog"
"github.com/golang/protobuf/proto"
@@ -27,10 +26,12 @@
func main() {
flag.StringVar(&flagConfigPath, "config_path", "./topo.pb.text", "Text proto configuration of Topo (per config.proto)")
- flag.StringVar(&flagNetboxHost, "netbox_host", "nebtox.bgp.wtf", "Netbox host")
+ flag.StringVar(&flagNetboxHost, "netbox_host", "netbox.bgp.wtf", "Netbox host")
flag.StringVar(&flagNetboxAPIKey, "netbox_api_key", "", "Netbox API key")
flag.Parse()
+ ctx := context.Background()
+
data, err := ioutil.ReadFile(flagConfigPath)
if err != nil {
glog.Exitf("Could not read config: %v", err)
@@ -45,12 +46,10 @@
glog.Exitf("Initial config load failed: %v", err)
}
- //client.DefaultSchemes = []string{"https"}
- //nb := netbox.NewNetboxWithAPIKey(flagNetboxHost, flagNetboxAPIKey)
- //req := &dcim.DcimInterfaceConnectionsListParams{
- // Device: swag.String("bc01n01"),
- // Context: context.Background(),
- //}
- //res, err := nb.Dcim.DcimInterfaceConnectionsList(req, nil)
- //glog.Infof("%+v, %v", res, err)
+ client.DefaultSchemes = []string{"https"}
+ nb := netbox.NewNetboxWithAPIKey(flagNetboxHost, flagNetboxAPIKey)
+ err = gr.FeedFromNetbox(ctx, nb)
+ if err != nil {
+ glog.Exitf("Initial netbox feed failed: %v", err)
+ }
}