Skip to content
Snippets Groups Projects
configurationmanagement.go 6.67 KiB
Newer Older
  • Learn to ignore specific revisions
  • Neil Schark's avatar
    wip
    Neil Schark committed
    package server
    
    import (
    	"context"
    
    Neil Schark's avatar
    Neil Schark committed
    	"encoding/json"
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    	"time"
    
    
    Neil Schark's avatar
    Neil Schark committed
    	cmpb "code.fbi.h-da.de/danet/gosdn/api/go/gosdn/configurationmanagement"
    
    	spb "code.fbi.h-da.de/danet/gosdn/api/go/gosdn/southbound"
    
    Neil Schark's avatar
    Neil Schark committed
    	"code.fbi.h-da.de/danet/gosdn/controller/interfaces/networkdomain"
    
    Neil Schark's avatar
    Neil Schark committed
    	"code.fbi.h-da.de/danet/gosdn/controller/interfaces/networkelement"
    
    	"code.fbi.h-da.de/danet/gosdn/controller/interfaces/southbound"
    
    	"code.fbi.h-da.de/danet/gosdn/controller/nucleus"
    
    Neil Schark's avatar
    Neil Schark committed
    	"code.fbi.h-da.de/danet/gosdn/controller/store"
    	"code.fbi.h-da.de/danet/gosdn/controller/topology"
    	"code.fbi.h-da.de/danet/gosdn/controller/topology/links"
    	"code.fbi.h-da.de/danet/gosdn/controller/topology/nodes"
    	"code.fbi.h-da.de/danet/gosdn/controller/topology/ports"
    
    
    	tpb "code.fbi.h-da.de/danet/gosdn/api/go/gosdn/transport"
    
    Neil Schark's avatar
    Neil Schark committed
    	"github.com/google/uuid"
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    )
    
    
    Neil Schark's avatar
    Neil Schark committed
    // ConfigurationManagementServer represents  ConfigurationManagementServer...
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    type ConfigurationManagementServer struct {
    
    Neil Schark's avatar
    Neil Schark committed
    	cmpb.UnimplementedConfigurationManagementServiceServer
    
    Neil Schark's avatar
    Neil Schark committed
    	pndStore        networkdomain.PndStore
    	topologyService topology.Service
    	nodeService     nodes.Service
    	portService     ports.Service
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    }
    
    
    Neil Schark's avatar
    Neil Schark committed
    // NewConfigurationManagementServer creates the ConfigurationManagementServer..
    
    Neil Schark's avatar
    Neil Schark committed
    func NewConfigurationManagementServer(
    	pndStore networkdomain.PndStore,
    	topologyService topology.Service,
    
    	nodeService nodes.Service,
    	portService ports.Service,
    
    Neil Schark's avatar
    Neil Schark committed
    ) *ConfigurationManagementServer {
    	return &ConfigurationManagementServer{
    		pndStore:        pndStore,
    		topologyService: topologyService,
    		nodeService:     nodeService,
    		portService:     portService}
    }
    
    
    // sdnConfig is used to parse the sdnConfig into JSON.
    type sdnConfig struct {
    
    	PndID           string                           `json:"pndID"`
    	Nodes           []nodes.Node                     `json:"nodes"`
    	Ports           []ports.Port                     `json:"ports"`
    	Links           []links.Link                     `json:"links"`
    	Sbis            []southbound.SouthboundInterface `json:"sbis"`
    	NetworkElements []networkelement.NetworkElement  `json:"networkelements"`
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    }
    
    
    // loadedSDNConfig is used to parse the stringified JSON sdnConfig into objects.
    type loadedSDNConfig struct {
    	PndID           string                                `json:"pndID"`
    	Nodes           []nodes.Node                          `json:"nodes"`
    	Ports           []ports.Port                          `json:"ports"`
    	Links           []links.Link                          `json:"links"`
    	Sbis            []southbound.LoadedSbi                `json:"sbis"`
    	NetworkElements []networkelement.LoadedNetworkElement `json:"networkelements"`
    }
    
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    // ExportSDNConfig returns the SDN configuration.
    
    Neil Schark's avatar
    Neil Schark committed
    func (c ConfigurationManagementServer) ExportSDNConfig(ctx context.Context, request *cmpb.ExportSDNConfigRequest) (*cmpb.ExportSDNConfigResponse, error) {
    
    	var sdnConfig = sdnConfig{}
    
    Neil Schark's avatar
    Neil Schark committed
    	sdnConfig.PndID = request.Pid
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    
    
    Neil Schark's avatar
    Neil Schark committed
    	pndUUID := uuid.MustParse(request.Pid)
    	pnd, err := c.pndStore.Get(store.Query{ID: pndUUID})
    	if err != nil {
    		return nil, err
    	}
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    
    
    Neil Schark's avatar
    Neil Schark committed
    	sdnConfig.NetworkElements = pnd.NetworkElements()
    
    	sdnConfig.Sbis, err = pnd.GetSBIs()
    	if err != nil {
    		return nil, err
    	}
    
    Neil Schark's avatar
    Neil Schark committed
    	sdnConfig.Nodes, err = c.nodeService.GetAll()
    	if err != nil {
    		return nil, err
    	}
    	sdnConfig.Ports, err = c.portService.GetAll()
    	if err != nil {
    		return nil, err
    	}
    	sdnConfig.Links, err = c.topologyService.GetAll()
    	if err != nil {
    		return nil, err
    	}
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    
    
    	jsonSDNConfig, err := json.MarshalIndent(sdnConfig, "", "   ")
    
    Neil Schark's avatar
    Neil Schark committed
    	if err != nil {
    		return nil, err
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    	}
    
    
    Neil Schark's avatar
    Neil Schark committed
    	sdnConfigDataString := string(jsonSDNConfig)
    
    Neil Schark's avatar
    Neil Schark committed
    	return &cmpb.ExportSDNConfigResponse{
    
    Neil Schark's avatar
    Neil Schark committed
    		Timestamp:     time.Now().UnixNano(),
    		SdnConfigData: sdnConfigDataString}, nil
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    }
    
    // ImportSDNConfig receives an SDN configuration and imports it.
    
    func (c ConfigurationManagementServer) ImportSDNConfig(ctx context.Context, request *cmpb.ImportSDNConfigRequest) (*cmpb.ImportSDNConfigResponse, error) {
    	var sdnConfig = loadedSDNConfig{}
    
    	err := json.Unmarshal([]byte(request.SdnConfigData), &sdnConfig)
    	if err != nil {
    		return nil, err
    	}
    
    	//TODO: Clear DB
    
    
    Neil Schark's avatar
    Neil Schark committed
    	c.createElementsFromSDNConfig(&sdnConfig, request)
    
    Neil Schark's avatar
    Neil Schark committed
    	if err != nil {
    		return nil, err
    	}
    
    	return &cmpb.ImportSDNConfigResponse{
    		Timestamp: time.Now().UnixNano()}, nil
    }
    
    
    Neil Schark's avatar
    Neil Schark committed
    func (c ConfigurationManagementServer) createElementsFromSDNConfig(sdnConfig *loadedSDNConfig, request *cmpb.ImportSDNConfigRequest) error {
    
    	pndUUID := uuid.MustParse(request.Pid)
    	pnd, err := c.pndStore.Get(store.Query{ID: pndUUID})
    	if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    		return err
    
    	}
    
    	for _, inputSBI := range sdnConfig.Sbis {
    		sbi, err := nucleus.NewSBI(inputSBI.Type, uuid.MustParse(inputSBI.ID))
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    		err = pnd.AddSbi(sbi)
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    	}
    
    	for _, inputNode := range sdnConfig.Nodes {
    		node := nodes.Node{
    			ID:   inputNode.ID,
    			Name: inputNode.Name,
    		}
    		_, err = c.nodeService.EnsureExists(node)
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    	}
    
    	for _, inputPort := range sdnConfig.Ports {
    		port := ports.Port{
    			ID:            inputPort.ID,
    			Name:          inputPort.Name,
    			Configuration: inputPort.Configuration,
    		}
    		_, err = c.portService.EnsureExists(port)
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    	}
    
    	for _, inputPort := range sdnConfig.Links {
    		sourceNode, err := c.nodeService.Get(store.Query{ID: inputPort.SourceNode.ID})
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    		targetNode, err := c.nodeService.Get(store.Query{ID: inputPort.TargetNode.ID})
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    		sourcePort, err := c.portService.Get(store.Query{ID: inputPort.SourcePort.ID})
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    		targetPort, err := c.portService.Get(store.Query{ID: inputPort.TargetPort.ID})
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    		link := links.Link{
    			ID:         inputPort.ID,
    			Name:       inputPort.Name,
    			SourceNode: sourceNode,
    			TargetNode: targetNode,
    			SourcePort: sourcePort,
    			TargetPort: targetPort,
    		}
    		err = c.topologyService.AddLink(link)
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    	}
    
    	for _, inputNetworkElement := range sdnConfig.NetworkElements {
    		transportOption := tpb.TransportOption{
    			Address:  inputNetworkElement.TransportAddress,
    			Username: inputNetworkElement.TransportUsername,
    			Password: inputNetworkElement.TransportPassword,
    			TransportOption: &tpb.TransportOption_GnmiTransportOption{
    				GnmiTransportOption: &tpb.GnmiTransportOption{},
    			},
    			Type: spb.Type_TYPE_OPENCONFIG,
    		}
    		_, err := pnd.AddNetworkElement(
    			inputNetworkElement.Name,
    			&transportOption,
    			uuid.MustParse(inputNetworkElement.SBI),
    			uuid.MustParse(inputNetworkElement.ID),
    		)
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    
    		networkelement, err := pnd.GetNetworkElement(inputNetworkElement.ID)
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    		}
    
    		err = pnd.UpdateNetworkElement(networkelement, inputNetworkElement.Model)
    		if err != nil {
    
    Neil Schark's avatar
    Neil Schark committed
    			return err
    
    Neil Schark's avatar
    Neil Schark committed
    	return nil
    
    Neil Schark's avatar
    wip
    Neil Schark committed
    }