Cassandra Docker helper

cassandra

#!/bin/bash

CASSANDRA_IMAGE='cassandra'
CASSANDRA_VERSION='2.2.9'
CASSANDRA_CONTAINER="${CASSANDRA_IMAGE}-${CASSANDRA_VERSION}"

cassandraCreate() {
    docker run -d -p 9042:9042 --name ${CASSANDRA_CONTAINER} ${CASSANDRA_IMAGE}:${CASSANDRA_VERSION}
}

cassandraStart() {
    docker start ${CASSANDRA_CONTAINER}
}

cassandraStop() {
    docker stop ${CASSANDRA_CONTAINER}
}

cassandraStatus() {
    docker ps --filter "name=${CASSANDRA_CONTAINER}"
}

cassandraConnect() {
    docker exec -it ${CASSANDRA_CONTAINER} cqlsh -k mitigation

    if [ $? -ne 0 ]; then
        docker exec -it ${CASSANDRA_CONTAINER} cqlsh
    fi
}

cassandraTruncate() {
    docker exec -it ${CASSANDRA_CONTAINER} cqlsh -k mitigation --execute "TRUNCATE table_one; TRUNCATE table_two;"
}

case "$1" in
    create)
        cassandaCreate;;
    start)
        cassandraStart;;
    stop)
        cassandraStop;;
    status)
        cassandraStatus;;
    connect)
        cassandraConnect;;
    truncate)
        cassandraTruncate;;
    *)
        echo $"Usage: $0 {create|start|stop|connect|truncate}"
        exit 1
esac