subscribe.go

package cli

import (
	"code.fbi.h-da.de/cocsn/gosdn/forks/goarista/gnmi"
	"code.fbi.h-da.de/cocsn/gosdn/nucleus"
	"context"
	"fmt"
	"github.com/google/uuid"
	gpb "github.com/openconfig/gnmi/proto/gnmi"
	log "github.com/sirupsen/logrus"
	"os"
	"os/signal"
	"syscall"
	"time"
)

func Subscribe() error{
	log.SetLevel(log.DebugLevel)
	sbi := &nucleus.OpenConfig{}

	device, err := nucleus.NewDevice(sbi,
		&nucleus.GnmiTransportOptions{
			Config: gnmi.Config{
				Addr:     "portainer.danet.fbi.h-da.de:6030",
				Username: "admin",
				Password: "arista",
				Encoding: gpb.Encoding_JSON_IETF,
			},
			SetNode:  sbi.SetNode(),
			RespChan: make(chan *gpb.SubscribeResponse),
		})
	if err != nil {
		return err
	}
	pnd, err := nucleus.NewPND("openconfig", "a simple openconfig PND", uuid.New(), sbi)
	if err != nil {
		return err
	}
	if err := pnd.AddDevice(device); err != nil {
		return err
	}

	paths := []string{"/interfaces/interface/name"}

	opts := &gnmi.SubscribeOptions{
		UpdatesOnly:       false,
		Prefix:            "",
		Mode:              "stream",
		StreamMode:        "sample",
		SampleInterval:    uint64(10 * time.Second.Nanoseconds()),
		SuppressRedundant: false,
		HeartbeatInterval: uint64(time.Second.Nanoseconds()),
		Paths:             gnmi.SplitPaths(paths),
		Origin:            "",
		Target:            "portainer.danet.fbi.h-da.de:6030",
	}
	done := make(chan os.Signal, 1)
	signal.Notify(done, syscall.SIGILL, syscall.SIGTERM)
	ctx := context.WithValue(context.Background(), "opts", opts)
	go func() {
		if err := device.Transport.Subscribe(ctx); err != nil {
			log.Fatal(err)
		}
	}()
	fmt.Println("awaiting signal")
	<-done
	fmt.Println("exiting")
	return nil
}