planb-cassandra.sh

#!/bin/sh
# CLUSTER_NAME
# LISTEN_ADDRESS
# BROADCAST_ADDRESS
# SNITCH
# DATA_DIR
# COMMIT_LOG_DIR
# TRUSTSTORE
# KEYSTORE
# ADMIN_PASSWORD

if [ -z "$CLUSTER_NAME" ] ;
then
    echo "Cluster name is not defined."
    exit 1
fi

EC2_META_URL=http://169.254.169.254/latest/meta-data

if [ -z "$LISTEN_ADDRESS" ] ;
then
    export LISTEN_ADDRESS=$(curl -Ls -m 4 ${EC2_META_URL}/local-ipv4)
fi
echo "Local IP address is $LISTEN_ADDRESS ..."

if [ "x$SUBNET_TYPE" = xinternal ];
then
    export BROADCAST_ADDRESS=$LISTEN_ADDRESS

    if [ -z $SNITCH ] ;
    then
        export SNITCH="Ec2Snitch"
    fi
else
    while [ -z "$BROADCAST_ADDRESS" ] ;
    do
        echo "Waiting for Public IP address to be assigned ..."
        export BROADCAST_ADDRESS=$(curl -Ls -m 4 ${EC2_META_URL}/public-ipv4)
        sleep 5
    done

    if [ -z $SNITCH ] ;
    then
        export SNITCH="Ec2MultiRegionSnitch"
    fi
fi
echo "Broadcast IP address is $BROADCAST_ADDRESS ..."

export DATA_DIR=${DATA_DIR:-/var/lib/cassandra}
export COMMIT_LOG_DIR=${COMMIT_LOG_DIR:-/var/lib/cassandra/commit_logs}

if [ -z "$TRUSTSTORE" ]; then
    echo "TRUSTSTORE must be set (base64 encoded)."
    exit 1
fi

if [ -z "$KEYSTORE" ]; then
    echo "KEYSTORE must be set (base64 encoded)."
    exit 1
fi

echo $TRUSTSTORE | base64 -d > /etc/cassandra/truststore
echo $KEYSTORE | base64 -d > /etc/cassandra/keystore

echo "Finished bootstrapping node."
# Add route 53record seed1.${CLUSTER_NAME}.domain.tld ?

ncores=$(grep -c ^processor /proc/cpuinfo)
ncores_4=$(( ncores / 4 ))
[ $ncores_4 -gt 0 ] || ncores_4=1

#
# Assuming we are using SSD storage, set memtable_flush_writers to the
# number of CPU cores divided by 4:
#
export MEMTABLE_FLUSH_WRITERS=$ncores_4

# the same for concurrent_compactors setting:
export CONCURRENT_COMPACTORS=$ncores_4

echo "Generating configuration from template ..."
python -c "import sys, os; sys.stdout.write(os.path.expandvars(open('/etc/cassandra/cassandra_template.yaml').read()))" > /etc/cassandra/cassandra.yaml

echo "Starting Cassandra ..."
/usr/sbin/cassandra -f &

#
# Try to override default superuser password (we don't care if it
# fails, that would just mean we are not the first one to do that).
#
sleep 60
cqlsh -u cassandra -p cassandra \
      -e "\
ALTER USER cassandra WITH PASSWORD '$ADMIN_PASSWORD'; \
ALTER KEYSPACE system_auth WITH replication = { 'class': 'NetworkTopologyStrategy' $(echo $REGIONS | sed "s/\([^ ]*\)-1/, '\1': $CLUSTER_SIZE/g") };"

# Make sure the script don't exit at this point, if cassandra is still there.
wait