blob: 34825bd283cc8940125ecbae910d1d414b9291c7 (
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
96
97
98
99
100
101
102
103
104
105
106
107
|
#!/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
chown -R cassandra /var/lib/cassandra /var/log/cassandra "$CASSANDRA_CONFIG"
exec gosu cassandra "$BASH_SOURCE" "$@"
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
}
'
}
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
echo "#############################################"
echo "############## Update music.cql #############"
echo "#############################################"
if [ "$CASSANDRA_BROADCAST_ADDRESS" = 'auto' ]; then
CASSANDRA_BROADCAST_ADDRESS="$(_ip_address)"
fi
if [ "${MUSIC_REPLICATION_FACTOR}" ]; then
sed -ri 's/REPLICATION_FACTOR/'${MUSIC_REPLICATION_FACTOR}'/' "$f"
fi
done
echo "#############################################"
echo "############## Update music.cql #############"
echo "#############################################"
for f in /docker-entrypoint-initdb.d/a_music.cql; do
if [ "${MUSIC_REPLICATION_CLASS}" ]; then
sed -ri 's/REPLICATION_CLASS/'${MUSIC_REPLICATION_CLASS}'/' "$f"
fi
if [ "${MUSIC_REPLICATION_FACTOR}" ]; then
sed -ri 's/REPLICATION_FACTOR/'${MUSIC_REPLICATION_FACTOR}'/' "$f"
fi
done
echo "#############################################"
echo "######Updating username and password #######"
echo "#############################################"
for f in /docker-entrypoint-initdb.d/b_pw.cql; do
if [ "${CASSUSER}" ]; then
sed -ri 's/CASSUSER/'${CASSUSER}'/' "$f"
fi
if [ "${CASSPASS}" ]; then
sed -ri 's/CASSPASS/'${CASSPASS}'/' "$f"
fi
done
echo "#############################################"
echo "############## Let run cql's ################"
echo "#############################################"
for f in /docker-entrypoint-initdb.d/*; do
case "$f" in
*zzz*.cql)
echo "$0: running $f" && until $AM && cqlsh -u ${CASSUSER} -p ${CASSPASS} -f "$f";
do >&2 echo "Cassandra is unavailable - sleeping [${f}] $C";let C=C+1; sleep 5; done & ;;
*a_music.cql)
echo "$0: running $f" && until $PW && cqlsh -u ${CASSUSER} -p ${CASSPASS} -f "$f" && AM=true;
do >&2 echo "Cassandra is unavailable - sleeping [${f}] $D";let D=D+1; sleep 5; done & ;;
*b_pw.cql)
echo "$0: running $f" && until cqlsh -u cassandra -p cassandra -f "$f" && PW=true;
do >&2 echo "Cassandra is unavailable - sleeping [${f}] $E";let E=E+1; sleep 5; done & ;;
*) echo "$0: ignoring $f" ;;
esac
echo
done
echo "#############################################"
echo "########### Running Password CQL ############"
echo "#############################################"
#echo "$0: running $f" &&
#until cqlsh -u cassandra -p cassandra -f /pw.cql;
#do >&2 echo "Cassandra is unavailable - sleeping"; sleep 10; done
echo "#############################################"
echo "########### Cassandra Running ###############"
echo "#############################################"
exec "$@"
|
