|
| 1 | +#!/bin/bash |
| 2 | +set -e |
| 3 | + |
| 4 | +# first arg is `-f` or `--some-option` |
| 5 | +# or there are no args |
| 6 | +if [ "$#" -eq 0 ] || [ "${1#-}" != "$1" ]; then |
| 7 | + set -- dse cassandra -f "$@" |
| 8 | +fi |
| 9 | + |
| 10 | +if [ "$CASSANDRA_CONF" == "" ]; then |
| 11 | + export CASSANDRA_CONF=/opt/dse/resources/cassandra/conf |
| 12 | +fi |
| 13 | + |
| 14 | +# allow the container to be started with `--user` |
| 15 | +if [ "$1" = 'mgmtapi' -a "$(id -u)" = '0' ]; then |
| 16 | + find "$CASSANDRA_CONF" /var/lib/cassandra /var/log/cassandra \ |
| 17 | + \! -user dse -exec chown dse '{}' + |
| 18 | +fi |
| 19 | + |
| 20 | +_ip_address() { |
| 21 | + # scrape the first non-localhost IP address of the container |
| 22 | + # in Swarm Mode, we often get two IPs -- the container IP, and the (shared) VIP, and the container IP should always be first |
| 23 | + ip address | awk ' |
| 24 | + $1 == "inet" && $NF != "lo" { |
| 25 | + gsub(/\/.+$/, "", $2) |
| 26 | + print $2 |
| 27 | + exit |
| 28 | + } |
| 29 | + ' |
| 30 | +} |
| 31 | + |
| 32 | +# "sed -i", but without "mv" (which doesn't work on a bind-mounted file, for example) |
| 33 | +_sed-in-place() { |
| 34 | + local filename="$1"; shift |
| 35 | + local tempFile |
| 36 | + tempFile="$(mktemp)" |
| 37 | + sed "$@" "$filename" > "$tempFile" |
| 38 | + cat "$tempFile" > "$filename" |
| 39 | + rm "$tempFile" |
| 40 | +} |
| 41 | + |
| 42 | +if [ "$1" = 'mgmtapi' ]; then |
| 43 | + echo "Starting Management API" |
| 44 | + |
| 45 | + if ! grep -qxF "JVM_OPTS=\"\$JVM_OPTS -javaagent:/opt/dse/datastax-mgmtapi-agent-0.1.0-SNAPSHOT.jar\"" < /opt/dse/resources/cassandra/conf/cassandra-env.sh ; then |
| 46 | + # ensure newline at end of file |
| 47 | + echo "" >> /opt/dse/resources/cassandra/conf/cassandra-env.sh |
| 48 | + echo "JVM_OPTS=\"\$JVM_OPTS -javaagent:/opt/dse/datastax-mgmtapi-agent-0.1.0-SNAPSHOT.jar\"" >> /opt/dse/resources/cassandra/conf/cassandra-env.sh |
| 49 | + fi |
| 50 | + |
| 51 | + # Set this if you want to ignore default env variables, i.e. when running inside an operator |
| 52 | + if [ $IGNORE_DEFAULTS ]; then |
| 53 | + CASSANDRA_RPC_ADDRESS='0.0.0.0' |
| 54 | + CASSANDRA_BROADCAST_RPC_ADDRESS="$(_ip_address)" |
| 55 | + else |
| 56 | + : ${CASSANDRA_RPC_ADDRESS='0.0.0.0'} |
| 57 | + |
| 58 | + : ${CASSANDRA_LISTEN_ADDRESS='auto'} |
| 59 | + if [ "$CASSANDRA_LISTEN_ADDRESS" = 'auto' ]; then |
| 60 | + CASSANDRA_LISTEN_ADDRESS="$(_ip_address)" |
| 61 | + fi |
| 62 | + |
| 63 | + : ${CASSANDRA_BROADCAST_ADDRESS="$CASSANDRA_LISTEN_ADDRESS"} |
| 64 | + |
| 65 | + if [ "$CASSANDRA_BROADCAST_ADDRESS" = 'auto' ]; then |
| 66 | + CASSANDRA_BROADCAST_ADDRESS="$(_ip_address)" |
| 67 | + fi |
| 68 | + : ${CASSANDRA_BROADCAST_RPC_ADDRESS:=$CASSANDRA_BROADCAST_ADDRESS} |
| 69 | + |
| 70 | + if [ -n "${CASSANDRA_NAME:+1}" ]; then |
| 71 | + : ${CASSANDRA_SEEDS:="cassandra"} |
| 72 | + fi |
| 73 | + : ${CASSANDRA_SEEDS:="$CASSANDRA_BROADCAST_ADDRESS"} |
| 74 | + |
| 75 | + CASSANDRA_YAML="cassandra.yaml" |
| 76 | + if [ $CASSANDRA_DEPLOYMENT ]; then |
| 77 | + CASSANDRA_DEPLOYMENT=`echo "$CASSANDRA_DEPLOYMENT" | awk '{print tolower($0)}'` |
| 78 | + CASSANDRA_YAML="cassandra-$CASSANDRA_DEPLOYMENT.yaml" |
| 79 | + fi |
| 80 | + |
| 81 | + _sed-in-place "$CASSANDRA_CONF/$CASSANDRA_YAML" \ |
| 82 | + -r 's/(- seeds:).*/\1 "'"$CASSANDRA_SEEDS"'"/' |
| 83 | + |
| 84 | + for yaml in \ |
| 85 | + broadcast_address \ |
| 86 | + broadcast_rpc_address \ |
| 87 | + cluster_name \ |
| 88 | + endpoint_snitch \ |
| 89 | + listen_address \ |
| 90 | + num_tokens \ |
| 91 | + rpc_address \ |
| 92 | + start_rpc \ |
| 93 | + ; do |
| 94 | + var="CASSANDRA_${yaml^^}" |
| 95 | + val="${!var}" |
| 96 | + if [ "$val" ]; then |
| 97 | + _sed-in-place "$CASSANDRA_CONF/$CASSANDRA_YAML" \ |
| 98 | + -r 's/^(# )?('"$yaml"':).*/\2 '"$val"'/' |
| 99 | + fi |
| 100 | + done |
| 101 | + |
| 102 | + for rackdc in dc rack; do |
| 103 | + var="CASSANDRA_${rackdc^^}" |
| 104 | + val="${!var}" |
| 105 | + if [ "$val" ]; then |
| 106 | + _sed-in-place "$CASSANDRA_CONF/cassandra-rackdc.properties" \ |
| 107 | + -r 's/^('"$rackdc"'=).*/\1 '"$val"'/' |
| 108 | + fi |
| 109 | + done |
| 110 | + fi |
| 111 | + |
| 112 | + MGMT_API_ARGS="" |
| 113 | + |
| 114 | + # Hardcoding these for now |
| 115 | + MGMT_API_CASSANDRA_SOCKET="--db-socket /tmp/db.sock" |
| 116 | + MGMT_API_LISTEN_TCP="--host tcp://0.0.0.0:8080" |
| 117 | + MGMT_API_LISTEN_SOCKET="--host file:///tmp/oss-mgmt.sock" |
| 118 | + |
| 119 | + MGMT_API_ARGS="$MGMT_API_ARGS $MGMT_API_CASSANDRA_SOCKET $MGMT_API_LISTEN_TCP $MGMT_API_LISTEN_SOCKET" |
| 120 | + |
| 121 | + # These will generally come from the k8s operator |
| 122 | + if [ ! -z "$MGMT_API_EXPLICIT_START" ]; then |
| 123 | + MGMT_API_EXPLICIT_START="--explicit-start $MGMT_API_EXPLICIT_START" |
| 124 | + MGMT_API_ARGS="$MGMT_API_ARGS $MGMT_API_EXPLICIT_START" |
| 125 | + fi |
| 126 | + |
| 127 | + if [ ! -z "$MGMT_API_TLS_CA_CERT_FILE" ]; then |
| 128 | + MGMT_API_TLS_CA_CERT_FILE="--tlscacert $MGMT_API_TLS_CA_CERT_FILE" |
| 129 | + MGMT_API_ARGS="$MGMT_API_ARGS $MGMT_API_TLS_CA_CERT_FILE" |
| 130 | + fi |
| 131 | + if [ ! -z "$MGMT_API_TLS_CERT_FILE" ]; then |
| 132 | + MGMT_API_TLS_CERT_FILE="--tlscert $MGMT_API_TLS_CERT_FILE" |
| 133 | + MGMT_API_ARGS="$MGMT_API_ARGS $MGMT_API_TLS_CERT_FILE" |
| 134 | + fi |
| 135 | + if [ ! -z "$MGMT_API_TLS_KEY_FILE" ]; then |
| 136 | + MGMT_API_TLS_KEY_FILE="--tlskey $MGMT_API_TLS_KEY_FILE" |
| 137 | + MGMT_API_ARGS="$MGMT_API_ARGS $MGMT_API_TLS_KEY_FILE" |
| 138 | + fi |
| 139 | + |
| 140 | + if [ ! -z "$MGMT_API_PID_FILE" ]; then |
| 141 | + MGMT_API_PID_FILE="--pidfile $MGMT_API_PID_FILE" |
| 142 | + MGMT_API_ARGS="$MGMT_API_ARGS $MGMT_API_PID_FILE" |
| 143 | + fi |
| 144 | + |
| 145 | + MGMT_API_DSE_HOME="--db-home ${DSE_HOME}" |
| 146 | + MGMT_API_ARGS="$MGMT_API_ARGS $MGMT_API_DSE_HOME" |
| 147 | + |
| 148 | + if [ ! -z "$MGMT_API_NO_KEEP_ALIVE" ]; then |
| 149 | + MGMT_API_NO_KEEP_ALIVE="--no-keep-alive $MGMT_API_NO_KEEP_ALIVE" |
| 150 | + MGMT_API_ARGS="$MGMT_API_ARGS $MGMT_API_NO_KEEP_ALIVE" |
| 151 | + fi |
| 152 | + |
| 153 | + MGMT_API_JAR="$(find "/opt/mgmtapi" -name *server*.jar)" |
| 154 | + |
| 155 | + echo "Running" java ${MGMT_API_JAVA_OPTS} -Xms128m -Xmx128m -jar "$MGMT_API_JAR" $MGMT_API_ARGS |
| 156 | + exec java ${MGMT_API_JAVA_OPTS} -Xms128m -Xmx128m -jar "$MGMT_API_JAR" $MGMT_API_ARGS |
| 157 | + |
| 158 | +fi |
| 159 | + |
| 160 | +exec "$@" |
0 commit comments