#!/bin/bash
set -e
LC_ALL=C
LANG=C
unset TZ
file_path="./"
host="192.168.0.32"
username="test"
password="test.123"
dbname="test"
jmeter_path="./apache-jmeter-3.0/bin/jmeter.sh"
jmx_path="./apache-jmeter-3.0/jmxs/termplan_dq.jmx"
jtl_path="./apache-jmeter-3.0/jtls/`date +%Y%m%d-%H%M`.jtl"
sql_path="./sqlFiles/"
#now=`date +date +%Y%m%d-%H%M`

###############33######
####数据库初始化
###############33######
function init_data(){
        echo "===========开始初始化相关数据============="
        mysql -h${host} --default-character-set=utf8 -u${username} -p${password} ${dbname} -e"
        DELETE FROM t_user WHERE id > 50000;
        DELETE FROM t_person WHERE id > 50000;
        SELECT count(*) AS ‘压测前预约总数‘ FROM t_test_table;
        "

        cd ${sql_path}
        sql_name=`ls`
        for file_name in ${sql_name[@]}
        do
                if [ -f "$file_name" ];then
                        #command="source $file_name"
                        mysql -h${host} --default-character-set=utf8 -u${username} -p${password} ${dbname} -e "source ${file_name}"
                else
                        echo -e "\033[31;31m SQL文件不存在！！！\033[0m"
                fi
        done
        cd ..

        mysql -h${host} --default-character-set=utf8 -u${username} -p${password} ${dbname} -e"
        SELECT count(*) AS ‘新增用户数‘ FROM t_user WHERE id > 50000;
        "
        echo "==========已完成初始化相关数据============"
}
###############33######
####缓存初始化
###############33######
function init_cache(){
        echo "=============开始初始化缓存==============="
        echo "==============初始化(kiwi)================"
        curl "http://127.0.0.1:10000/test/test.action"
        echo -e "\n==========初始化冻结缓存(kiwi)============"
        curl "http://127.0.0.1:10000/test/test/zen.action"
        echo -e "\n============已完成初始化缓存=============="
}
###############33######
####执行jmeter脚本
###############33######
function auto_jmeter(){
        sh ${jmeter_path} -n -t ${jmx_path} -l ${jtl_path}
}
###############33######
####查询数据(数据太多无作用，因为缓存到数据库需要很长时间)
###############33######
function select_data(){
        echo "============开始查询相关数据=============="
        mysql -h${host} --default-character-set=utf8 -u${username} -p${password} ${dbname} -e"
        select count(*) as ‘压测后预约总数‘  from t_test_table;
        quit
        "
        echo "============相关数据查询完毕=============="
        echo -e "\033[36;36m============压力测试执行完毕==============\033[0m"
}
#######################
#######################
init_data
init_cache
auto_jmeter
sleep 10
select_data