cloudstack/agent-simulator/scripts/zucchini/__old__/deployBulk-sg.sh

#!/usr/bin/env bash
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
# 
#   http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
  
usage() {
  printf "Deploy many VMs: %s: -h management-server -n numberofvms [[-b batchsize] [-w wait for success] [-d db-node]]\n" $(basename $0) >&2
  printf "-w option when specifed waits for successful deployment of batchsize (specifed by -b option) number of VMs. default batchsize =100\n"
}

getSgListRandom() {
    num_groups=$((RANDOM%groups_per_vm))
    if [[ $num_groups -eq 0 ]]; then
        num_groups=1 #set back to default
    fi

    #form sg list string
    local sg_list=""
    for ((i=0;i<$num_groups;i++))
    do
        sgid=$((RANDOM%numberofgroups))
        if [[ $sgid -eq 0 || $sgid -eq 1 ]]; then
            sgid=1 #set back to default security group
            sg_list=$sgid","$sg_list            
            continue
        fi
        
        sg_exists="GET  http://$host/client/?command=listSecurityGroups&id=$sgid  HTTP/1.0\n\n"
        sg_out=$(echo -e $sg_exists | nc -v -q 60 $host 8096)
        count=$(echo $sg_out | sed 's/\(.*<count>\)\([0-9]*\)\(.*\)/\2/g')        
        if [[ $count != "1" ]]; then #FAIL: Invalid security group was randomly selected
            continue
        fi        
        sg_list=$sgid","$sg_list
    done
    echo "$sg_list"
}


waitDeploy() {
   local dbnode=$1
   local batchsize=$2
   while [ 1 ]
   do
  	    donecount=$(mysql -uroot -Dcloud -h$dbnode -s -r --skip-column-names -e"select count(*) from async_job where job_cmd like '%DeployVM%' and last_updated is null")
        echo "[DEBUG] " $(date) " " $donecount " VMs still deploying"
        if [[ $donecount == "0" || $donecount -eq 0 ]]
        then
           break
        fi
        sleep $(($donecount*2))s #2 seconds per VM
   done
}

hflag=
nflag=1
wflag=
bflag= 
dflag=
iterator=0

declare -a sg_array=('79' '79' '79' '79' '79' '79' '79' '79' '79')
#declare -a sg_array=('72' '73' '74' '75' '76' '77' '78' '79' '80')

host="127.0.0.1" #defaults to locahost
numberofvms=1040 #defaults
batchsize=100 #default
dbnode=
waitSuccess=false

while getopts 'h:n:b:d:w' OPTION
do
	case $OPTION in
		h)	  hflag=1
			host="$OPTARG"
			;;
		n)    nflag=1
			numberofvms="$OPTARG"
			;;
		w)    wflag=1
			waitSuccess=true
			;;
		b)    bflag=1
			batchsize="$OPTARG"
			;;
		d)    dflag=1
			dbnode="$OPTARG"
			;;
		?)	usage
			exit 2
			;;
	esac
done

if [ $hflag$nflag != "11" ]
then
 usage
 exit 2
fi

if [[ $wflag == "1" && $dflag != "1" ]]
then
   echo "please specify dbnode -d option"
   usage
   exit 2
fi


if [[ $bflag == "1" && $wflag != "1" ]]
then
   echo "-w option mandatory when -b is given"
   usage
   exit 2
fi

tag1=$(($numberofvms*5/13))
tag2=$(($numberofvms*2/13))
tag3=$(($numberofvms*6/13))

tag1_so=9 #defaults from a regular installation
tag2_so=10
tag3_so=11
vmcount=0

echo -n "Service Offering ID with TAG1 hosttag: "
read tag1_so

echo -n "Service Offering ID with TAG2 hosttag: "
read tag2_so

echo -n "Service Offering ID with TAG3 hosttag: "
read tag3_so

echo "Deploying TAG1 VMs with Service Offering: " $tag1_so
for ((c=1;c<$tag1;c++))
do
	if [[ $vmcount -eq $batchsize && waitSuccess ]]
	then
		waitDeploy $dbnode $batchsize
		vmcount=0
	fi
        sglist=${sg_array[$((iterator % 9))]}
        iterator=$((iterator+1))

        job_out=$(./deployVirtualMachine.sh -h $host -z 1 -t 2 -s $tag1_so -u -g $sglist)
				job_id=$(echo $job_out | sed 's/\(.*<jobid>\)\([0-9]*\)\(.*\)/\2/g') 
				vmid=$(echo $job_out | sed 's/\(.*<id>\)\([0-9]*\)\(.*\)/\2/g')
				echo "[DEBUG] $(date) deployed vm: " $vmid " in job: " $job_id

        vmcount=$((vmcount+1))
done

sleep 60s

echo "Deploying TAG2 VMs with Service Offering: " $tag2_so
for ((c=1;c<$tag2;c++))
do
	if [[ $vmcount -eq $batchsize && wflag == "1" ]]
	then
		waitDeploy $dbnode $batchsize
		vmcount=0
	fi
        sglist=${sg_array[$((iterator % 9))]}
        iterator=$((iterator+1))

        job_out=$(./deployVirtualMachine.sh -h $host -z 1 -t 2 -s $tag2_so -u -g $sglist)
				job_id=$(echo $job_out | sed 's/\(.*<jobid>\)\([0-9]*\)\(.*\)/\2/g') 
				vmid=$(echo $job_out | sed 's/\(.*<id>\)\([0-9]*\)\(.*\)/\2/g')
				echo "[DEBUG] $(date) deployed vm: " $vmid " in job: " $job_id

        vmcount=$((vmcount+1))
done

sleep 60s

echo "Deploying TAG3 VMs with Service Offering: " $tag3_so
for ((c=1;c<$tag3;c++))
do
	if [[ $vmcount -eq $batchsize && wflag == "1" ]]
	then
		waitDeploy $dbnode $batchsize
		vmcount=0
	fi
        sglist=${sg_array[$((iterator % 9))]}
        iterator=$((iterator+1))

        job_out=$(./deployVirtualMachine.sh -h $host -z 1 -t 2 -s $tag3_so -u -g $sglist)
				job_id=$(echo $job_out | sed 's/\(.*<jobid>\)\([0-9]*\)\(.*\)/\2/g') 
				vmid=$(echo $job_out | sed 's/\(.*<id>\)\([0-9]*\)\(.*\)/\2/g')
				echo "[DEBUG] $(date) deployed vm: " $vmid " in job: " $job_id

        vmcount=$((vmcount+1))
done