gnmi.go

package main

import (
	"code.fbi.h-da.de/cocsn/gosdn/forks/goarista/gnmi"
	"code.fbi.h-da.de/cocsn/gosdn/nucleus"
	schema "code.fbi.h-da.de/cocsn/yang-models/generated/arista"
	"context"
	"github.com/google/uuid"
	log "github.com/sirupsen/logrus"
)

func main() {
	log.SetLevel(log.DebugLevel)
	sbi := &nucleus.AristaOC{}
	transport := &nucleus.Gnmi{SetNode: sbi.SetNode()}
	device := nucleus.Device{
		Device: &schema.Device{},
		SBI:    sbi,
		Config: nucleus.DeviceConfig{
			Uuid:     uuid.New(),
			Address:  "localhost:9339",
			Username: "admin",
			Password: "arista",
		},
		Transport: transport,
	}
	pnd := nucleus.NewPND("openconfig", sbi)
	if err := pnd.AddDevice(device); err != nil {
		log.Fatal(err)
	}

	cfg := &gnmi.Config{
		Addr:     device.Config.Address,
		Password: device.Config.Password,
		Username: device.Config.Username,
	}
	ctx := gnmi.NewContext(context.Background(), cfg)
	ctx = context.WithValue(ctx, "config", cfg)

	p := []string{"/interfaces/interface[name=en0]/state/name"}
	errors := 0
	for _, path := range p {
		req, err := gnmi.NewGetRequest(gnmi.SplitPaths([]string{path}), "")
		resp, err := nucleus.GetWithRequest(ctx, req)
		if err != nil {
			log.Debug(err)
			errors++
			break
		}
		if err := device.Add(resp); err != nil {
			log.Debug(err)
			errors++
		}
	}

	percentage := float64(errors) / float64(len(p)) * 100.0
	log.Debugf("%v errors", errors)
	log.Debugf("%v percent failed", percentage)
}