add_node_with_weave.sh

#!/bin/bash


print_help() {
  cat <<EOF
  use $0 master/slave docker_image add_node_begin_index add_node_end_index cpuset cpu_period cpu_quota memory disk_speed cpu_core block_size 
EOF
}

if [ $# != 11 -a $# != 4 ]; then 
print_help
exit
fi

server_type=$1
image=$2
begin=$3
end=$4
prefix="slave"
ip="192.168.0."
ip_suffix="/24"

cpuset=$5
cpu_period=$6
cpu_quota=$7
memory=$8
disk_speed=$9
cpu_core=${10}
block_size=${11}

memory_unit="G"
disk_speed_unit="mb"


if [ $server_type = "slave" ]; then
	#add 0.5G for each container
	container_memory=$(echo $memory + 0.5| bc)
	echo "container memory is $container_memory"
	for ((i = 1; i < $begin; i++))
	do
        	container="$prefix$i"
		# for the nodes already existed on the cluster, just need to append the new node ip to the /etc/hosts and hadoop slave file, so the last three paraemeters are 
		# meaningless for them, just use fake values
        	docker exec -d $container /root/add_node_with_weave.sh $begin $end 0 0 0
	done

	for ((i = $begin; i <= $end; i++))
	do
        	container="$prefix$i"
		memory_size="$container_memory$memory_unit"
		disk_speed_value="$disk_speed$disk_speed_unit"
		if [ $cpu_period = -1 -o $cpu_quota = -1 ]; then
		docker run -d --name $container -h $container --cpuset-cpus=$cpuset -m $memory_size \
										    --device-read-bps /dev/mapper/ECEVM01--vg-root:$disk_speed_value \
										    --device-write-bps /dev/mapper/ECEVM01--vg-root:$disk_speed_value \
										    $image /root/add_node_with_weave.sh $begin $end $memory $cpu_core $block_size
		else	
        		docker run -d --name $container -h $container --cpuset-cpus=$cpuset --cpu-period=$cpu_period --cpu-quota=$cpu_quota \
-m $memory_size --device-read-bps /dev/mapper/ECEVM01--vg-root:$disk_speed_value \
	--device-write-bps /dev/mapper/ECEVM01--vg-root:$disk_speed_value $image /root/add_node_with_weave.sh $begin $end $memory $cpu_core $block_size
		fi
		#add ip with weave
		tm=$[i + 2]
		ipaddr="$ip$tm$ip_suffix"
		weave attach $ipaddr $container
        	docker exec -d $container /root/start_node.sh $begin $end
	done

elif [ $server_type = "master" ]; then
	docker exec -d master /root/add_node_with_weave.sh $begin $end 0 0 0 # the last three parameters are meaningless for master, so we give it fake values 
fi