configurationmanagement.go

package server

import (
	"context"
	"encoding/json"
	"time"

	cmpb "code.fbi.h-da.de/danet/gosdn/api/go/gosdn/configurationmanagement"
	"code.fbi.h-da.de/danet/gosdn/controller/interfaces/networkdomain"
	"code.fbi.h-da.de/danet/gosdn/controller/interfaces/networkelement"
	"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"
	"github.com/google/uuid"
)

// ConfigurationManagementServer represents  ConfigurationManagementServer...
type ConfigurationManagementServer struct {
	cmpb.UnimplementedConfigurationManagementServiceServer
	pndStore        networkdomain.PndStore
	topologyService topology.Service
	nodeService     nodes.Service
	portService     ports.Service
}

// NewConfigurationManagementServer creates the ConfigurationManagementServer..
func NewConfigurationManagementServer(
	pndStore networkdomain.PndStore,
	topologyService topology.Service,
	nodeService nodes.Service, portService ports.Service,
) *ConfigurationManagementServer {
	return &ConfigurationManagementServer{
		pndStore:        pndStore,
		topologyService: topologyService,
		nodeService:     nodeService,
		portService:     portService}
}

// SDNConfig is used to aprse 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"`
	NetworkElements []networkelement.NetworkElement `json:"networkelements"`
}

// ExportSDNConfig returns the SDN configuration.
func (c ConfigurationManagementServer) ExportSDNConfig(ctx context.Context, request *cmpb.ExportSDNConfigRequest) (*cmpb.ExportSDNConfigResponse, error) {
	var sdnConfig = SDNConfig{}
	sdnConfig.PndID = request.Pid

	pndUUID := uuid.MustParse(request.Pid)
	pnd, err := c.pndStore.Get(store.Query{ID: pndUUID})
	if err != nil {
		return nil, err
	}

	sdnConfig.NetworkElements = pnd.NetworkElements()
	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
	}

	jsonSDNConfig, err := json.MarshalIndent(sdnConfig, "", "   ")
	if err != nil {
		return nil, err
	}

	stringified := string(jsonSDNConfig)

	return &cmpb.ExportSDNConfigResponse{
		Timestamp: time.Now().UnixNano(),
		SdnConfig: stringified}, nil
}

// ImportSDNConfig receives an SDN configuration and imports it.
func (s *NetworkElementServer) ImportSDNConfig(ctx context.Context, request *cmpb.ImportSDNConfigRequest) (*cmpb.ImportSDNConfigResponse, error) {
	return &cmpb.ImportSDNConfigResponse{
		Timestamp: time.Now().UnixNano()}, nil
}