make_docker_compose_yml.sh

#!/bin/bash

###
# // small setup
# $ zookeeper=1 namenode=1 datanode=1 ./make_docker_compose_file.sh hdfs hbase > docker-compose.yml
#
# // hdfs + hbase setup
#  ./make_docker_compose_file.sh hdfs hbase > docker-compose.yml
# 
# ./make_docker_compose_yml.sh [hdfs] [hbase] [yarn] [drill]
###

DEBUG=${DEBUG:-0}

# docker network name
network_name=${network_name:-"vnet"}

# zookeeper scale size
zookeeper=${zookeeper:-3}

# hdfs scale size
journalnode=${journalnode:-3}
namenode=${namenode:-2}
datanode=${datanode:-3}

if [ "$namenode" -eq 1 ]; then
    journalnode=0
fi

nn_ha=()
if [ "$namenode" -gt 1 ]; then
    for i in `seq 1 ${namenode}`; do
        nn_ha+=("nn${i}")
    done
fi
NAMENODE_HA="$(IFS=,; echo "${nn_ha[*]}")"

# yarn scale size
resourcemanager=${resourcemanager:-1}
historyserver=${historyserver:-1}
nodemanager=${nodemanager:-${datanode}}

# hbase scale size
hmaster=${hmaster:-${namenode}}
regionserver=${regionserver:-${datanode}}
hbasethrift=${hbasethrift:-0}

# drill scale size
drillbit=${drillbit:-1}

debug() {
  [ ${DEBUG} -gt 0 ] && echo "[DEBUG] $@" 1>&2
}
 
services=()

zk_quorum=()

for z in `seq ${zookeeper}`; do
    zk_quorum+=("zookeeper-$z.${network_name}:2181")
done
ZOOKEEPER_QUORUM="$(IFS=,; echo "${zk_quorum[*]}")"

zookeeper_arg=0
for component in $@; do 
    if [ "${component}" == "zookeeper" ]; then
        zookeeper_arg=1
        break
    fi
done   

if [ $zookeeper_arg -eq 0 ]; then
    set -- zookeeper $@
fi

for component in $@; do 
    template=services.${component}.yml.tpl
    if [ ! -e ${template} ]; then
        continue
    fi
    debug "Template: $template ..."
    
    service_keys=$(cat $template | sed -e '/^## /!d' | sed -e 's/^##[ ]*//g' )

    for k in $service_keys; do
        part=$(cat $template \
            | sed -e '/^## '${k}'/,/^##\/ '${k}'/!d' \
            | sed -e '/^##/d' \
        )  
        
        KK=$(echo $k | tr '[a-z]' '[A-Z]')

        debug $k $KK
        
        scale_size=${!k}
        debug $KK scale_size: $scale_size
        
        for i in `seq 1 ${scale_size}`; do 
            if [ ${i} -gt ${scale_size} ]; then
                break
            fi
            swarm_filter="SWARM_FILTER_${KK}_${i}" 
            debug $swarm_filter
            # replace template variable
            filter=""
            if [ "${!swarm_filter}" != "" ]; then
                filter="- ${!swarm_filter}"            
            fi
            _part=$(echo "$part" \
                | sed -e 's/${i}/'$i'/g' \
                      -e 's/${network_name}/'$network_name'/g'  \
                      -e 's/${scale_size}/'$scale_size'/g' \
                      -e 's/${ZOOKEEPER_QUORUM}/'$ZOOKEEPER_QUORUM'/g' \
                      -e 's/${NAMENODE_HA}/'"${NAMENODE_HA}"'/g' \
                | sed -e 's/${'${swarm_filter}'}/'"${filter}"'/g' \
            )
            services+=("$_part" "")
        done
    done
done

# join 
docker_compose_services="$(IFS=$'\n'; echo "${services[*]}")"

# output docker-compose.yml v2 format
cat <<EOD
version: "2"
services:

$docker_compose_services

networks:
  $network_name:
    external:
      name: $network_name 
EOD