cmd/arouterd/arouterd.go

/*
 * Copyright 2018-present Open Networking Foundation

 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at

 * http://www.apache.org/licenses/LICENSE-2.0

 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package main

import (
	"context"
	"flag"
	"fmt"
	"github.com/opencord/voltha-api-server/internal/pkg/afrouterd"
	"github.com/opencord/voltha-lib-go/v2/pkg/probe"
	"github.com/opencord/voltha-lib-go/v2/pkg/version"
	"math"
	"os"
	"path"

	"github.com/opencord/voltha-lib-go/v2/pkg/log"
	pb "github.com/opencord/voltha-protos/v2/go/afrouter"
)

var (
	instanceID         = afrouterd.GetStrEnv("HOSTNAME", "arouterd001")
	afrouterApiAddress = afrouterd.GetStrEnv("AFROUTER_API_ADDRESS", "localhost:55554")
	probeHost          = afrouterd.GetStrEnv("PROBE_HOST", "")
	probePort          = afrouterd.GetIntEnv("PROBE_PORT", 0, math.MaxUint16, 8081)
)

type Configuration struct {
	DisplayVersionOnly *bool
}

func startup() int {
	config := &Configuration{}
	cmdParse := flag.NewFlagSet(path.Base(os.Args[0]), flag.ContinueOnError)
	config.DisplayVersionOnly = cmdParse.Bool("version", false, "Print version information and exit")

	if err := cmdParse.Parse(os.Args[1:]); err != nil {
		fmt.Printf("Error: %v\n", err)
		return 1
	}

	if *config.DisplayVersionOnly {
		fmt.Println("VOLTHA API Server (afrouterd)")
		fmt.Println(version.VersionInfo.String("  "))
		return 0
	}

	// Set up logging
	if _, err := log.SetDefaultLogger(log.JSON, 0, log.Fields{"instanceId": instanceID}); err != nil {
		log.With(log.Fields{"error": err}).Fatal("Cannot setup logging")
	}

	// Set up kubernetes api
	clientset := afrouterd.K8sClientSet()

	p := &probe.Probe{}
	go p.ListenAndServe(fmt.Sprintf("%s:%d", probeHost, probePort))

	p.RegisterService(
		"affinity-router",
		"message-bus",
	)

	for {
		// Connect to the affinity router
		conn, err := afrouterd.Connect(context.Background(), afrouterApiAddress) // This is a sidecar container so communicating over localhost
		if err != nil {
			panic(err)
		}
		p.UpdateStatus("affinity-router", probe.ServiceStatusRunning)

		// monitor the connection status, end context if connection is lost
		ctx := afrouterd.ConnectionActiveContext(conn, p)

		probeCtx := context.WithValue(ctx, probe.ProbeContextKey, p)
		// set up the client
		client := pb.NewConfigurationClient(conn)

		// start the discovery monitor and core monitor
		// these two processes do the majority of the work

		log.Info("Starting discovery monitoring")
		doneCh, _ := afrouterd.StartDiscoveryMonitor(probeCtx, client)

		log.Info("Starting core monitoring")
		afrouterd.CoreMonitor(probeCtx, client, clientset)

		//ensure the discovery monitor to quit
		<-doneCh

		conn.Close()
	}
}

func main() {
	status := startup()
	if status != 0 {
		os.Exit(status)
	}
}