forked from amey-sam/cassandra
-
Notifications
You must be signed in to change notification settings - Fork 0
/
docker-entrypoint.sh
executable file
·93 lines (74 loc) · 2.59 KB
/
docker-entrypoint.sh
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
#!/bin/bash
set -e
# first arg is `-f` or `--some-option`
if [ "${1:0:1}" = '-' ]; then
set -- cassandra -f "$@"
fi
# allow the container to be started with `--user`
if [ "$1" = 'cassandra' -a "$(id -u)" = '0' ]; then
chown -R cassandra /var/lib/cassandra /var/log/cassandra "$CASSANDRA_CONFIG"
exec gosu cassandra "$BASH_SOURCE" "$@"
fi
if [ "$1" = 'cassandra' ]; then
sleep $[ ( $RANDOM % 10 ) + 1]
echo 'LAUNCH NODETOOL REPAIR IN BACKGROUND,
SCRIPT WILL WAIT FOR CASSANDRA TO BE FULLY BOOTED'
nohup sh etc/cassandra/node-repair-after-full-boot.sh > /dev/stdout 2>&1 &
: ${CASSANDRA_RPC_ADDRESS='0.0.0.0'}
: ${HOST_COMMAND='hostname --ip-address'}
: ${SERVICE_NAME='cassadra'}
: ${SEEDS_COMMAND='echo $CASSANDRA_BROADCAST_ADDRESS'}
# Listen Address
: ${CASSANDRA_LISTEN_ADDRESS_COMMAND=$HOST_COMMAND}
: ${CASSANDRA_LISTEN_ADDRESS='auto'}
if [ "$CASSANDRA_LISTEN_ADDRESS" = 'auto' ]; then
CASSANDRA_LISTEN_ADDRESS=$(eval $CASSANDRA_LISTEN_ADDRESS_COMMAND)
fi
# Broadcast Address
: ${CASSANDRA_BROADCAST_ADDRESS_COMMAND=$HOST_COMMAND}
: ${CASSANDRA_BROADCAST_ADDRESS="$CASSANDRA_LISTEN_ADDRESS"}
if [ "$CASSANDRA_BROADCAST_ADDRESS" = 'auto' ]; then
CASSANDRA_BROADCAST_ADDRESS=$(eval $CASSANDRA_BROADCAST_ADDRESS_COMMAND)
fi
: ${CASSANDRA_BROADCAST_RPC_ADDRESS:=$CASSANDRA_BROADCAST_ADDRESS}
# Seeds
if [ -n "${CASSANDRA_NAME:+1}" ]; then
: ${CASSANDRA_SEEDS="cassandra"}
fi
if [ "$CASSANDRA_SEEDS" = 'auto' ]; then
tasks=`getent hosts tasks.$SERVICE_NAME | awk '{print $1}'`
Ntasks=`echo "$tasks" | wc -l`
seeds=`echo "$tasks" | sed "/$CASSANDRA_BROADCAST_ADDRESS/d" | paste -d, -s -`;
echo "found seeds $seeds"
CASSANDRA_SEEDS=$(printf "%s" $( [ ${Ntasks} -gt 1 ] && echo ${seeds} || echo "$CASSANDRA_BROADCAST_ADDRESS" ))
echo "cassandra_seeds set as $CASSANDRA_SEEDS"
# CASSANDRA_SEEDS=$(eval $SEEDS_COMMAND)
fi
sed -ri 's/(- seeds:).*/\1 "'"$CASSANDRA_SEEDS"'"/' "$CASSANDRA_CONFIG/cassandra.yaml"
# Substitution
for yaml in \
broadcast_address \
broadcast_rpc_address \
cluster_name \
endpoint_snitch \
listen_address \
listen_interface \
num_tokens \
rpc_address \
start_rpc \
; do
var="CASSANDRA_${yaml^^}"
val="${!var}"
if [ "$val" ]; then
sed -ri 's/^(# )?('"$yaml"':).*/\2 '"$val"'/' "$CASSANDRA_CONFIG/cassandra.yaml"
fi
done
for rackdc in dc rack; do
var="CASSANDRA_${rackdc^^}"
val="${!var}"
if [ "$val" ]; then
sed -ri 's/^('"$rackdc"'=).*/\1 '"$val"'/' "$CASSANDRA_CONFIG/cassandra-rackdc.properties"
fi
done
fi
exec "$@"