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

create-swarm-mode-cluster.sh

#!/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 "$@"