Files
sientia-dataops-model-manager/setup_port_forwards.sh

112 lines
3.7 KiB
Bash
Executable File

#!/bin/bash
# Exit on any error
set -e
# Color codes for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color
echo -e "${GREEN}=== Port Forward Setup Script ===${NC}\n"
# Define port forwards: LOCAL_PORT:NAMESPACE:SERVICE:REMOTE_PORT:DESCRIPTION
PORT_FORWARDS=(
"55432:paradedb:paradedb-rw:5432:PostgreSQL"
"45249:sientia-tracker:sientia-tracker-mlflow-tracking:80:MLflow"
"37463:temporal:temporal-frontend:7233:Temporal"
"8080:temporal:temporal-web:8080:Temporal UI"
"42297:mongodb:my-release-mongodb:27017:MongoDB"
"36577:minio:minio:9000:MinIO"
)
# Step 1: Kill existing port-forward jobs for these services
echo -e "${YELLOW}Step 1: Checking for existing port-forward jobs...${NC}"
for pf in "${PORT_FORWARDS[@]}"; do
IFS=':' read -r local_port namespace service remote_port description <<< "$pf"
# Check if there's a job with this service name
existing_jobs=$(jobs -l | grep "kubectl.*port-forward.*svc/$service" || true)
if [ -n "$existing_jobs" ]; then
echo -e "${YELLOW} Found existing port-forward for $description ($service)${NC}"
# Extract PIDs and kill them
pids=$(echo "$existing_jobs" | awk '{print $2}')
for pid in $pids; do
echo -e "${YELLOW} Killing job with PID $pid${NC}"
kill "$pid" 2>/dev/null || true
done
fi
done
# Wait a moment for ports to be released
sleep 1
echo -e "${GREEN} Cleanup complete${NC}\n"
# Step 2: Check if any of the ports are already in use
echo -e "${YELLOW}Step 2: Checking if ports are available...${NC}"
ports_in_use=()
for pf in "${PORT_FORWARDS[@]}"; do
IFS=':' read -r local_port namespace service remote_port description <<< "$pf"
# Check if port is in use using lsof or netstat
if command -v lsof &> /dev/null; then
if lsof -Pi :$local_port -sTCP:LISTEN -t >/dev/null 2>&1; then
ports_in_use+=("$local_port:$description")
fi
elif command -v netstat &> /dev/null; then
if netstat -tuln | grep -q ":$local_port "; then
ports_in_use+=("$local_port:$description")
fi
elif command -v ss &> /dev/null; then
if ss -tuln | grep -q ":$local_port "; then
ports_in_use+=("$local_port:$description")
fi
fi
done
# If any ports are in use, report and exit
if [ ${#ports_in_use[@]} -gt 0 ]; then
echo -e "${RED}ERROR: The following ports are already in use:${NC}"
for port_info in "${ports_in_use[@]}"; do
IFS=':' read -r port desc <<< "$port_info"
echo -e "${RED} - Port $port (for $desc)${NC}"
done
echo -e "\n${RED}Please free these ports before running this script.${NC}"
exit 1
fi
echo -e "${GREEN} All ports are available${NC}\n"
# Step 3: Create all port forwards
echo -e "${YELLOW}Step 3: Creating port forwards...${NC}"
for pf in "${PORT_FORWARDS[@]}"; do
IFS=':' read -r local_port namespace service remote_port description <<< "$pf"
echo -e "${GREEN} Starting port-forward: $description${NC}"
echo -e " Local port: $local_port -> $namespace/$service:$remote_port"
kubectl -n "$namespace" port-forward "svc/$service" "$local_port:$remote_port" &
# Give it a moment to start
sleep 0.5
done
echo -e "\n${GREEN}=== All port forwards created successfully ===${NC}"
echo -e "\n${YELLOW}Active port forwards:${NC}"
for pf in "${PORT_FORWARDS[@]}"; do
IFS=':' read -r local_port namespace service remote_port description <<< "$pf"
echo -e " - ${GREEN}localhost:$local_port${NC} -> $description ($namespace/$service)"
done
echo -e "\n${YELLOW}To stop all port forwards, run:${NC}"
echo -e " jobs -p | xargs kill"
echo -e "\n${YELLOW}To view active port forwards:${NC}"
echo -e " jobs -l"