121 lines
3.7 KiB
Bash
121 lines
3.7 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# Define a logging function
|
||
|
|
log() {
|
||
|
|
local MESSAGE="$1"
|
||
|
|
echo -e "$MESSAGE" | tee -a /var/log/cassandra/setup.log
|
||
|
|
}
|
||
|
|
|
||
|
|
log "RUNNING SETUP"
|
||
|
|
|
||
|
|
# Configuration
|
||
|
|
KEYSPACE=${KEYSPACE:-dev_keyspace_1}
|
||
|
|
DUMP_DIR=${DUMP_DIR:-/dump} # Ensure DUMP_DIR is defined
|
||
|
|
CASSANDRA_SEEDS=${CASSANDRA_SEEDS:-cassandra1,cassandra2,cassandra3}
|
||
|
|
PRIMARY_NODE=${PRIMARY_NODE:-false} # Default to false if not set
|
||
|
|
|
||
|
|
IP_ADDRESS=$(hostname -I | awk '{print $1}')
|
||
|
|
|
||
|
|
SCHEMA_PATH="$DUMP_DIR/schema/${KEYSPACE}_schema.cql"
|
||
|
|
DATA_DIR="/var/lib/cassandra/data/$KEYSPACE"
|
||
|
|
SNAPSHOT_DIR="$DUMP_DIR/snapshot"
|
||
|
|
|
||
|
|
# Initialize SEEDS array
|
||
|
|
SEEDS=(${CASSANDRA_SEEDS//,/ })
|
||
|
|
|
||
|
|
SLEEP_DURATION=5 # Sleep duration in seconds for waits
|
||
|
|
TIMEOUT=300 # Timeout in seconds for waits
|
||
|
|
|
||
|
|
# Function to wait for a command to succeed
|
||
|
|
wait_for_command() {
|
||
|
|
local COMMAND="$1"
|
||
|
|
local TIMEOUT="$2"
|
||
|
|
local START_TIME=$(date +%s)
|
||
|
|
local END_TIME=$((START_TIME + TIMEOUT))
|
||
|
|
|
||
|
|
while true; do
|
||
|
|
if eval "$COMMAND"; then
|
||
|
|
log "Command succeeded: $COMMAND"
|
||
|
|
break
|
||
|
|
else
|
||
|
|
local CURRENT_TIME=$(date +%s)
|
||
|
|
if [ "$CURRENT_TIME" -ge "$END_TIME" ]; then
|
||
|
|
log "Timed out waiting for command: $COMMAND"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
log "Command failed: $COMMAND, still waiting"
|
||
|
|
sleep $SLEEP_DURATION
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
}
|
||
|
|
|
||
|
|
log "setup KEYSPACE: $KEYSPACE"
|
||
|
|
log "setup DUMP_DIR: $DUMP_DIR"
|
||
|
|
log "setup SCHEMA_PATH: $SCHEMA_PATH"
|
||
|
|
log "setup CASSANDRA_SEEDS: $CASSANDRA_SEEDS"
|
||
|
|
|
||
|
|
# Check if the keyspace directory exists and is not empty
|
||
|
|
if [ -d "$DATA_DIR" ] && [ "$(ls -A $DATA_DIR)" ]; then
|
||
|
|
log "Data directory $DATA_DIR exists and is not empty. Skipping schema creation and data import."
|
||
|
|
EMPTY_DB=false
|
||
|
|
else
|
||
|
|
log "Data directory $DATA_DIR does not exist or is empty. Proceeding with schema creation and data import."
|
||
|
|
EMPTY_DB=true
|
||
|
|
fi
|
||
|
|
|
||
|
|
# # Wait for cassandra1 to be ready if this is not the primary node
|
||
|
|
# if [ "$PRIMARY_NODE" != "true" ]; then
|
||
|
|
# wait_for_command "/scripts/is_node_up.sh --node cassandra1 --cassandra_rpc_address $IP_ADDRESS" $TIMEOUT
|
||
|
|
# fi
|
||
|
|
|
||
|
|
# Start Cassandra in the background
|
||
|
|
cassandra -R &
|
||
|
|
|
||
|
|
# Wait for Cassandra to be ready
|
||
|
|
wait_for_command "/scripts/is_cassandra_ready.sh" $TIMEOUT
|
||
|
|
|
||
|
|
# Log the value of PRIMARY_NODE for debugging
|
||
|
|
log "PRIMARY_NODE is set to: $PRIMARY_NODE"
|
||
|
|
|
||
|
|
# Step 1: Wait for all nodes to be up and ready
|
||
|
|
log "Waiting for all nodes to be up and ready..."
|
||
|
|
wait_for_command "/scripts/is_node_up.sh --node $seed " $TIMEOUT
|
||
|
|
// TODO: aspettare tutti i nodi
|
||
|
|
|
||
|
|
# Function to wait for all nodes to be up
|
||
|
|
wait_for_all_nodes_up() {
|
||
|
|
for seed in "${SEEDS[@]}"; do
|
||
|
|
wait_for_command "/scripts/is_node_up.sh --node $seed " $TIMEOUT
|
||
|
|
done
|
||
|
|
log "All nodes are up."
|
||
|
|
}
|
||
|
|
|
||
|
|
wait_for_all_nodes_up
|
||
|
|
|
||
|
|
# Step 2: Create keyspace and schema on the primary node
|
||
|
|
if [ "$PRIMARY_NODE" = "true" ]; then
|
||
|
|
log "Checking if keyspace $KEYSPACE exists..."
|
||
|
|
if ! /scripts/is_keyspace_exists.sh --keyspace "$KEYSPACE"; then
|
||
|
|
log "Keyspace $KEYSPACE does not exist. Creating keyspace and tables..."
|
||
|
|
cqlsh $IP_ADDRESS -f "$SCHEMA_PATH"
|
||
|
|
else
|
||
|
|
log "Keyspace $KEYSPACE already exists. Ensuring tables exist..."
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Step 3: Wait for schema to be created and agreed upon across all nodes
|
||
|
|
log "Waiting for schema agreement across all nodes..."
|
||
|
|
wait_for_command "/scripts/is_schema_agreed.sh --keyspace $KEYSPACE --cassandra_seeds $CASSANDRA_SEEDS" $TIMEOUT
|
||
|
|
|
||
|
|
# Step 4: Import data using sstableloader if not previously imported
|
||
|
|
if [ "$EMPTY_DB" = true ]; then
|
||
|
|
log "Importing snapshots using sstableloader..."
|
||
|
|
/scripts/import.sh --keyspace "$KEYSPACE" --dump_dir "$SNAPSHOT_DIR" --cassandra_seeds "$CASSANDRA_SEEDS"
|
||
|
|
fi
|
||
|
|
|
||
|
|
log "FINISHED IMPORT"
|
||
|
|
|
||
|
|
# Keep the container running
|
||
|
|
tail -f /dev/null
|