Script to create a #docker swarm mode cluster

Views: 1790

I found this script on GitHub.  I have been writing a similar one, with less functionality and flexibility, but that’s OK it is for Databases with volumes.

#!/bin/bash

Size=3

if [ -z “${DOCKER_MACHINE_DRIVER}” ]; then
DOCKER_MACHINE_DRIVER=virtualbox
fi

# REGISTRY_MIRROR_OPTS=”–engine-registry-mirror https://jxus37ac.mirror.aliyuncs.com”
# INSECURE_OPTS=”–engine-insecure-registry 192.168.99.0/24″
STORAGE_OPTS=”–engine-storage-driver overlay2″

MACHINE_OPTS=”${STORAGE_OPTS} ${INSECURE_OPTS} ${REGISTRY_MIRROR_OPTS}”

function create_nodes() {
for i in $(seq 1 ${Size})
do
# Create a docker host by docker-machine
set -xe
docker-machine create -d ${DOCKER_MACHINE_DRIVER} ${MACHINE_OPTS} node-${i}
set +xe
done
}

function remove_nodes() {
for i in $(seq 1 ${Size})
do
# Remove the docker host
docker-machine rm -y node-${i}
done
}

function create_swarm() {
NumberOfManager=$1

if [ -z “${NumberOfManager}” ]; then
# if nothing specified, just let everyone are managers.
NumberOfManager=${Size}
else
# NumberOfManager cannot be greater than the node size
if [ “${NumberOfManager}” -gt “${Size}” ]; then
NumberOfManager=${Size}
fi
fi

# Create Swarm Manager
ManagerIP=`docker-machine ip node-1`
echo “Manager IP: ${ManagerIP}”
echo “node-1 inits swarm and becomes a Manager”

set -xe
docker-machine ssh node-1 docker swarm init –advertise-addr ${ManagerIP}
set +xe

# Fetch Tokens
ManagerToken=`docker-machine ssh node-1 docker swarm join-token manager | grep token | awk ‘{ print $2 }’`
WorkerToken=`docker-machine ssh node-1 docker swarm join-token worker | grep token | awk ‘{ print $2 }’`

echo “Manager Token: ${ManagerToken}”
echo “Workder Token: ${WorkerToken}”

# Managers
if [ “${NumberOfManager}” -gt 2 ]; then
for i in $(seq 2 ${NumberOfManager})
do
echo “node-${i} join swarm as a Manager”

set -xe
docker-machine ssh node-${i} “docker swarm join –token ${ManagerToken} ${ManagerIP}:2377”
set +xe

done
fi

# Workers
if [ “${NumberOfManager}” -lt “${Size}” ]; then
for i in $(seq “$(expr ${NumberOfManager} + 1)” ${Size})
do
echo “node-${i} join swarm as a Worker”

set -xe
docker-machine ssh node-${i} “docker swarm join –token ${WorkerToken} ${ManagerIP}:2377”
set +xe

done
fi
}

function remove_swarm() {
for i in $(seq 1 ${Size})
do
# Leave Swarm
set -xe
docker-machine ssh node-${i} docker swarm leave –force
set +xe

done
}

function create() {
Command=$1
shift
case “${Command}” in
nodes) create_nodes “$@” ;;
swarm) create_swarm “$@” ;;
*) echo “Usage: $0 create <nodes|swarm>” ;;
esac
}

function remove() {
Command=$1
shift
case “${Command}” in
nodes) remove_nodes “$@” ;;
swarm) remove_swarm “$@” ;;
*) echo “Usage: $0 remove <nodes|swarm>” ;;
esac
}

function main() {
Command=$1
shift
case “${Command}” in
create) create “$@” ;;
remove) remove “$@” ;;
*) echo “Usage: $0 <create|remove>” ;;
esac
}

main “$@”

Source: Script to create a swarm mode cluster which introduced in Docker 1.12