common.sh

#!/bin/bash

# Copyright The KubeDB Authors.
#
# 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.

# ref: https://github.com/kubernetes/charts/blob/master/stable/mongodb-replicaset/init/on-start.sh

DEFAULT_WAIT_SECS=5
script_name=${0##*/}
count=0
log() {
    local msg="$1"
    local timestamp
    timestamp=$(date --iso-8601=ns)
    echo "[$timestamp] [$script_name] $msg" | tee -a /work-dir/log.txt
}

init() {
    count=0
}

exitScript() {
    total=10
    count=$((count + 1))
    if [ "$count" -ge "$total" ]; then
        exit 1
    fi
}

retry() {
    local delay=1
    local numberOfTry=300
    local tryNo=1
    while [[ $tryNo -le $numberOfTry ]]; do
        str_command="$*"
        log "Running command $str_command . . ."
        out=$("$@")
        log "$out"
        tryNo=$((tryNo + 1))

        if [ "$(echo $out | jq -r '.ok')" == "1" ]; then
            return 0
        elif echo $out | jq -r '.errmsg' | grep "HostUnreachable"; then
            sleep $delay
        elif echo $out | jq -r '.errmsg' | grep "Host not found"; then
            sleep $delay
        elif echo $out | grep "connection attempt failed: SocketException: stream truncated"; then
            # To handle ReconfigureTLS-situation like, current-pod has tls configured, but other peers dont
            return 0
        elif echo $out | grep "SocketException"; then
            # SocketException occurs in 3 commands[rs.add(), rs.addArb(), isMaster()] & 2 variation['connection attempt failed', 'host not found'] mainly.
            sleep $delay
        elif [ "$(echo $out | jq -r '.ok')" == "0" ]; then
            exit 1 # kill the container
        else
            return 0
        fi
    done
    exit 1
}

# bug: https://jira.mongodb.org/browse/SERVER-42065
# ref: https://www.golinuxcloud.com/linux-check-ipv6-enabled/#Method_1_Check_IPv6_module_status
ipv6=
if [ $(cat /sys/module/ipv6/parameters/disable) -eq "0" ]; then
    ipv6="--ipv6"
fi