summaryrefslogtreecommitdiffstats
path: root/src/rsync/pkg/grpc/register.go
blob: fb46250566d55332a15856e88d88468598dba6b0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
/*
Copyright 2020 Intel Corporation.
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 grpc

import (
	"os"
	"strconv"
	"strings"

	log "github.com/onap/multicloud-k8s/src/orchestrator/pkg/infra/logutils"
	controller "github.com/onap/multicloud-k8s/src/orchestrator/pkg/module/controller"
	mtypes "github.com/onap/multicloud-k8s/src/orchestrator/pkg/module/types"
)

const default_host = "localhost"
const default_port = 9031
const default_rsync_name = "rsync"
const ENV_RSYNC_NAME = "RSYNC_NAME"

func GetServerHostPort() (string, int) {

	// expect name of this rsync program to be in env variable "RSYNC_NAME" - e.g. RSYNC_NAME="rsync"
	serviceName := os.Getenv(ENV_RSYNC_NAME)
	if serviceName == "" {
		serviceName = default_rsync_name
		log.Info("Using default name for RSYNC service name", log.Fields{
			"Name": serviceName,
		})
	}

	// expect service name to be in env variable - e.g. RSYNC_SERVICE_HOST
	host := os.Getenv(strings.ToUpper(serviceName) + "_SERVICE_HOST")
	if host == "" {
		host = default_host
		log.Info("Using default host for rsync gRPC controller", log.Fields{
			"Host": host,
		})
	}

	// expect service port to be in env variable - e.g. RSYNC_SERVICE_PORT
	port, err := strconv.Atoi(os.Getenv(strings.ToUpper(serviceName) + "_SERVICE_PORT"))
	if err != nil || port < 0 {
		port = default_port
		log.Info("Using default port for rsync gRPC controller", log.Fields{
			"Port": port,
		})
	}
	return host, port
}

func RegisterGrpcServer(host string, port int) error {
	// expect name of this rsync program to be in env variable "RSYNC_NAME" - e.g. RSYNC_NAME="rsync"
	// This will be the name of the controller that is registered in the orchestrator controller API
	// This same name will be used as the key name for intents in the deployment intent group
	serviceName := os.Getenv(ENV_RSYNC_NAME)
	if serviceName == "" {
		serviceName = default_rsync_name
		log.Info("Using default name for rsync service name", log.Fields{
			"Name": serviceName,
		})
	}

	client := controller.NewControllerClient()

	// Create or update the controller entry
	controller := controller.Controller{
		Metadata: mtypes.Metadata{
			Name: serviceName,
		},
		Spec: controller.ControllerSpec{
			Host:     host,
			Port:     port,
			Type:     controller.CONTROLLER_TYPE_ACTION,
			Priority: controller.MinControllerPriority,
		},
	}
	_, err := client.CreateController(controller, true)
	if err != nil {
		log.Error("Failed to create/update a gRPC controller", log.Fields{
			"Error":      err,
			"Controller": serviceName,
		})
		return err
	}

	return nil
}