mirror of
https://github.com/Tencent/WeKnora.git
synced 2026-10-31 03:47:56 +08:00
wizardchen
77b6133e2a
- Introduced new MCP services functionality, including management routes and integration with the agent service. - Updated Docker Compose configuration to include a new SQL migration for MCP services. - Enhanced Makefile with new migration commands for versioning and creating migrations. - Improved chat component to handle selected MCP services in message processing. - Updated migration documentation to reflect new strategies and added MCP services migration files.
81 lines
2.6 KiB
Bash
Executable File
81 lines
2.6 KiB
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
|
|
# Database connection details (can be overridden by environment variables)
|
|
DB_HOST=${DB_HOST:-localhost}
|
|
DB_PORT=${DB_PORT:-5432}
|
|
DB_USER=${DB_USER:-postgres}
|
|
DB_PASSWORD=${DB_PASSWORD:-postgres}
|
|
DB_NAME=${DB_NAME:-WeKnora}
|
|
|
|
# Use versioned migrations directory
|
|
MIGRATIONS_DIR="${MIGRATIONS_DIR:-migrations/versioned}"
|
|
|
|
# Check if migrate tool is installed
|
|
if ! command -v migrate &> /dev/null; then
|
|
echo "Error: migrate tool is not installed"
|
|
echo "Install it with: go install -tags 'postgres' github.com/golang-migrate/migrate/v4/cmd/migrate@latest"
|
|
exit 1
|
|
fi
|
|
|
|
# Construct the database URL
|
|
DB_URL="postgres://${DB_USER}:${DB_PASSWORD}@${DB_HOST}:${DB_PORT}/${DB_NAME}?sslmode=disable"
|
|
|
|
# Execute migration based on command
|
|
case "$1" in
|
|
up)
|
|
echo "Running migrations up..."
|
|
echo "DB_URL: ${DB_URL}"
|
|
echo "DB_USER: ${DB_USER}"
|
|
echo "DB_PASSWORD: ${DB_PASSWORD}"
|
|
echo "DB_HOST: ${DB_HOST}"
|
|
echo "DB_PORT: ${DB_PORT}"
|
|
echo "DB_NAME: ${DB_NAME}"
|
|
echo "MIGRATIONS_DIR: ${MIGRATIONS_DIR}"
|
|
migrate -path ${MIGRATIONS_DIR} -database ${DB_URL} up
|
|
;;
|
|
down)
|
|
echo "Running migrations down..."
|
|
migrate -path ${MIGRATIONS_DIR} -database ${DB_URL} down
|
|
;;
|
|
create)
|
|
if [ -z "$2" ]; then
|
|
echo "Error: Migration name is required"
|
|
echo "Usage: $0 create <migration_name>"
|
|
exit 1
|
|
fi
|
|
echo "Creating migration files for $2..."
|
|
migrate create -ext sql -dir ${MIGRATIONS_DIR} -seq $2
|
|
echo "Created:"
|
|
echo " - ${MIGRATIONS_DIR}/$(ls -t ${MIGRATIONS_DIR} | head -1)"
|
|
echo " - ${MIGRATIONS_DIR}/$(ls -t ${MIGRATIONS_DIR} | head -2 | tail -1)"
|
|
;;
|
|
version)
|
|
echo "Checking current migration version..."
|
|
migrate -path ${MIGRATIONS_DIR} -database ${DB_URL} version
|
|
;;
|
|
force)
|
|
if [ -z "$2" ]; then
|
|
echo "Error: Version number is required"
|
|
echo "Usage: $0 force <version>"
|
|
exit 1
|
|
fi
|
|
echo "Forcing migration version to $2..."
|
|
migrate -path ${MIGRATIONS_DIR} -database ${DB_URL} force $2
|
|
;;
|
|
goto)
|
|
if [ -z "$2" ]; then
|
|
echo "Error: Version number is required"
|
|
echo "Usage: $0 goto <version>"
|
|
exit 1
|
|
fi
|
|
echo "Migrating to version $2..."
|
|
migrate -path ${MIGRATIONS_DIR} -database ${DB_URL} goto $2
|
|
;;
|
|
*)
|
|
echo "Usage: $0 {up|down|create <migration_name>|version|force <version>|goto <version>}"
|
|
exit 1
|
|
;;
|
|
esac
|
|
|
|
echo "Migration command completed successfully" |