diff options
Diffstat (limited to 'test/kafka/loadtest/setup_seaweed_infrastructure.sh')
| -rwxr-xr-x | test/kafka/loadtest/setup_seaweed_infrastructure.sh | 131 |
1 files changed, 131 insertions, 0 deletions
diff --git a/test/kafka/loadtest/setup_seaweed_infrastructure.sh b/test/kafka/loadtest/setup_seaweed_infrastructure.sh new file mode 100755 index 000000000..448119097 --- /dev/null +++ b/test/kafka/loadtest/setup_seaweed_infrastructure.sh @@ -0,0 +1,131 @@ +#!/bin/bash + +# Script to set up SeaweedFS infrastructure for Kafka Gateway testing +# This script will start Master, Filer, and MQ Broker components + +set -e + +BASE_DIR="/tmp/seaweedfs" +LOG_DIR="$BASE_DIR/logs" +DATA_DIR="$BASE_DIR/data" + +echo "=== SeaweedFS Infrastructure Setup ===" +echo "Setup Date: $(date)" +echo "Base Directory: $BASE_DIR" +echo "" + +# Create directories +mkdir -p "$BASE_DIR/master" "$BASE_DIR/filer" "$BASE_DIR/broker" "$LOG_DIR" + +# Function to check if a service is running +check_service() { + local host_port=$1 + local service_name=$2 + + if timeout 3 bash -c "</dev/tcp/${host_port//://}" 2>/dev/null; then + echo "✓ $service_name is already running on $host_port" + return 0 + else + echo "✗ $service_name is NOT running on $host_port" + return 1 + fi +} + +# Function to start a service in background +start_service() { + local cmd="$1" + local service_name="$2" + local log_file="$3" + local check_port="$4" + + echo "Starting $service_name..." + echo "Command: $cmd" + echo "Log: $log_file" + + # Start in background + nohup $cmd > "$log_file" 2>&1 & + local pid=$! + echo "PID: $pid" + + # Wait for service to be ready + local retries=30 + while [ $retries -gt 0 ]; do + if check_service "$check_port" "$service_name" 2>/dev/null; then + echo "✓ $service_name is ready" + return 0 + fi + retries=$((retries - 1)) + sleep 1 + echo -n "." + done + echo "" + echo "❌ $service_name failed to start within 30 seconds" + return 1 +} + +# Stop any existing processes +echo "=== Cleaning up existing processes ===" +pkill -f "weed master" || true +pkill -f "weed filer" || true +pkill -f "weed mq.broker" || true +sleep 2 + +echo "" +echo "=== Starting SeaweedFS Components ===" + +# Start Master +if ! check_service "localhost:9333" "SeaweedFS Master"; then + start_service \ + "weed master -defaultReplication=001 -mdir=$BASE_DIR/master" \ + "SeaweedFS Master" \ + "$LOG_DIR/master.log" \ + "localhost:9333" + echo "" +fi + +# Start Filer +if ! check_service "localhost:8888" "SeaweedFS Filer"; then + start_service \ + "weed filer -master=localhost:9333 -filer.dir=$BASE_DIR/filer" \ + "SeaweedFS Filer" \ + "$LOG_DIR/filer.log" \ + "localhost:8888" + echo "" +fi + +# Start MQ Broker +if ! check_service "localhost:17777" "SeaweedFS MQ Broker"; then + start_service \ + "weed mq.broker -filer=localhost:8888 -master=localhost:9333" \ + "SeaweedFS MQ Broker" \ + "$LOG_DIR/broker.log" \ + "localhost:17777" + echo "" +fi + +echo "=== Infrastructure Status ===" +check_service "localhost:9333" "Master (gRPC)" +check_service "localhost:9334" "Master (HTTP)" +check_service "localhost:8888" "Filer (HTTP)" +check_service "localhost:18888" "Filer (gRPC)" +check_service "localhost:17777" "MQ Broker" + +echo "" +echo "=== Infrastructure Ready ===" +echo "Log files:" +echo " Master: $LOG_DIR/master.log" +echo " Filer: $LOG_DIR/filer.log" +echo " Broker: $LOG_DIR/broker.log" +echo "" +echo "To view logs in real-time:" +echo " tail -f $LOG_DIR/master.log" +echo " tail -f $LOG_DIR/filer.log" +echo " tail -f $LOG_DIR/broker.log" +echo "" +echo "To stop all services:" +echo " pkill -f \"weed master\"" +echo " pkill -f \"weed filer\"" +echo " pkill -f \"weed mq.broker\"" +echo "" +echo "[OK] SeaweedFS infrastructure is ready for testing!" + |
