#!/bin/bash
INST_OFFSET=`expr $UID % 63`
SIMULATE_SIZE=50000 # Unit: MB
LOG_SIZE=30000 # Unit: MB
declare inst_count=''
declare last_id=''
declare mes_cfg=''
DSS_PORT_BASE=30000

init_nodes_list()
{
    inst_count=$1
    last_id=`expr $inst_count - 1`
    for i in `seq 0 $last_id`
    do
        inst_id=`expr $i + $INST_OFFSET`
        port=`expr $i + $DSS_PORT_BASE`
        if [ $i != $last_id ]; then
            mes_cfg=$mes_cfg$inst_id":127.0.0.1:"$port","
        else
            mes_cfg=$mes_cfg$inst_id":127.0.0.1:"$port
        fi
    done
}

init_nodes_list_standby_cluster()
{
    inst_count=$1
    last_id=`expr $inst_count - 1`
    for i in `seq 0 $last_id`
    do
        inst_id=`expr $i + $INST_OFFSET - 2`
        port=`expr $i + $DSS_PORT_BASE + 10000`
        if [ $i != $last_id ]; then
            mes_cfg=$mes_cfg$inst_id":127.0.0.1:"$port","
        else
            mes_cfg=$mes_cfg$inst_id":127.0.0.1:"$port
        fi
    done
}

init_dss_conf()
{
    dss_home=$1
    simu_path=$3
    lock_path=$4
    cluster_mode=$5
    if [ ${cluster_mode} == 'standby_cluster' ]; then
        inst_id=`expr $2 + $INST_OFFSET - 2`
    else
        inst_id=`expr $2 + $INST_OFFSET`
    fi
    echo "init ${dss_home}"

    mkdir -p ${dss_home}/cfg
    mkdir -p ${dss_home}/log

    echo "data:${simu_path}/dss_data.dmp" > ${dss_home}/cfg/dss_vg_conf.ini
    echo "log:${simu_path}/dss_log.dmp" >> ${dss_home}/cfg/dss_vg_conf.ini
    echo "INST_ID = ${inst_id}" > ${dss_home}/cfg/dss_inst.ini
    echo "_LOG_LEVEL = 255" >> ${dss_home}/cfg/dss_inst.ini
    echo "_LOG_BACKUP_FILE_COUNT = 128" >> ${dss_home}/cfg/dss_inst.ini
    echo "_LOG_MAX_FILE_SIZE = 100M" >> ${dss_home}/cfg/dss_inst.ini
    echo "LSNR_PATH = ${dss_home}" >> ${dss_home}/cfg/dss_inst.ini
    echo "_DISK_LOCK_FILE_PATH = ${lock_path}" >> ${dss_home}/cfg/dss_inst.ini
    echo "DSS_NODES_LIST = ${mes_cfg}" >> ${dss_home}/cfg/dss_inst.ini
}

create_vg()
{
    dss_home=$1
    simu_path=$2

    export DSS_HOME=${dss_home}
    rm -rf ${simu_path}
    mkdir ${simu_path}

    echo " =========== truncate `expr ${SIMULATE_SIZE} / 1000`G =========== "
#    dd if=/dev/zero bs=1048576 count=${SIMULATE_SIZE} of=${simu_path}/dss_data.dmp
    truncate -s `expr ${SIMULATE_SIZE} / 1000`G ${simu_path}/dss_data.dmp
    chmod 777 ${simu_path}/dss_data.dmp

    echo " =========== truncate `expr ${LOG_SIZE} / 1000`G =========== "
#   dd if=/dev/zero bs=1048576 count=${LOG_SIZE} of=${simu_path}/dss_log.dmp
    truncate -s `expr ${SIMULATE_SIZE} / 1000`G ${simu_path}/dss_log.dmp
    chmod 777 ${simu_path}/dss_log.dmp

    echo "> creating volume group ${simu_path}/dss_data.dmp"
    ${GAUSSHOME}/bin/dsscmd cv -g data -v ${simu_path}/dss_data.dmp -s 2048 -D ${dss_home}

    echo "> creating volume group ${simu_path}/dss_log.dmp"
    ${GAUSSHOME}/bin/dsscmd cv -g log -v ${simu_path}/dss_log.dmp -s 2048 -D ${dss_home}
}

start_dss()
{
    dsshome_pre=$1
    inst_count=$2
    cluster_mode=$3
    if [ ${cluster_mode} == 'standby_cluster' ]; then
        inst_count=4
    fi    
    echo " =================   starting $inst_count dssserver process   ================="
    for i in `seq 0 $last_id`
    do
        dss="${dsshome_pre}${i}"
        echo "> starting ${dss}" && nohup ${GAUSSHOME}/bin/dssserver -D ${dss} &
        sleep 1
    done

    # check start node number equals input value number
    dss_pids=`ps ux | grep dssserver | grep -v grep | wc -l`
    if [ $inst_count != ${dss_pids} ]; then
        echo "dssserver start failed, or parameter error"
        exit 1;
    else
        echo " =================   $inst_count dssserver process started    ================="
    fi
}

function main() {
    inst_count=$1
    last_id=`expr $inst_count - 1`
    pre_path=$2
    if [ ! -d ${pre_path} ]; then
        mkdir -p ${pre_path}
    fi
    simu_path=$3
    cluster_mode=$4
    if [ ${cluster_mode} == 'standby_cluster' ]; then
        init_nodes_list_standby_cluster $inst_count
        echo "init & start $inst_count dss node for standby_cluster"
    else
        cluster_mode='single_cluster'
        init_nodes_list $inst_count
        echo "init & start $inst_count dss node"
    fi 
    for i in `seq 0 $last_id`
        do
            echo "init_dss_conf ${pre_path}/dss_home$i"
            init_dss_conf ${pre_path}/dss_home$i $i ${simu_path} ${pre_path} ${cluster_mode}
        done

    create_vg ${pre_path}/dss_home0 ${simu_path}
    start_dss ${pre_path}/dss_home ${inst_count} ${cluster_mode}
}

main $@