mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-10-10 21:26:01 +02:00
21dc3a2456
* Add error clarity to restart containers script * erroneous cleanup on exit * fix when starting containers for the first time --------- Co-authored-by: Wenxi Onyx <wenxi-onyx@Wenxis-MacBook-Pro.local>
65 lines
2.4 KiB
Bash
Executable File
65 lines
2.4 KiB
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
|
|
cleanup() {
|
|
echo "Error occurred. Cleaning up..."
|
|
docker stop onyx_postgres onyx_vespa onyx_redis 2>/dev/null || true
|
|
docker rm onyx_postgres onyx_vespa onyx_redis 2>/dev/null || true
|
|
}
|
|
|
|
# Trap errors and output a message, then cleanup
|
|
trap 'echo "Error occurred on line $LINENO. Exiting script." >&2; cleanup' ERR
|
|
|
|
# Usage of the script with optional volume arguments
|
|
# ./restart_containers.sh [vespa_volume] [postgres_volume] [redis_volume]
|
|
|
|
VESPA_VOLUME=${1:-""} # Default is empty if not provided
|
|
POSTGRES_VOLUME=${2:-""} # Default is empty if not provided
|
|
REDIS_VOLUME=${3:-""} # Default is empty if not provided
|
|
|
|
# Stop and remove the existing containers
|
|
echo "Stopping and removing existing containers..."
|
|
docker stop onyx_postgres onyx_vespa onyx_redis 2>/dev/null || true
|
|
docker rm onyx_postgres onyx_vespa onyx_redis 2>/dev/null || true
|
|
|
|
# Start the PostgreSQL container with optional volume
|
|
echo "Starting PostgreSQL container..."
|
|
if [[ -n "$POSTGRES_VOLUME" ]]; then
|
|
docker run -p 5432:5432 --name onyx_postgres -e POSTGRES_PASSWORD=password -d -v $POSTGRES_VOLUME:/var/lib/postgresql/data postgres -c max_connections=250
|
|
else
|
|
docker run -p 5432:5432 --name onyx_postgres -e POSTGRES_PASSWORD=password -d postgres -c max_connections=250
|
|
fi
|
|
|
|
# Start the Vespa container with optional volume
|
|
echo "Starting Vespa container..."
|
|
if [[ -n "$VESPA_VOLUME" ]]; then
|
|
docker run --detach --name onyx_vespa --hostname vespa-container --publish 8081:8081 --publish 19071:19071 -v $VESPA_VOLUME:/opt/vespa/var vespaengine/vespa:8
|
|
else
|
|
docker run --detach --name onyx_vespa --hostname vespa-container --publish 8081:8081 --publish 19071:19071 vespaengine/vespa:8
|
|
fi
|
|
|
|
# Start the Redis container with optional volume
|
|
echo "Starting Redis container..."
|
|
if [[ -n "$REDIS_VOLUME" ]]; then
|
|
docker run --detach --name onyx_redis --publish 6379:6379 -v $REDIS_VOLUME:/data redis
|
|
else
|
|
docker run --detach --name onyx_redis --publish 6379:6379 redis
|
|
fi
|
|
|
|
# Ensure alembic runs in the correct directory
|
|
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
|
|
PARENT_DIR="$(dirname "$SCRIPT_DIR")"
|
|
cd "$PARENT_DIR"
|
|
|
|
# Give Postgres a second to start
|
|
sleep 1
|
|
|
|
# Run Alembic upgrade
|
|
echo "Running Alembic migration..."
|
|
alembic upgrade head
|
|
|
|
# Run the following instead of the above if using MT cloud
|
|
# alembic -n schema_private upgrade head
|
|
|
|
echo "Containers restarted and migration completed."
|