set.go

package cli

import (
	"context"
	"os"

	"code.fbi.h-da.de/danet/forks/goarista/gnmi"
	"code.fbi.h-da.de/danet/gosdn/nucleus"
	"code.fbi.h-da.de/danet/gosdn/nucleus/util/proto"
	pb "google.golang.org/protobuf/proto"
)

// Set sends a gNMI Set request to the specified target. Only one
// request per invocation supported.
func Set(a, u, p, typ string, args ...string) error {
	opts := &nucleus.GnmiTransportOptions{
		Config: gnmi.Config{
			Addr:     a,
			Username: u,
			Password: p,
		},
	}
	t, err := nucleus.NewGnmiTransport(opts)
	if err != nil {
		return err
	}

	path := gnmi.SplitPath(args[0])
	req := []interface{}{
		&gnmi.Operation{
			Type:   typ,
			Origin: "",
			Target: "",
			Path:   path,
			Val:    args[1],
		},
	}

	resp, err := t.Set(context.Background(), req...)
	if err != nil {
		return err
	}

	_, tap := os.LookupEnv("GOSDN_TAP")
	if tap {
		if err := proto.Write(resp.(pb.Message), "resp-set-system-config-hostname"); err != nil {
			return err
		}
	}
	return nil
}