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
|
#!/bin/bash
# ================================================================================
# Copyright (c) 2019 AT&T Intellectual Property. All rights reserved.
# ================================================================================
# 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.
# ============LICENSE_END=========================================================
# Push service registrations and key-value pairs to consul
#
# Environment variables control the consul address used:
# -- CONSUL_PROTO: The protocol (http or https) used to access consul. DEFAULT: http
# -- CONSUL_HOST: The Consul host address. DEFAULT: consul
# -- CONSUL_PORT: The Consul API port. DEFAULT: 8500
#
# Command line options
# --service name|address|port : Register a service with name 'name', address 'address', and port 'port'.
# --key keyname|filepath: Register a key-value pair with key 'keyname' and the contents of a file at 'filepath' as its value
# A command can include multiple instances of each option.
CONSUL_ADDR=${CONSUL_PROTO:-http}://${CONSUL_HOST:-consul}:${CONSUL_PORT:-8500}
KV_URL=${CONSUL_ADDR}/v1/kv
REG_URL=${CONSUL_ADDR}/v1/catalog/register
# Register a service into Consul so that it can be discovered via the Consul service discovery API
# $1: Name under which service is registered
# $2: Address (typically DNS name, but can be IP) of the service
# $3: Port used by the service
function register_service {
service="{\"Node\": \"dcae\", \"Address\": \"$2\", \"Service\": {\"Service\": \"$1\", \"Address\": \"$2\", \"Port\": $3}}"
echo $service
curl -v -X PUT --data-binary "${service}" -H 'Content-Type: application/json' $REG_URL
}
# Store the contents of a file into Consul KV store
# $1: Key under which content is stored
# $2: Path to file whose content will be the value associated with the key
function put_key {
curl -v -X PUT --data-binary @$2 -H 'Content-Type: application/json' ${KV_URL}/$1
}
set -x
# Check Consul readiness
# The readiness container waits for a "consul-server" container to be ready,
# but this isn't always enough. We need the Consul API to be up and for
# the cluster to be formed, otherwise our Consul accesses might fail.
# Wait for Consul API to come up
until curl ${CONSUL_ADDR}/v1/agent/services
do
echo Waiting for Consul API
sleep 60
done
# Wait for a leader to be elected
until [[ "$(curl -Ss {$CONSUL_ADDR}/v1/status/leader)" != '""' ]]
do
echo Waiting for leader
sleep 30
done
while (( "$#" ))
do
case $1 in
"--service")
# ${2//|/ } turns all of the | characters in argument 2 into spaces
# () uses the space delimited string to initialize an array
# this turns an argument like inventory-api|inventory.onap|8080 into
# a three-element array with elements "inventory-api", "inventory.onap", and "8080"
s=(${2//|/ })
register_service ${s[@]}
shift 2;
;;
"--key")
# See above for explanation of (${2//|/ })
kv=(${2//|/ })
put_key ${kv[@]}
shift 2;
;;
*)
echo "ignoring $1"
shift
;;
esac
done
|
