63 lines
2.3 KiB
Bash
Executable File
63 lines
2.3 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Wrapper script to run the benchmark suite and automatically shut down when complete
|
|
|
|
set -e
|
|
|
|
# Determine docker-compose command
|
|
if docker compose version &> /dev/null 2>&1; then
|
|
DOCKER_COMPOSE="docker compose"
|
|
else
|
|
DOCKER_COMPOSE="docker-compose"
|
|
fi
|
|
|
|
# Clean old data directories (may be owned by root from Docker)
|
|
if [ -d "data" ]; then
|
|
echo "Cleaning old data directories..."
|
|
if ! rm -rf data/ 2>/dev/null; then
|
|
# If normal rm fails (permission denied), provide clear instructions
|
|
echo ""
|
|
echo "ERROR: Cannot remove data directories due to permission issues."
|
|
echo "This happens because Docker creates files as root."
|
|
echo ""
|
|
echo "Please run one of the following to clean up:"
|
|
echo " sudo rm -rf data/"
|
|
echo " sudo chown -R \$(id -u):\$(id -g) data/ && rm -rf data/"
|
|
echo ""
|
|
echo "Then run this script again."
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
# Stop any running containers from previous runs
|
|
echo "Stopping any running containers..."
|
|
$DOCKER_COMPOSE down 2>/dev/null || true
|
|
|
|
# Create fresh data directories with correct permissions
|
|
echo "Preparing data directories..."
|
|
|
|
# Clean Neo4j data to prevent "already running" errors
|
|
if [ -d "data/neo4j" ]; then
|
|
echo "Cleaning Neo4j data directory..."
|
|
rm -rf data/neo4j/*
|
|
fi
|
|
|
|
mkdir -p data/{next-orly-badger,next-orly-dgraph,next-orly-neo4j,dgraph-zero,dgraph-alpha,neo4j,neo4j-logs,khatru-sqlite,khatru-badger,relayer-basic,strfry,nostr-rs-relay,postgres}
|
|
chmod 777 data/{next-orly-badger,next-orly-dgraph,next-orly-neo4j,dgraph-zero,dgraph-alpha,neo4j,neo4j-logs,khatru-sqlite,khatru-badger,relayer-basic,strfry,nostr-rs-relay,postgres}
|
|
|
|
echo "Building fresh Docker images..."
|
|
# Force rebuild to pick up latest code changes
|
|
$DOCKER_COMPOSE build --no-cache benchmark-runner next-orly-badger next-orly-dgraph next-orly-neo4j
|
|
|
|
echo ""
|
|
echo "Starting benchmark suite..."
|
|
echo "This will automatically shut down all containers when the benchmark completes."
|
|
echo ""
|
|
|
|
# Run docker compose with flags to exit when benchmark-runner completes
|
|
$DOCKER_COMPOSE up --exit-code-from benchmark-runner --abort-on-container-exit
|
|
|
|
echo ""
|
|
echo "Benchmark suite has completed and all containers have been stopped."
|
|
echo "Check the ./reports/ directory for results."
|