summaryrefslogtreecommitdiffstats
path: root/kubernetes/common/cassandra/resources/config/docker-entrypoint.sh
blob: 2d30f2e0686a7b97c545fccedb0db5b6cee6da0e (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
#!/bin/bash

set -e

# first arg is `-f` or `--some-option`
# or there are no args
if [ "$#" -eq 0 ] || [ "${1#-}" != "$1" ]; then
        set -- cassandra -f "$@"
fi

# allow the container to be started with `--user`
if [ "$1" = 'cassandra' -a "$(id -u)" = '0' ]; then
        find /var/lib/cassandra /var/log/cassandra "$CASSANDRA_CONFIG" \
                \! -user cassandra -exec chown cassandra '{}' +
        exec gosu cassandra "$0" "$@"
fi

_ip_address() {
        # scrape the first non-localhost IP address of the container
        # in Swarm Mode, we often get two IPs -- the container IP, and the (shared) VIP, and the container IP should always be first
        ip address | awk '
                $1 == "inet" && $NF != "lo" {
                        gsub(/\/.+$/, "", $2)
                        print $2
                        exit
                }
        '
}

# "sed -i", but without "mv" (which doesn't work on a bind-mounted file, for example)
_sed_in_place() {
        local filename="$1"; shift
        local tempFile
        tempFile="$(mktemp)"
        sed "$@" "$filename" > "$tempFile"
        cat "$tempFile" > "$filename"
        rm "$tempFile"
}

if [ "$1" = 'cassandra' ]; then
        : ${CASSANDRA_RPC_ADDRESS='0.0.0.0'}

        : ${CASSANDRA_LISTEN_ADDRESS='auto'}
        if [ "$CASSANDRA_LISTEN_ADDRESS" = 'auto' ]; then
                CASSANDRA_LISTEN_ADDRESS="$(_ip_address)"
        fi

        : ${CASSANDRA_BROADCAST_ADDRESS="$CASSANDRA_LISTEN_ADDRESS"}

        if [ "$CASSANDRA_BROADCAST_ADDRESS" = 'auto' ]; then
                CASSANDRA_BROADCAST_ADDRESS="$(_ip_address)"
        fi
        : ${CASSANDRA_BROADCAST_RPC_ADDRESS:=$CASSANDRA_BROADCAST_ADDRESS}

        if [ -n "${CASSANDRA_NAME:+1}" ]; then
                : ${CASSANDRA_SEEDS:="cassandra"}
        fi
        : ${CASSANDRA_SEEDS:="$CASSANDRA_BROADCAST_ADDRESS"}

        _sed_in_place "$CASSANDRA_CONFIG/cassandra.yaml" \
                -r 's/(- seeds:).*/\1 "'"$CASSANDRA_SEEDS"'"/'

        for yaml in \
                broadcast_address \
                broadcast_rpc_address \
                cluster_name \
                endpoint_snitch \
                listen_address \
                num_tokens \
                rpc_address \
                start_rpc \
                authenticator \
        ; do
                var="CASSANDRA_$(echo $yaml | tr '[:lower:]' '[:upper:]')"
                # eval presents no security issue here because of limited possible values of var
                eval val=\$$var
                if [ "$val" ]; then
                        _sed_in_place "$CASSANDRA_CONFIG/cassandra.yaml" \
                                -r 's/^(# )?('"$yaml"':).*/\2 '"$val"'/'
                fi
        done

        for rackdc in dc rack; do
                var="CASSANDRA_$(echo $rackdc | tr '[:lower:]' '[:upper:]')"
                # eval presents no security issue here because of limited possible values of var
                eval val=\$$var
                if [ "$val" ]; then
                        _sed_in_place "$CASSANDRA_CONFIG/cassandra-rackdc.properties" \
                                -r 's/^('"$rackdc"'=).*/\1 '"$val"'/'
                fi
        done
fi

exec "$@"