Improve cleanup and run scripts for local dev

Author: chupe

hack/cleanup.sh

@@ -61,16 +61,19 @@ cleanup_env() {
 for env in "${ENVS_TO_CLEAN[@]}"; do
     case $env in
         rs)
-            cleanup_env "rs" "$SCRIPT_DIR/rs/compose.yml"
-            docker volume ls -q | grep -E '^rs[0-9]+' | xargs -r docker volume rm 2>/dev/null || true
+            cleanup_env "rs" "$SCRIPT_DIR/rs/compose.yml" "r1"
+            docker volume ls -q | grep -E '^r1_rs[0-9]+' | xargs -r docker volume rm 2>/dev/null || true
+            docker network ls -q --filter "name=r1_" | xargs -r docker network rm 2>/dev/null || true
             ;;
         sh)
             cleanup_env "sh" "$SCRIPT_DIR/sh/compose.yml" "s1"
             docker volume ls -q | grep -E '^(src-|tgt-|s1_)' | xargs -r docker volume rm 2>/dev/null || true
+            docker network ls -q --filter "name=s1_" | xargs -r docker network rm 2>/dev/null || true
             ;;
         sh-ha)
             cleanup_env "sh-ha" "$SCRIPT_DIR/sh-ha/compose.yml" "s1"
             docker volume ls -q | grep -E '^s1_' | xargs -r docker volume rm 2>/dev/null || true
+            docker network ls -q --filter "name=s1_" | xargs -r docker network rm 2>/dev/null || true
             ;;
     esac
 done

hack/rs/run.sh

@@ -1,6 +1,19 @@
 #!/usr/bin/env bash
+# Start MongoDB replica sets for source and target
+#
+# Usage:
+#   ./run.sh           # Default: start both rs0 and rs1
+#   RS_COUNT=1 ./run.sh   # Start only rs0
+#   RS_COUNT=2 ./run.sh   # Start both rs0 and rs1
+#
+# Limits: Max 2 replica sets (rs0 and rs1)
+# Each replica set has 3 members (rs{N}0, rs{N}1, rs{N}2)
+# Ports: rs0 uses 30000-30002, rs1 uses 30100-30102
+
+export COMPOSE_PROJECT_NAME=r1
 
 BASE=$(dirname "$(dirname "$0")")
+echo "BASE=$BASE"
 
 # shellcheck source=/dev/null
 source "$BASE/util"
@@ -9,16 +22,55 @@ RDIR="$BASE/rs"
 
 export compose="$RDIR/compose.yml"
 
-dcf up -d rs00 rs01 rs02 rs10 rs11 rs12
+# Port mappings for each replica set (must match compose.yml)
+# rs0: 30000, 30001, 30002
+# rs1: 30100, 30101, 30102
+RS_PORTS=(
+    "30000 30001 30002"
+    "30100 30101 30102"
+)
+
+# Dynamically defined by the length of sets of ports
+MAX_RS_COUNT=${#RS_PORTS[@]}
+
+# Configuration: Number of replica sets to start
+# Can be overridden with environment variable: RS_COUNT=1 ./run.sh
+RS_COUNT="${RS_COUNT:-$MAX_RS_COUNT}"
+
+if [ "$RS_COUNT" -gt "$MAX_RS_COUNT" ]; then
+    echo "ERROR: RS_COUNT=$RS_COUNT exceeds maximum $MAX_RS_COUNT"
+    echo "Available init scripts: $RDIR/mongo/scripts/rs{0..$((MAX_RS_COUNT-1))}.js"
+    exit 1
+fi
+
+if [ "$RS_COUNT" -lt 1 ]; then
+    echo "ERROR: RS_COUNT=$RS_COUNT must be at least 1"
+    exit 1
+fi
+
+echo "Starting $RS_COUNT replica set(s) (max: $MAX_RS_COUNT)"
+
+# Build list of services to start and initialize each replica set
+SERVICES=""
+for rs_idx in $(seq 0 $((RS_COUNT - 1))); do
+    echo "Starting replica set rs${rs_idx}"
+    SERVICES="$SERVICES rs${rs_idx}0 rs${rs_idx}1 rs${rs_idx}2"
+done
 
-mwait "rs00:30000"
-mwait "rs01:30001"
-mwait "rs02:30002"
+# Start the selected services
+dcf up -d $SERVICES
 
-rsinit "scripts/rs0" "rs00:30000"
+# Initialize each replica set
+for rs_idx in $(seq 0 $((RS_COUNT - 1))); do
+    # Get ports for this replica set
+    read -r -a PORTS <<< "${RS_PORTS[$rs_idx]}"
 
-mwait "rs10:30100"
-mwait "rs11:30101"
-mwait "rs12:30102"
+    # Wait for all members
+    for member_idx in 0 1 2; do
+        PORT=${PORTS[$member_idx]}
+        mwait "rs${rs_idx}${member_idx}:${PORT}"
+    done
 
-rsinit "scripts/rs1" "rs10:30100"
+    # Initialize replica set
+    rsinit "scripts/rs${rs_idx}" "rs${rs_idx}0:${PORTS[0]}"
+done