wehub-resource-sync
101 lines
4.2 KiB
Bash
Executable File
101 lines
4.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# ============================================================================
|
|
# stop-demo.sh — one-command teardown for the banking demo (self-hosted mode).
|
|
#
|
|
# cd examples/showcases/banking && ./stop-demo.sh
|
|
#
|
|
# The companion to run-demo.sh. That script detaches everything except the
|
|
# Next.js dev server (`docker compose up -d`, native TEI via `nohup … & disown`,
|
|
# then `exec pnpm dev`), so Ctrl-C on the dev server leaves the docker stack and
|
|
# the native embedder running. This script brings those leftovers down.
|
|
#
|
|
# It tears down, in order:
|
|
# - the Next.js dev server on :3000 (defensive; usually already gone via Ctrl-C)
|
|
# - the docker compose stack (project `banking-memory`) — containers only by
|
|
# default, so a re-run of run-demo.sh reuses the built image + seeded data
|
|
# - the native Metal TEI on :7067 (Apple Silicon only; the host process
|
|
# run-demo.sh started outside docker's knowledge)
|
|
#
|
|
# Idempotent: safe to re-run — anything already down is skipped.
|
|
#
|
|
# Flags:
|
|
# --purge also delete the docker volumes (postgres/redis/minio/tei cache).
|
|
# Full reset: next run-demo.sh re-seeds the DB and re-downloads the
|
|
# embedding model. Without this, data + model cache persist.
|
|
# --keep-tei leave the native Metal TEI running (it's slow to warm up; handy
|
|
# if you're only bouncing the stack and want to skip the reload).
|
|
# ============================================================================
|
|
set -euo pipefail
|
|
|
|
DEMO_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
cd "$DEMO_DIR"
|
|
|
|
PURGE=0
|
|
KEEP_TEI=0
|
|
for arg in "$@"; do
|
|
case "$arg" in
|
|
--purge) PURGE=1 ;;
|
|
--keep-tei) KEEP_TEI=1 ;;
|
|
# Print only the leading banner: skip the shebang, then every comment
|
|
# line up to the first non-comment line (stops before the code body).
|
|
-h|--help) awk 'NR>1 && !/^#/{exit} NR>1{sub(/^# ?/,""); print}' "$0"; exit 0 ;;
|
|
*) printf 'unknown flag: %s (try --help)\n' "$arg" >&2; exit 2 ;;
|
|
esac
|
|
done
|
|
|
|
say() { printf '\n\033[1;36m==> %s\033[0m\n' "$*"; }
|
|
ok() { printf ' \033[1;32m✓\033[0m %s\n' "$*"; }
|
|
warn(){ printf ' \033[1;33m!\033[0m %s\n' "$*"; }
|
|
|
|
# Kill whatever is listening on a TCP port (best-effort, no error if nothing is).
|
|
kill_port() { # port label
|
|
local port="$1" label="$2" pids
|
|
pids="$(lsof -ti "tcp:${port}" -sTCP:LISTEN 2>/dev/null || true)"
|
|
if [ -n "$pids" ]; then
|
|
# shellcheck disable=SC2086 # word-splitting the pid list is intentional
|
|
kill $pids 2>/dev/null || true
|
|
sleep 1
|
|
# Escalate to SIGKILL for anything that ignored SIGTERM.
|
|
pids="$(lsof -ti "tcp:${port}" -sTCP:LISTEN 2>/dev/null || true)"
|
|
# shellcheck disable=SC2086
|
|
[ -n "$pids" ] && kill -9 $pids 2>/dev/null || true
|
|
ok "$label on :$port stopped"
|
|
else
|
|
ok "$label on :$port already stopped"
|
|
fi
|
|
}
|
|
|
|
# --- Next.js dev server -----------------------------------------------------
|
|
say "Stopping the Next.js dev server (:3000)"
|
|
kill_port 3000 "dev server"
|
|
|
|
# --- Docker stack -----------------------------------------------------------
|
|
# run-demo.sh may have brought the stack up with or without the cpu-fallback
|
|
# `tei` profile. `down` ignores unknown profiles, but pass --profile so the
|
|
# profiled `tei` container is included in the teardown on amd64/CI.
|
|
say "Stopping the docker stack (project banking-memory)"
|
|
if docker info >/dev/null 2>&1; then
|
|
DOWN_ARGS=(--profile cpu-fallback down --remove-orphans)
|
|
if [ "$PURGE" -eq 1 ]; then
|
|
DOWN_ARGS+=(--volumes)
|
|
warn "--purge: deleting volumes (postgres data, redis, minio, tei model cache)"
|
|
fi
|
|
docker compose "${DOWN_ARGS[@]}"
|
|
ok "docker stack down${PURGE:+ (volumes removed)}"
|
|
else
|
|
warn "Docker is not running — assuming the stack is already down"
|
|
fi
|
|
|
|
# --- Native Metal TEI (Apple Silicon) ---------------------------------------
|
|
# The one piece docker doesn't manage: run-demo.sh starts text-embeddings-router
|
|
# on the host with nohup/disown. Only present on arm64; a no-op elsewhere.
|
|
if [ "$KEEP_TEI" -eq 1 ]; then
|
|
say "Leaving native Metal TEI running (:7067) — --keep-tei"
|
|
else
|
|
say "Stopping native Metal TEI (:7067)"
|
|
kill_port 7067 "native TEI"
|
|
fi
|
|
|
|
say "Demo stopped."
|
|
[ "$PURGE" -eq 0 ] && printf ' (volumes kept — re-run ./run-demo.sh for a warm restart; use --purge for a clean slate)\n'
|