1 files changed, 148 insertions, 0 deletions
diff --git a/mdbc-packages/mdbc-docker/src/main/docker/docker-files/scripts/start-cassandra.sh b/mdbc-packages/mdbc-docker/src/main/docker/docker-files/scripts/start-cassandra.sh
new file mode 100644
index 0000000..362c9e9
--- /dev/null
+++ b/mdbc-packages/mdbc-docker/src/main/docker/docker-files/scripts/start-cassandra.sh
@@ -0,0 +1,148 @@
+#!/bin/bash
+
+# This enhances the standard docker entrypoint script by running cql scripts
+# and shell scripts present in /docker-entrypoint-initdb.d. The public native
+# transport port is not exposed until all the scripts have been completed.
+
+if [[ $PRIVATE_PORT == "" ]]
+then
+	PRIVATE_PORT=19042
+fi
+
+if [[ $START_TIMEOUT_SECS == "" ]]
+then
+	START_TIMEOUT_SECS=300
+fi
+
+function getPort
+{
+	grep "^native_transport_port:" /etc/cassandra/cassandra.yaml | cut -d: -f2 | tr -d '\t '
+}
+
+function setPort
+{
+	local port=$1
+	sed -i "s/^native_transport_port:.*/native_transport_port: $port/" /etc/cassandra/cassandra.yaml
+}
+
+function usage
+{
+	echo "usage: $(basename $0) [-sX]"
+}
+
+sleep=0
+args=
+
+while [[ $# != 0 ]]
+do
+	case "$1" in
+	-s*)
+		sleep=${1:2}
+		shift 1
+		;;
+	*)
+		usage && exit 1
+		;;
+	esac
+done
+
+if [[ $# != 0 ]]
+then
+	usage && exit 1
+fi
+
+if [[ $sleep != 0 ]]
+then
+	echo "Delaying startup by $sleep seconds"
+	sleep $sleep
+fi
+
+# Note: this does not start up cassandra
+echo "Running docker-entrypoint.sh to configure cassandra"
+/docker-entrypoint.sh true || exit 1
+
+echo "Enabling password authentication"
+sed -i 's/^authenticator: AllowAllAuthenticator/authenticator: PasswordAuthenticator/g' /etc/cassandra/cassandra.yaml || exit 1
+
+if [[ $(/bin/ls -1 /docker-entrypoint-initdb.d 2>/dev/null | wc -l) == 0 ]]
+then
+	echo "Starting cassandra"
+else
+	# Get the public native transport port from cassandra.yaml
+
+	public_port=$(getPort)
+
+	if [[ $public_port == "" ]]
+	then
+		echo "ERROR: No native_transport_port in /etc/cassandra/cassandra.yaml"
+		exit 1
+	fi
+
+	echo "Starting cassandra on port $PRIVATE_PORT"
+	setPort $PRIVATE_PORT || exit 1
+
+	/docker-entrypoint.sh cassandra
+
+	if [[ $? != 0 ]]
+	then
+		setPort $public_port
+		exit 1
+	fi
+
+	tries=$START_TIMEOUT_SECS
+
+	while true
+	do
+		if echo "describe keyspaces;" | cqlsh -u cassandra -p cassandra 127.0.0.1 $PRIVATE_PORT >/dev/null 2>&1
+		then
+			break
+		fi
+
+		tries=$((tries-1))
+
+		if [[ $tries == 0 ]]
+		then
+			setPort $public_port
+			echo "Timed out waiting for cassandra to start on port $PRIVATE_PORT"
+			exit 1
+		fi
+
+		sleep 1
+	done
+
+	echo "Cassandra is started on port $PRIVATE_PORT"
+
+	for file in $(/bin/ls -1 /docker-entrypoint-initdb.d 2>/dev/null)
+	do
+		case "$file" in
+		*.sh)
+			echo "Running $file"
+			source "/docker-entrypoint-initdb.d/$file"
+			;;
+		*.cql)
+			echo "Running $file"
+			cqlsh -u cassandra -p cassandra -f "/docker-entrypoint-initdb.d/$file" 127.0.0.1 $PRIVATE_PORT
+			;;
+        	*)
+			echo "Ignoring $file"
+    		esac
+	done
+
+	sleep 5
+
+	echo "Restarting cassandra on port $public_port"
+
+	setPort $public_port
+	pkill java
+
+	if [[ $? != 0 ]]
+	then
+		echo "Failed to restart cassandra (kill failed)"
+		exit 1
+	fi
+
+	sleep 5
+fi
+
+# NOTE: this starts cassandra in the foreground
+/docker-entrypoint.sh cassandra -f