dragonfly/deploy/docker-compose/run.sh

106 lines
3.3 KiB
Bash
Executable File

#!/bin/sh
set -e
REPO=${REPO:-dragonflyoss}
TAG=${TAG:-latest}
DIR=$(cd "$(dirname "$0")" && pwd)
cd $DIR
prepare(){
mkdir -p config
ip=${IP:-$(hostname -i)}
sed "s,__IP__,$ip," template/dfget.template.yaml > config/dfget.yaml
sed "s,__IP__,$ip," template/seed-peer.template.yaml > config/seed-peer.yaml
sed "s,__IP__,$ip," template/scheduler.template.yaml > config/scheduler.yaml
sed "s,__IP__,$ip," template/manager.template.yaml > config/manager.yaml
}
run_container(){
RUNTIME=${RUNTIME:-docker}
echo use container runtime: ${RUNTIME}
echo try to clean old containers
${RUNTIME} rm -f dragonfly-redis dragonfly-mysql dragonfly-manager dragonfly-scheduler \
dragonfly-dfdaemon dragonfly-seed-peer
printf "create dragonfly-redis "
${RUNTIME} run -d --name dragonfly-redis --restart=always -p 6379:6379 \
redis:6-alpine \
--requirepass "dragonfly"
printf "create dragonfly-mysql "
${RUNTIME} run -d --name dragonfly-mysql --restart=always -p 3306:3306 \
--env MARIADB_USER="dragonfly" \
--env MARIADB_PASSWORD="dragonfly" \
--env MARIADB_DATABASE="manager" \
--env MARIADB_ALLOW_EMPTY_ROOT_PASSWORD="yes" \
mariadb:10.6
printf "create dragonfly-manager "
${RUNTIME} run -d --name dragonfly-manager --restart=always --net=host \
-v /tmp/log/dragonfly:/var/log/dragonfly \
-v ${DIR}/config/manager.yaml:/etc/dragonfly/manager.yaml \
${REPO}/manager:${TAG}
printf "create dragonfly-seed-peer "
${RUNTIME} run -d --name dragonfly-seed-peer --restart=always --net=host \
-v /tmp/log/dragonfly:/var/log/dragonfly \
-v ${DIR}/config/seed-peer.yaml:/etc/dragonfly/dfget.yaml \
${REPO}/dfdaemon:${TAG}
printf "create dragonfly-scheduler "
${RUNTIME} run -d --name dragonfly-scheduler --restart=always --net=host \
-v /tmp/log/dragonfly:/var/log/dragonfly \
-v ${DIR}/config/scheduler.yaml:/etc/dragonfly/scheduler.yaml \
${REPO}/scheduler:${TAG}
printf "create dragonfly-dfdaemon "
${RUNTIME} run -d --name dragonfly-dfdaemon --restart=always --net=host \
-v /tmp/log/dragonfly:/var/log/dragonfly \
-v ${DIR}/config/dfget.yaml:/etc/dragonfly/dfget.yaml \
${REPO}/dfdaemon:${TAG}
}
prepare
case "$1" in
container)
run_container
;;
*)
if [ -z "$1" ]; then
# start all of docker-compose defined service
COMPOSE=docker-compose
# use docker compose plugin
if docker compose version; then
COMPOSE="docker compose"
fi
$COMPOSE up -d
# docker-compose version 3 depends_on does not wait for redis and mysql to be “ready” before starting manager ...
# doc https://docs.docker.com/compose/compose-file/compose-file-v3/#depends_on
for i in $(seq 0 10); do
service_num=$($COMPOSE ps --services |wc -l)
ready_num=$($COMPOSE ps | grep healthy | wc -l)
if [ "$service_num" -eq "$ready_num" ]; then
break
fi
echo "wait for all service ready: $ready_num/$service_num,$i times check"
sleep 2
done
# print service list info
$COMPOSE ps
exit 0
fi
echo "unknown argument: $1"
exit 1
;;
esac