114 lines
2.7 KiB
Bash
Executable File
114 lines
2.7 KiB
Bash
Executable File
#!/bin/bash
|
|
# Horus Stack Benchmark Runner
|
|
# This script ensures the Horus stack is running before executing benchmarks
|
|
|
|
set -e
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
|
|
|
|
# Colors for output
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
YELLOW='\033[1;33m'
|
|
NC='\033[0m' # No Color
|
|
|
|
# Configuration
|
|
SUPERVISOR_URL="http://127.0.0.1:3030"
|
|
OSIRIS_URL="http://127.0.0.1:8081"
|
|
REDIS_URL="127.0.0.1:6379"
|
|
|
|
echo -e "${GREEN}=== Horus Stack Benchmark Runner ===${NC}\n"
|
|
|
|
# Function to check if a service is running
|
|
check_service() {
|
|
local url=$1
|
|
local name=$2
|
|
|
|
if curl -s -f "$url/health" > /dev/null 2>&1 || curl -s -f "$url" > /dev/null 2>&1; then
|
|
echo -e "${GREEN}✓${NC} $name is running"
|
|
return 0
|
|
else
|
|
echo -e "${RED}✗${NC} $name is not running"
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
# Function to check if Redis is running
|
|
check_redis() {
|
|
if redis-cli -h 127.0.0.1 -p 6379 ping > /dev/null 2>&1; then
|
|
echo -e "${GREEN}✓${NC} Redis is running"
|
|
return 0
|
|
else
|
|
echo -e "${RED}✗${NC} Redis is not running"
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
# Check prerequisites
|
|
echo "Checking prerequisites..."
|
|
echo ""
|
|
|
|
REDIS_OK=false
|
|
OSIRIS_OK=false
|
|
SUPERVISOR_OK=false
|
|
|
|
if check_redis; then
|
|
REDIS_OK=true
|
|
fi
|
|
|
|
if check_service "$OSIRIS_URL" "Osiris"; then
|
|
OSIRIS_OK=true
|
|
fi
|
|
|
|
if check_service "$SUPERVISOR_URL" "Supervisor"; then
|
|
SUPERVISOR_OK=true
|
|
fi
|
|
|
|
echo ""
|
|
|
|
# If any service is not running, provide instructions
|
|
if [ "$REDIS_OK" = false ] || [ "$OSIRIS_OK" = false ] || [ "$SUPERVISOR_OK" = false ]; then
|
|
echo -e "${YELLOW}Some services are not running. Please start the Horus stack:${NC}"
|
|
echo ""
|
|
|
|
if [ "$REDIS_OK" = false ]; then
|
|
echo " 1. Start Redis:"
|
|
echo " redis-server"
|
|
echo ""
|
|
fi
|
|
|
|
echo " 2. Start Horus stack:"
|
|
echo " cd $PROJECT_ROOT"
|
|
echo " RUST_LOG=info ./target/release/horus all --admin-secret SECRET --kill-ports"
|
|
echo ""
|
|
echo " Or run in the background:"
|
|
echo " RUST_LOG=info ./target/release/horus all --admin-secret SECRET --kill-ports &"
|
|
echo ""
|
|
|
|
read -p "Do you want to continue anyway? (y/N) " -n 1 -r
|
|
echo
|
|
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
|
|
echo -e "${RED}Benchmark cancelled.${NC}"
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
# Build the project first
|
|
echo -e "${GREEN}Building project...${NC}"
|
|
cd "$PROJECT_ROOT"
|
|
cargo build --release
|
|
|
|
echo ""
|
|
echo -e "${GREEN}Running benchmarks...${NC}"
|
|
echo ""
|
|
|
|
# Run benchmarks with any additional arguments passed to this script
|
|
cargo bench --bench horus_stack "$@"
|
|
|
|
echo ""
|
|
echo -e "${GREEN}=== Benchmark Complete ===${NC}"
|
|
echo ""
|
|
echo "Results saved to: target/criterion/"
|
|
echo "View HTML reports: open target/criterion/report/index.html"
|