[icn/sdwan.git] /

// SPDX-License-Identifier: Apache-2.0\r
// Copyright (c) 2020 Intel Corporation\r
\r
package controller\r
\r
import (\r
        "encoding/json"\r
\r
        "github.com/open-ness/EMCO/src/orchestrator/pkg/infra/db"\r
        log "github.com/open-ness/EMCO/src/orchestrator/pkg/infra/logutils"\r
        rpc "github.com/open-ness/EMCO/src/orchestrator/pkg/infra/rpc"\r
        mtypes "github.com/open-ness/EMCO/src/orchestrator/pkg/module/types"\r
        pkgerrors "github.com/pkg/errors"\r
)\r
\r
// Controller contains the parameters needed for Controllers\r
// It implements the interface for managing the Controllers\r
type Controller struct {\r
        Metadata mtypes.Metadata `json:"metadata"`\r
        Spec     ControllerSpec  `json:"spec"`\r
}\r
\r
type ControllerSpec struct {\r
        Host     string `json:"host"`\r
        Port     int    `json:"port"`\r
        Type     string `json:"type"`\r
        Priority int    `json:"priority"`\r
}\r
\r
const MinControllerPriority = 1\r
const MaxControllerPriority = 1000000\r
const CONTROLLER_TYPE_ACTION string = "action"\r
const CONTROLLER_TYPE_PLACEMENT string = "placement"\r
\r
var CONTROLLER_TYPES = [...]string{CONTROLLER_TYPE_ACTION, CONTROLLER_TYPE_PLACEMENT}\r
\r
// ControllerKey is the key structure that is used in the database\r
type ControllerKey struct {\r
        ControllerName string `json:"controller-name"`\r
}\r
\r
// We will use json marshalling to convert to string to\r
// preserve the underlying structure.\r
func (mk ControllerKey) String() string {\r
        out, err := json.Marshal(mk)\r
        if err != nil {\r
                return ""\r
        }\r
\r
        return string(out)\r
}\r
\r
// ControllerManager is an interface exposes the Controller functionality\r
type ControllerManager interface {\r
        CreateController(ms Controller, mayExist bool) (Controller, error)\r
        GetController(name string) (Controller, error)\r
        GetControllers() ([]Controller, error)\r
        InitControllers()\r
        DeleteController(name string) error\r
}\r
\r
// ControllerClient implements the Manager\r
// It will also be used to maintain some localized state\r
type ControllerClient struct {\r
        collectionName string\r
        tagMeta        string\r
}\r
\r
// NewControllerClient returns an instance of the ControllerClient\r
// which implements the Manager\r
func NewControllerClient() *ControllerClient {\r
        return &ControllerClient{\r
                collectionName: "controller",\r
                tagMeta:        "controllermetadata",\r
        }\r
}\r
\r
// CreateController a new collection based on the Controller\r
func (mc *ControllerClient) CreateController(m Controller, mayExist bool) (Controller, error) {\r
\r
        //Construct the composite key to select the entry\r
        key := ControllerKey{\r
                ControllerName: m.Metadata.Name,\r
        }\r
\r
        //Check if this Controller already exists\r
        _, err := mc.GetController(m.Metadata.Name)\r
        if err == nil && !mayExist {\r
                return Controller{}, pkgerrors.New("Controller already exists")\r
        }\r
\r
        err = db.DBconn.Insert(mc.collectionName, key, nil, mc.tagMeta, m)\r
        if err != nil {\r
                return Controller{}, pkgerrors.Wrap(err, "Creating DB Entry")\r
        }\r
\r
        // send message to create/update the  rpc connection\r
        rpc.UpdateRpcConn(m.Metadata.Name, m.Spec.Host, m.Spec.Port)\r
\r
        return m, nil\r
}\r
\r
// GetController returns the Controller for corresponding name\r
func (mc *ControllerClient) GetController(name string) (Controller, error) {\r
\r
        //Construct the composite key to select the entry\r
        key := ControllerKey{\r
                ControllerName: name,\r
        }\r
        value, err := db.DBconn.Find(mc.collectionName, key, mc.tagMeta)\r
        if err != nil {\r
                return Controller{}, pkgerrors.Wrap(err, "Get Controller")\r
        }\r
\r
        if value != nil {\r
                microserv := Controller{}\r
                err = db.DBconn.Unmarshal(value[0], &microserv)\r
                if err != nil {\r
                        return Controller{}, pkgerrors.Wrap(err, "Unmarshaling Value")\r
                }\r
                return microserv, nil\r
        }\r
\r
        return Controller{}, pkgerrors.New("Error getting Controller")\r
}\r
\r
// GetControllers returns all the  Controllers that are registered\r
func (mc *ControllerClient) GetControllers() ([]Controller, error) {\r
\r
        //Construct the composite key to select the entry\r
        key := ControllerKey{\r
                ControllerName: "",\r
        }\r
\r
        var resp []Controller\r
        values, err := db.DBconn.Find(mc.collectionName, key, mc.tagMeta)\r
        if err != nil {\r
                return []Controller{}, pkgerrors.Wrap(err, "Get Controller")\r
        }\r
\r
        for _, value := range values {\r
                microserv := Controller{}\r
                err = db.DBconn.Unmarshal(value, &microserv)\r
                if err != nil {\r
                        return []Controller{}, pkgerrors.Wrap(err, "Unmarshaling Value")\r
                }\r
\r
                resp = append(resp, microserv)\r
        }\r
\r
        return resp, nil\r
}\r
\r
// DeleteController the  Controller from database\r
func (mc *ControllerClient) DeleteController(name string) error {\r
\r
        //Construct the composite key to select the entry\r
        key := ControllerKey{\r
                ControllerName: name,\r
        }\r
        err := db.DBconn.Remove(mc.collectionName, key)\r
        if err != nil {\r
                return pkgerrors.Wrap(err, "Delete Controller Entry;")\r
        }\r
\r
        // send message to close rpc connection\r
        rpc.RemoveRpcConn(name)\r
\r
        return nil\r
}\r
\r
// InitControllers initializes connctions for controllers in the DB\r
func (mc *ControllerClient) InitControllers() {\r
        vals, _ := mc.GetControllers()\r
        for _, v := range vals {\r
                log.Info("Initializing RPC connection for controller", log.Fields{\r
                        "Controller": v.Metadata.Name,\r
                })\r
                rpc.UpdateRpcConn(v.Metadata.Name, v.Spec.Host, v.Spec.Port)\r
        }\r
}\r