assignForwardingHandler.go

package kmsHandler

import (
	"fmt"

	"code.fbi.h-da.de/danet/gnmi-target/handler"
	"code.fbi.h-da.de/danet/quant/goKMS/kms"
	gnmitargetygot "code.fbi.h-da.de/danet/quant/goKMS/model"
	"github.com/openconfig/gnmi/proto/gnmi"
	"github.com/openconfig/ygot/ygot"
	log "github.com/sirupsen/logrus"
)

// AssignForwardingHandler is the implementation of a gnmitarget.PathHandler.
type AssignForwardingHandler struct {
	handler.DefaultPathHandler
	kms *kms.KMS
}

func NewAssignForwardingHandler(kms *kms.KMS) *AssignForwardingHandler {
	return &AssignForwardingHandler{
		DefaultPathHandler: handler.DefaultPathHandler{
			Name: "kms-assign-forwarding-handler",
			Paths: map[string]struct{}{
				"/assign-forwarding": {},
			},
		},
		kms: kms,
	}
}

func (yh *AssignForwardingHandler) Init(config *handler.Config, publishToSubsFunc func([]*gnmi.Notification) error) error {
	// A initialization is not needed for this handler. The default startup has
	// no assign forwarding applied.
	yh.Config = config
	yh.PublishToSubs = publishToSubsFunc

	return nil
}

func (yh *AssignForwardingHandler) Update(c ygot.ValidatedGoStruct, jobs []*gnmi.Update) error {
	config, ok := c.(*gnmitargetygot.Gnmitarget)
	if !ok {
		return fmt.Errorf("failed type assertion for newConfig %T", (*gnmitargetygot.Gnmitarget)(nil))
	}

	forwarding := config.GetOrCreateAssignForwarding()

	pathId := forwarding.GetPathId()
	nextHop := forwarding.GetNextHop()
	prevHop := forwarding.GetPrevHop()

	var nextHopString string
	var prevHopString string
	var initiatingKmsAddressString string

	if nextHop != nil {
		nextHopString = nextHop.GetNodeId()
	}
	if prevHop != nil {
		prevHopString = prevHop.GetNodeId()
	}

	var initKMS *kms.RemoteKMS
	// check if initiating kms address is set
	if initiatingKmsAddress := forwarding.GetInitiatingKmsAddress(); initiatingKmsAddress != nil {
		addressPrefix := ""
		if initiatingKmsAddress.GetIpAddress() != "" {
			addressPrefix = initiatingKmsAddress.GetIpAddress()
		} else if initiatingKmsAddress.GetHostname() != "" {
			addressPrefix = initiatingKmsAddress.GetHostname()
		} else {
			log.Error("initiatingKmsAddress must have either an IP address or a hostname")
			return fmt.Errorf("initiatingKmsAddress must have either an IP address or a hostname")
		}
		initiatingKmsAddressString = fmt.Sprintf("%s:%d", addressPrefix, initiatingKmsAddress.GetPort())
		initKMS = &kms.RemoteKMS{
			Id:      initiatingKmsAddress.GetNodeId(),
			Address: initiatingKmsAddressString,
			Port:    initiatingKmsAddress.GetPort(),
		}
	}

	if err := yh.kms.AssignForwardingRoute(pathId, prevHopString, nextHopString, initKMS); err != nil {
		return err
	}

	return nil
}