2c7d53e5fb
Why I did it Need to share docker image for telemetry and gnmi, and only use telemetry container for 202305 branch Work item tracking Microsoft ADO (number only): How I did it Add a new docker image, base-gnmi, build sonic-gnmi and sonic-telemetry on this docker image. Enable telemetry container. How to verify it Run end to end test for telemetry and gnmi.
106 lines
3.3 KiB
Bash
Executable File
106 lines
3.3 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
EXIT_TELEMETRY_VARS_FILE_NOT_FOUND=1
|
|
INCORRECT_TELEMETRY_VALUE=2
|
|
TELEMETRY_VARS_FILE=/usr/share/sonic/templates/telemetry_vars.j2
|
|
|
|
if [ ! -f "$TELEMETRY_VARS_FILE" ]; then
|
|
echo "Telemetry vars template file not found"
|
|
exit $EXIT_TELEMETRY_VARS_FILE_NOT_FOUND
|
|
fi
|
|
|
|
# Try to read telemetry and certs config from ConfigDB.
|
|
# Use default value if no valid config exists
|
|
TELEMETRY_VARS=$(sonic-cfggen -d -t $TELEMETRY_VARS_FILE)
|
|
TELEMETRY_VARS=${TELEMETRY_VARS//[\']/\"}
|
|
X509=$(echo $TELEMETRY_VARS | jq -r '.x509')
|
|
GNMI=$(echo $TELEMETRY_VARS | jq -r '.gnmi')
|
|
CERTS=$(echo $TELEMETRY_VARS | jq -r '.certs')
|
|
|
|
TELEMETRY_ARGS=" -logtostderr"
|
|
export CVL_SCHEMA_PATH=/usr/sbin/schema
|
|
|
|
if [ -n "$CERTS" ]; then
|
|
SERVER_CRT=$(echo $CERTS | jq -r '.server_crt')
|
|
SERVER_KEY=$(echo $CERTS | jq -r '.server_key')
|
|
if [ -z $SERVER_CRT ] || [ -z $SERVER_KEY ]; then
|
|
TELEMETRY_ARGS+=" --insecure"
|
|
else
|
|
TELEMETRY_ARGS+=" --server_crt $SERVER_CRT --server_key $SERVER_KEY "
|
|
fi
|
|
|
|
CA_CRT=$(echo $CERTS | jq -r '.ca_crt')
|
|
if [ ! -z $CA_CRT ]; then
|
|
TELEMETRY_ARGS+=" --ca_crt $CA_CRT"
|
|
fi
|
|
elif [ -n "$X509" ]; then
|
|
SERVER_CRT=$(echo $X509 | jq -r '.server_crt')
|
|
SERVER_KEY=$(echo $X509 | jq -r '.server_key')
|
|
if [ -z $SERVER_CRT ] || [ -z $SERVER_KEY ]; then
|
|
TELEMETRY_ARGS+=" --insecure"
|
|
else
|
|
TELEMETRY_ARGS+=" --server_crt $SERVER_CRT --server_key $SERVER_KEY "
|
|
fi
|
|
|
|
CA_CRT=$(echo $X509 | jq -r '.ca_crt')
|
|
if [ ! -z $CA_CRT ]; then
|
|
TELEMETRY_ARGS+=" --ca_crt $CA_CRT"
|
|
fi
|
|
else
|
|
TELEMETRY_ARGS+=" --noTLS"
|
|
fi
|
|
|
|
# If no configuration entry exists for TELEMETRY, create one default port
|
|
if [ -z "$GNMI" ]; then
|
|
PORT=8080
|
|
else
|
|
PORT=$(echo $GNMI | jq -r '.port')
|
|
fi
|
|
TELEMETRY_ARGS+=" --port $PORT"
|
|
|
|
CLIENT_AUTH=$(echo $GNMI | jq -r '.client_auth')
|
|
if [ -z $CLIENT_AUTH ] || [ $CLIENT_AUTH == "false" ]; then
|
|
TELEMETRY_ARGS+=" --allow_no_client_auth"
|
|
fi
|
|
|
|
LOG_LEVEL=$(echo $GNMI | jq -r '.log_level')
|
|
if [[ $LOG_LEVEL =~ ^[0-9]+$ ]]; then
|
|
TELEMETRY_ARGS+=" -v=$LOG_LEVEL"
|
|
else
|
|
TELEMETRY_ARGS+=" -v=2"
|
|
fi
|
|
|
|
# Enable ZMQ for SmartSwitch
|
|
LOCALHOST_SUBTYPE=`sonic-db-cli CONFIG_DB hget localhost "subtype"`
|
|
if [[ x"${LOCALHOST_SUBTYPE}" == x"SmartSwitch" ]]; then
|
|
TELEMETRY_ARGS+=" -zmq_address=tcp://127.0.0.1:8100"
|
|
fi
|
|
|
|
# Server will handle threshold connections consecutively
|
|
THRESHOLD_CONNECTIONS=$(echo $GNMI | jq -r '.threshold')
|
|
if [[ $THRESHOLD_CONNECTIONS =~ ^[0-9]+$ ]]; then
|
|
TELEMETRY_ARGS+=" --threshold $THRESHOLD_CONNECTIONS"
|
|
else
|
|
if [ -z "$GNMI" ] || [[ $THRESHOLD_CONNECTIONS == "null" ]]; then
|
|
TELEMETRY_ARGS+=" --threshold 100"
|
|
else
|
|
echo "Incorrect threshold value, expecting positive integers" >&2
|
|
exit $INCORRECT_TELEMETRY_VALUE
|
|
fi
|
|
fi
|
|
|
|
# Close idle connections after certain duration (in seconds)
|
|
IDLE_CONN_DURATION=$(echo $GNMI | jq -r '.idle_conn_duration')
|
|
if [[ $IDLE_CONN_DURATION =~ ^[0-9]+$ ]]; then
|
|
TELEMETRY_ARGS+=" --idle_conn_duration $IDLE_CONN_DURATION"
|
|
else
|
|
if [ -z "$GNMI" ] || [[ $IDLE_CONN_DURATION == "null" ]]; then
|
|
TELEMETRY_ARGS+=" --idle_conn_duration 5"
|
|
else
|
|
echo "Incorrect idle_conn_duration value, expecting positive integers" >&2
|
|
exit $INCORRECT_TELEMETRY_VALUE
|
|
fi
|
|
fi
|
|
|
|
exec /usr/sbin/telemetry ${TELEMETRY_ARGS}
|