utils/infra_setup/vm_dev_setup/setup_env.sh

   1 #!/bin/bash
   2
   3 set -ex
   4
   5 wait_vm_ok() {
   6     ip=$1
   7
   8     retry=0
   9     until timeout 1s ssh $ssh_args ec2-user@$ip "exit" >/dev/null 2>&1
  10     do
  11         echo "retry connect rubbos vm ip $ip $retry"
  12         sleep 1
  13         let retry+=1
  14         if [[ $retry -ge $2 ]];then
  15             echo "rubbos control start timeout !!!"
  16             exit 1
  17         fi
  18     done
  19 }
  20
  21 bottlenecks_prepare_env()
  22 {
  23     echo "Bottlenecks prepare env in VMs"
  24
  25     # configue rubbos control ssh key
  26     generate_ssh_key
  27
  28     # wait all other VMs ok
  29     for i in $rubbos_benchmark $rubbos_client1 $rubbos_client2 \
  30              $rubbos_client3 $rubbos_client4 $rubbos_httpd $rubbos_mysql1 $rubbos_tomcat1
  31     do
  32         wait_vm_ok $i 120
  33     done
  34
  35     # asynchronous configue other VMs
  36     for i in $rubbos_benchmark $rubbos_client1 $rubbos_client2 \
  37              $rubbos_client3 $rubbos_client4 $rubbos_httpd $rubbos_mysql1 $rubbos_tomcat1
  38     do
  39           scp $ssh_args -r $SCRIPT_DIR ec2-user@$i:$SCRIPT_DIR
  40           ssh $ssh_args ec2-user@$i "sudo bash $SCRIPT_DIR/vm_prepare_setup.sh" &
  41     done
  42
  43     # ugly use ssh execute script to fix ec2-user previlege issue
  44     ssh $ssh_args ec2-user@$rubbos_control "sudo bash $SCRIPT_DIR/vm_prepare_setup.sh"
  45 }
  46
  47 bottlenecks_download_repo()
  48 {
  49     echo "Bottlenecks: download bottlenecks repo"
  50
  51     if [ -d $BOTTELENECKS_REPO_DIR/.git ]; then
  52         cd $BOTTLENECKS_REPO_DIR
  53         sudo git pull origin master
  54         cd -
  55     else
  56         sudo rm -rf $BOTTLENECKS_REPO_DIR
  57         sudo git clone $BOTTLENECKS_REPO $BOTTLENECKS_REPO_DIR
  58     fi
  59 }
  60
  61 bottlenecks_config_hosts_ip()
  62 {
  63     sed -i -e "s/REPLACE_CONTROL_HOST/$rubbos_control/g" \
  64            -e "s/REPLACE_HTTPD_HOST/$rubbos_httpd/g" \
  65            -e "s/REPLACE_MYSQL1_HOST/$rubbos_mysql1/g" \
  66            -e "s/REPLACE_TOMCAT1_HOST/$rubbos_tomcat1/g" \
  67            -e "s/REPLACE_CLIENT1_HOST/$rubbos_client1/g" \
  68            -e "s/REPLACE_CLIENT2_HOST/$rubbos_client2/g" \
  69            -e "s/REPLACE_CLIENT3_HOST/$rubbos_client3/g" \
  70            -e "s/REPLACE_CLIENT4_HOST/$rubbos_client4/g" \
  71            -e "s/REPLACE_BENCHMARK_HOST/$rubbos_benchmark/g" \
  72            $BOTTELENECKS_REPO_DIR/rubbos/rubbos_scripts/1-1-1/set_bottlenecks_rubbos_env.sh
  73 }
  74
  75 bottlenecks_download_packages()
  76 {
  77     echo "Bottlenecks: download rubbos dependent packages from artifacts"
  78
  79     curl --connect-timeout 10 -o /tmp/app_tools.tar.gz $RUBBOS_APP_TOOLS_URL 2>/dev/null
  80     sudo tar zxvf /tmp/app_tools.tar.gz -C $RUBBOS_DIR
  81     rm -rf /tmp/app_tools.tar.gz
  82     curl --connect-timeout 10 -o /tmp/rubbosMulini6.tar.gz $RUBBOS_MULINI6_URL 2>/dev/null
  83     sudo tar zxvf /tmp/rubbosMulini6.tar.gz -C $RUBBOS_MULINI6_DIR
  84     rm -rf /tmp/rubbosMulini6.tar.gz
  85 }
  86
  87 bottlenecks_rubbos_install_exe()
  88 {
  89     echo "Bottlenecks: install and run rubbos"
  90
  91     cd $RUBBOS_RUN_DIR
  92     sudo ./run.sh
  93     cd $RUBBOS_EXE_DIR
  94     sudo ./CONTROL_rubbos_exec.sh
  95 }
  96
  97 main()
  98 {
  99     SCRIPT_DIR=`cd ${BASH_SOURCE[0]%/*};pwd`
 100
 101     ssh_args="-o StrictHostKeyChecking=no -o BatchMode=yes"
 102     source $SCRIPT_DIR/package.conf
 103     source $SCRIPT_DIR/hosts.conf
 104     source $SCRIPT_DIR/common.sh
 105
 106     bottlenecks_prepare_env
 107     bottlenecks_download_repo
 108     bottlenecks_download_packages
 109     bottlenecks_rubbos_install_exe
 110 }
 111
 112 main
 113 set +ex