93 lines
2.2 KiB
Bash
93 lines
2.2 KiB
Bash
|
#!/usr/bin/env sh
|
||
|
|
|
||
|
|
get_container_id() {
|
||
|
|
local compose_file=$1
|
||
|
|
local service=$2
|
||
|
|
if [ -z "${compose_file}" ] || [ -z "${service}" ]; then
|
||
|
|
echo "usage: get_container_id <compose_file> <service>"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
docker compose -f $compose_file ps --format json --status running \
|
||
|
|
| jq -r '.[]? | select(.Service == "'${service}'") | .ID'
|
||
|
|
}
|
||
|
|
|
||
|
|
# Exports all environment variables
|
||
|
|
export_env() {
|
||
|
|
export_pg_env
|
||
|
|
export_redis_env
|
||
|
|
}
|
||
|
|
|
||
|
|
# Exports postgres environment variables
|
||
|
|
export_pg_env() {
|
||
|
|
# Based on creds in compose.yaml
|
||
|
|
export PGPORT=5433
|
||
|
|
export PGHOST=localhost
|
||
|
|
export PGUSER=pg
|
||
|
|
export PGPASSWORD=password
|
||
|
|
export PGDATABASE=postgres
|
||
|
|
export DB_POSTGRES_URL="postgresql://pg:password@127.0.0.1:5433/postgres"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Exports redis environment variables
|
||
|
|
export_redis_env() {
|
||
|
|
export REDIS_HOST="127.0.0.1:6380"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Main entry point
|
||
|
|
main() {
|
||
|
|
# Expect a SERVICES env var to be set with the docker service names
|
||
|
|
local services=${SERVICES}
|
||
|
|
|
||
|
|
dir=$(dirname $0)
|
||
|
|
compose_file="${dir}/docker-compose.yaml"
|
||
|
|
|
||
|
|
# whether this particular script started the container(s)
|
||
|
|
started_container=false
|
||
|
|
|
||
|
|
# trap SIGINT and performs cleanup as necessary, i.e.
|
||
|
|
# taking down containers if this script started them
|
||
|
|
trap "on_sigint ${services}" INT
|
||
|
|
on_sigint() {
|
||
|
|
local services=$@
|
||
|
|
echo # newline
|
||
|
|
if $started_container; then
|
||
|
|
docker compose -f $compose_file rm -f --stop --volumes ${services}
|
||
|
|
fi
|
||
|
|
exit $?
|
||
|
|
}
|
||
|
|
|
||
|
|
# check if all services are running already
|
||
|
|
not_running=false
|
||
|
|
for service in $services; do
|
||
|
|
container_id=$(get_container_id $compose_file $service)
|
||
|
|
if [ -z $container_id ]; then
|
||
|
|
not_running=true
|
||
|
|
break
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
|
||
|
|
# if any are missing, recreate all services
|
||
|
|
if $not_running; then
|
||
|
|
docker compose -f $compose_file up --wait --force-recreate ${services}
|
||
|
|
started_container=true
|
||
|
|
else
|
||
|
|
echo "all services ${services} are already running"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# setup environment variables and run args
|
||
|
|
export_env
|
||
|
|
"$@"
|
||
|
|
# save return code for later
|
||
|
|
code=$?
|
||
|
|
|
||
|
|
# performs cleanup as necessary, i.e. taking down containers
|
||
|
|
# if this script started them
|
||
|
|
echo # newline
|
||
|
|
if $started_container; then
|
||
|
|
docker compose -f $compose_file rm -f --stop --volumes ${services}
|
||
|
|
fi
|
||
|
|
|
||
|
|
exit ${code}
|
||
|
|
}
|