kubernetes/common/cassandra/resources/config/docker-entrypoint.sh

   1 #!/bin/bash
   2
   3 set -e
   4
   5 # first arg is `-f` or `--some-option`
   6 # or there are no args
   7 if [ "$#" -eq 0 ] || [ "${1#-}" != "$1" ]; then
   8         set -- cassandra -f "$@"
   9 fi
  10
  11 # allow the container to be started with `--user`
  12 if [ "$1" = 'cassandra' -a "$(id -u)" = '0' ]; then
  13         find /var/lib/cassandra /var/log/cassandra "$CASSANDRA_CONFIG" \
  14                 \! -user cassandra -exec chown cassandra '{}' +
  15         exec gosu cassandra "$0" "$@"
  16 fi
  17
  18 _ip_address() {
  19         # scrape the first non-localhost IP address of the container
  20         # in Swarm Mode, we often get two IPs -- the container IP, and the (shared) VIP, and the container IP should always be first
  21         ip address | awk '
  22                 $1 == "inet" && $NF != "lo" {
  23                         gsub(/\/.+$/, "", $2)
  24                         print $2
  25                         exit
  26                 }
  27         '
  28 }
  29
  30 # "sed -i", but without "mv" (which doesn't work on a bind-mounted file, for example)
  31 _sed_in_place() {
  32         local filename="$1"; shift
  33         local tempFile
  34         tempFile="$(mktemp)"
  35         sed "$@" "$filename" > "$tempFile"
  36         cat "$tempFile" > "$filename"
  37         rm "$tempFile"
  38 }
  39
  40 if [ "$1" = 'cassandra' ]; then
  41         : ${CASSANDRA_RPC_ADDRESS='0.0.0.0'}
  42
  43         : ${CASSANDRA_LISTEN_ADDRESS='auto'}
  44         if [ "$CASSANDRA_LISTEN_ADDRESS" = 'auto' ]; then
  45                 CASSANDRA_LISTEN_ADDRESS="$(_ip_address)"
  46         fi
  47
  48         : ${CASSANDRA_BROADCAST_ADDRESS="$CASSANDRA_LISTEN_ADDRESS"}
  49
  50         if [ "$CASSANDRA_BROADCAST_ADDRESS" = 'auto' ]; then
  51                 CASSANDRA_BROADCAST_ADDRESS="$(_ip_address)"
  52         fi
  53         : ${CASSANDRA_BROADCAST_RPC_ADDRESS:=$CASSANDRA_BROADCAST_ADDRESS}
  54
  55         if [ -n "${CASSANDRA_NAME:+1}" ]; then
  56                 : ${CASSANDRA_SEEDS:="cassandra"}
  57         fi
  58         : ${CASSANDRA_SEEDS:="$CASSANDRA_BROADCAST_ADDRESS"}
  59
  60         _sed_in_place "$CASSANDRA_CONFIG/cassandra.yaml" \
  61                 -r 's/(- seeds:).*/\1 "'"$CASSANDRA_SEEDS"'"/'
  62
  63         for yaml in \
  64                 broadcast_address \
  65                 broadcast_rpc_address \
  66                 cluster_name \
  67                 endpoint_snitch \
  68                 listen_address \
  69                 num_tokens \
  70                 rpc_address \
  71                 start_rpc \
  72                 authenticator \
  73         ; do
  74                 var="CASSANDRA_$(echo $yaml | tr '[:lower:]' '[:upper:]')"
  75                 # eval presents no security issue here because of limited possible values of var
  76                 eval val=\$$var
  77                 if [ "$val" ]; then
  78                         _sed_in_place "$CASSANDRA_CONFIG/cassandra.yaml" \
  79                                 -r 's/^(# )?('"$yaml"':).*/\2 '"$val"'/'
  80                 fi
  81         done
  82
  83         for rackdc in dc rack; do
  84                 var="CASSANDRA_$(echo $rackdc | tr '[:lower:]' '[:upper:]')"
  85                 # eval presents no security issue here because of limited possible values of var
  86                 eval val=\$$var
  87                 if [ "$val" ]; then
  88                         _sed_in_place "$CASSANDRA_CONFIG/cassandra-rackdc.properties" \
  89                                 -r 's/^('"$rackdc"'=).*/\1 '"$val"'/'
  90                 fi
  91         done
  92 fi
  93
  94 exec "$@"
  95