Skip to content
Snippets Groups Projects
grpc.go 8.1 KiB
Newer Older
  • Learn to ignore specific revisions
  • Manuel Kieweg's avatar
    Manuel Kieweg committed
    package api
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    
    import (
    	"context"
    	"errors"
    	"time"
    
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	pb "code.fbi.h-da.de/danet/api/go/gosdn/core"
    	ppb "code.fbi.h-da.de/danet/api/go/gosdn/pnd"
    	spb "code.fbi.h-da.de/danet/api/go/gosdn/southbound"
    	tpb "code.fbi.h-da.de/danet/api/go/gosdn/transport"
    	nbi "code.fbi.h-da.de/danet/gosdn/northbound/client"
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	"github.com/google/uuid"
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	log "github.com/sirupsen/logrus"
    	"github.com/spf13/viper"
    
    	"google.golang.org/grpc"
    
    Malte Bauch's avatar
    Malte Bauch committed
    	"google.golang.org/grpc/credentials/insecure"
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    )
    
    var dialOptions []grpc.DialOption
    
    func init() {
    	dialOptions = []grpc.DialOption{
    
    Malte Bauch's avatar
    Malte Bauch committed
    		grpc.WithTransportCredentials(insecure.NewCredentials()),
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	}
    }
    
    // Init initialises the CLI client.
    func Init(addr string) error {
    	ctx := context.Background()
    	resp, err := getAllCore(ctx, addr)
    	if err != nil {
    		return err
    	}
    	if len(resp.Pnd) > 0 {
    		pid := resp.Pnd[0].Id
    		viper.Set("CLI_PND", pid)
    		log.Infof("PND: %v", pid)
    
    		// if len(resp.Pnd[0].Sbi) != 0 {
    		// 	sbi := resp.Pnd[0].Sbi[0].Id
    		// 	viper.Set("CLI_SBI", sbi)
    		// 	log.Infof("SBI: %v", sbi)
    		// }
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	}
    
    	return viper.WriteConfig()
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    }
    
    // GetIds requests all UUID information from the controller
    func GetIds(addr string) ([]*ppb.PrincipalNetworkDomain, error) {
    	ctx := context.Background()
    	resp, err := getAllCore(ctx, addr)
    	if err != nil {
    		return nil, err
    	}
    	return resp.Pnd, nil
    }
    
    
    func getAllCore(ctx context.Context, addr string) (*pb.GetPndListResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	coreClient, err := nbi.CoreClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    	}
    
    	req := &pb.GetPndListRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Timestamp: time.Now().UnixNano(),
    	}
    
    	return coreClient.GetPndList(ctx, req)
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    }
    
    // AddPnd takes a name, description and SBI UUID to create a new
    // PrincipalNetworkDomain on the controller
    
    func AddPnd(addr, name, description, sbi string) (*pb.CreatePndListResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	coreClient, err := nbi.CoreClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    	}
    	ctx := context.Background()
    
    	req := &pb.CreatePndListRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Timestamp: time.Now().UnixNano(),
    
    		Pnd: []*pb.PndCreateProperties{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    			{
    				Name:        name,
    				Description: description,
    				Sbi:         sbi,
    			},
    		},
    	}
    
    
    	return coreClient.CreatePndList(ctx, req)
    
    // GetPnd requests one PrincipalNetworkDomain from the
    // controller.
    func GetPnd(addr string, args ...string) (*pb.GetPndResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	coreClient, err := nbi.CoreClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    	}
    	if len(args) <= 0 {
    		return nil, errors.New("not enough arguments")
    	}
    	ctx := context.Background()
    
    	req := &pb.GetPndRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Timestamp: time.Now().UnixNano(),
    		Pid:       args,
    	}
    
    	return coreClient.GetPnd(ctx, req)
    }
    
    // GetPnds requests all PrincipalNetworkDomains from the
    // controller.
    func GetPnds(addr string, args ...string) (*pb.GetPndListResponse, error) {
    	coreClient, err := nbi.CoreClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    	}
    	if len(args) <= 0 {
    		return nil, errors.New("not enough arguments")
    	}
    	ctx := context.Background()
    	req := &pb.GetPndListRequest{
    		Timestamp: time.Now().UnixNano(),
    	}
    	return coreClient.GetPndList(ctx, req)
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    // getChanges requests all pending and unconfirmed changes from the controller
    
    func getChanges(addr, pnd string) (*ppb.GetChangeListResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	ctx := context.Background()
    	client, err := nbi.PndClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    	}
    
    	req := &ppb.GetChangeListRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Timestamp: time.Now().UnixNano(),
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	}
    
    	return client.GetChangeList(ctx, req)
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    // commit sends a commit request for one or multiple changes to the
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    // controller.
    
    func commit(addr, pnd string, cuids ...string) (*ppb.SetChangeListResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	changes := make([]*ppb.SetChange, len(cuids))
    	for i, arg := range cuids {
    		changes[i] = &ppb.SetChange{
    			Cuid: arg,
    
    			Op:   ppb.Operation_OPERATION_COMMIT,
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		}
    	}
    	return commitConfirm(addr, pnd, changes)
    }
    
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    // confirm sends a confirm request for one or multiple changes to the
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    // controller
    
    func confirm(addr, pnd string, cuids ...string) (*ppb.SetChangeListResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	changes := make([]*ppb.SetChange, len(cuids))
    	for i, arg := range cuids {
    		changes[i] = &ppb.SetChange{
    			Cuid: arg,
    
    			Op:   ppb.Operation_OPERATION_CONFIRM,
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		}
    	}
    	return commitConfirm(addr, pnd, changes)
    }
    
    
    func commitConfirm(addr, pnd string, changes []*ppb.SetChange) (*ppb.SetChangeListResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	ctx := context.Background()
    	client, err := nbi.PndClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    	}
    
    	req := &ppb.SetChangeListRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Timestamp: time.Now().UnixNano(),
    		Change:    changes,
    		Pid:       pnd,
    	}
    
    	return client.SetChangeList(ctx, req)
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    // addDevice adds a new device to the controller. The device name is optional.
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    // If no name is provided a name will be generated upon device creation.
    
    func addDevice(addr, deviceName string, opt *tpb.TransportOption, sid, pid uuid.UUID) (*ppb.SetOndListResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	pndClient, err := nbi.PndClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    	}
    
    
    	req := &ppb.SetOndListRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Timestamp: time.Now().UnixNano(),
    		Ond: []*ppb.SetOnd{
    			{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    				Address: opt.GetAddress(),
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    				Sbi: &spb.SouthboundInterface{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    					Id: sid.String(),
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    				},
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    				DeviceName:      deviceName,
    				TransportOption: opt,
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    			},
    		},
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Pid: pid.String(),
    	}
    
    Malte Bauch's avatar
    Malte Bauch committed
    	switch t := opt.Type; t {
    
    	case spb.Type_TYPE_CONTAINERISED, spb.Type_TYPE_PLUGIN:
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		req.Ond[0].Sbi.Id = uuid.Nil.String()
    
    Malte Bauch's avatar
    Malte Bauch committed
    		req.Ond[0].Sbi.Type = t
    		req.Ond[0].TransportOption.Type = t
    	default:
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	}
    	ctx := context.Background()
    
    	return pndClient.SetOndList(ctx, req)
    
    // getDevice requests one device belonging to a given
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    // PrincipalNetworkDomain from the controller. If no device identifier
    
    // is provided, an error is thrown.
    func getDevice(addr, pid string, did ...string) (*ppb.GetOndResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	pndClient, err := nbi.PndClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    	}
    
    	if len(did) == 0 {
    
    	req := &ppb.GetOndRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Timestamp: time.Now().UnixNano(),
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	}
    	ctx := context.Background()
    
    	return pndClient.GetOnd(ctx, req)
    
    Malte Bauch's avatar
    Malte Bauch committed
    //nolint
    // NOTE: currently not in use, but could be of value later
    
    // getDevice requests all devices belonging to a given
    // PrincipalNetworkDomain from the controller.
    func getDevices(addr, pid string) (*ppb.GetOndListResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	pndClient, err := nbi.PndClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	}
    
    	req := &ppb.GetOndListRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Timestamp: time.Now().UnixNano(),
    
    		Pid:       pid,
    	}
    	ctx := context.Background()
    	return pndClient.GetOndList(ctx, req)
    }
    
    func getPath(addr, pid, did, path string) (*ppb.GetPathResponse, error) {
    	pndClient, err := nbi.PndClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    	}
    
    	req := &ppb.GetPathRequest{
    		Timestamp: time.Now().UnixNano(),
    		Did:       did,
    		Pid:       pid,
    		Path:      path,
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	}
    	ctx := context.Background()
    
    	return pndClient.GetPath(ctx, req)
    
    func deleteDevice(addr, pid, did string) (*ppb.DeleteOndResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	pndClient, err := nbi.PndClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	}
    
    	req := &ppb.DeleteOndRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Timestamp: time.Now().UnixNano(),
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Pid:       pid,
    	}
    	ctx := context.Background()
    
    	return pndClient.DeleteOnd(ctx, req)
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    // change creates a ChangeRequest for the specified OND. ApiOperations are
    // used to specify the type of the change (update, replace, delete as specified
    // in https://github.com/openconfig/reference/blob/master/rpc/gnmi/gnmi-specification.md#34-modifying-state)
    // For delete operations the value field needs to contain an empty string.
    
    func changeRequest(addr, did, pid, path, value string, op ppb.ApiOperation) (*ppb.SetPathListResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	req := &ppb.ChangeRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Path:  path,
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Value: value,
    		ApiOp: op,
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	}
    	return sendChangeRequest(addr, pid, req)
    }
    
    
    func sendChangeRequest(addr, pid string, req *ppb.ChangeRequest) (*ppb.SetPathListResponse, error) {
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    	pndClient, err := nbi.PndClient(addr, dialOptions...)
    	if err != nil {
    		return nil, err
    	}
    	ctx := context.Background()
    
    	r := &ppb.SetPathListRequest{
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    		Timestamp:     time.Now().UnixNano(),
    		ChangeRequest: []*ppb.ChangeRequest{req},
    		Pid:           pid,
    	}
    
    	return pndClient.SetPathList(ctx, r)
    
    Manuel Kieweg's avatar
    Manuel Kieweg committed
    }