entrypoint.sh

#!/bin/bash
set -e

NOW=$(date +"%Y%m%d-%H%M%S")
JMETER_HOME="/opt/jmeter/apache-jmeter"
java -version

if [ -z "${JMX_FOLDER}" ]; then
   echo "Give at least the jmx parent folder name as parameter with"
   echo "JMX_FOLDER=my-scenario docker-compose -p jmeter up -d" 
   exit 1
fi

source "/scenario/${JMX_FOLDER}/.env"

# Environment variable available :

if [ -z "${host}" ]; then
    host=google.com
fi

if [ -z "${protocol}" ]; then
    protocol=https
fi

if [ -z "${port}" ]; then
    port=443
fi

if [ -z "${XMX}" ]; then
    XMX="1g"
fi

if [ -z "${XMS}" ]; then
    XMS="1g"
fi

# Copying JMX, modules files to bin folder
cp /scenario/modules/*.jmx "${JMETER_HOME}/bin"
cp "/scenario/${JMX_FOLDER}/${JMX_FOLDER}.jmx" "${JMETER_HOME}/bin"

# Setting report and log dir
LOGS_DIR="${JMETER_HOME}/logs/${JMX_FOLDER}"
RESULTS_DIR="${JMETER_HOME}/results/${JMX_FOLDER}"
RESULTS_FILE="${RESULTS_DIR}/${NOW}-load-test-${JMX_FOLDER}-result.csv"

# Preparing JMeter vars
JMX_FILE_PATH="${JMETER_HOME}/bin/${JMX_FOLDER}.jmx"
PARAM_HOSTS_ARGS="-Ghost=${host} -Gport=${port} -Gprotocol=${protocol}"
PARAM_USERS_ARGS="-Gthreads=${threads} -Gduration=${duration} -Grampup=${rampup} -Gjmx=${JMX_FOLDER}.jmx"
echo "server.rmi.ssl.disable=true" >> ${JMETER_HOME}/bin/jmeter.properties

# JVM args
JVM_ARGS="$JVM_ARGS -Duser.timezone=CET"
JVM_ARGS="$JVM_ARGS -Djava.net.preferIPv4Stack=true -Djava.net.preferIPv6Addresses=false"
JVM_ARGS="$JVM_ARGS -Dcom.sun.management.jmxremote.authenticate=false"
JVM_ARGS="$JVM_ARGS -Dcom.sun.management.jmxremote.ssl=false"
JVM_ARGS="$JVM_ARGS -XX:+UseG1GC -XX:MaxGCPauseMillis=100 -Xms${XMX} -Xmx${XMS} -XX:G1ReservePercent=20 -Xss256k"
export JVM_ARGS


# Runtime
CSV=$(find "/scenario/data" -type f -name "*.csv")

if [[ -z "${SLAVE}" ]]; then
    # Building host list
        START=1
        END=${nbInjector}
        i=${START}
        HOST_LIST=()

        while [[ ${i} -le ${END} ]]; do
            # Warning, the below line could change depending the docker-compose version (could be _ separated instead of -)
            HOST_LIST+=("jmeter-jmeter-slave-${i}")
            i=$((i + 1))
        done

        echo "Injector hostname list : ${HOST_LIST[@]}"

        # Building IP list of slaves
        for HOST in "${HOST_LIST[@]}"; do
            HOST_IP_LIST+=( "$(getent hosts "${HOST}" | awk -F" " '{print $1}')" )
        done

        echo "Injectors IP list : ${HOST_IP_LIST[@]}"
fi

if [[ -n "${CSV}" ]]; then
    if [[ "${SLAVE}" -eq 1 ]]; then

        IP=$(hostname -i)
        echo "Slave at ${IP} is starting"

        while ! ls /scenario/data/split/*${IP};
        do
            echo "Waiting for dataset to be splitted by controller "
            sleep 2
        done

        for DATASET_FILE_PATH in $(ls /scenario/data/split/*"${IP}"); do
            DATASET_FILE=$(basename "${DATASET_FILE_PATH}")
            echo "copying ${DATASET_FILE_PATH} to ${JMETER_HOME}/${DATASET_FILE}/.${IP}"
            cp "${DATASET_FILE_PATH}" "${JMETER_HOME}/bin/${DATASET_FILE/.${IP}/}"
        done

        ls -ltra ${JMETER_HOME}/bin/*.csv

    else

        echo "Found csv dataset to split: ${CSV}"
        echo "Controller dataset management starting"

        # Dataset splitting
        mkdir -p "/scenario/data/split"
        
        START=0
        END=$((nbInjector -1))
        i=${START}

        # Splitting dataset to equal parts
        for DATASET_FILE_PATH in $(ls /scenario/data/*.*); do
            echo "Splitting ${DATASET_FILE_PATH}"
            DATASET_FILE=$(basename "${DATASET_FILE_PATH}")
            TOTAL_LINE=$(wc -l < "${DATASET_FILE_PATH}")
            LINES_PER_FILES=$(((TOTAL_LINE + nbInjector - 1) / nbInjector))
            split -d -a 1 -l ${LINES_PER_FILES} "${DATASET_FILE_PATH}" "/scenario/data/split/splitted_${DATASET_FILE}"

            echo "Splitting folder content"

            # Appending slave IP to dataset file 
            while [[ "${i}" -le ${END} ]]; do
                echo "Generating dataset for ${HOST_IP_LIST[${i}]}"
                mv "/scenario/data/split/splitted_${DATASET_FILE}${i}" "/scenario/data/split/${DATASET_FILE}.${HOST_IP_LIST[${i}]}"
                i=$((i + 1))
            done
        done

    fi
    else 
    echo "No dataset found, starting JMeter..."
fi

if [[ "${SLAVE}" -eq 1 ]]; then

    echo "Starting JMeter on slave ${IP}"

    LOG_FILE="${LOGS_DIR}/jmeter-${IP}-${JMX_FOLDER}.jmx-${NOW}.log"

    echo "Installing plugins for JMX ${JMX_FOLDER}.jmx"
    ${JMETER_HOME}/bin/PluginsManagerCMD.sh install-for-jmx "${JMX_FILE_PATH}"

    set -x

    ${JMETER_HOME}/bin/jmeter \
    -LINFO \
    -j "${LOG_FILE}" \
    -s \
    -n \
    -d "${JMETER_HOME}" \
    -Gserver.exitaftertest=true \
    ${PARAM_HOSTS_ARGS} \
    ${PARAM_USERS_ARGS}

else 

    echo "Starting JMeter on controller"

    echo "Waiting for injectors to start"
    slave_array=(${HOST_IP_LIST[@]})
    slave_num=${#slave_array[@]}
    index=${slave_num} 
    
    while [ ${index} -gt 0 ]; do 
        for slave in ${slave_array[@]}; do 
            if echo 'test open port' 2>/dev/null > /dev/tcp/${slave}/1099
            then    
                echo "${slave} ready"
                slave_array=(${slave_array[@]/${slave}/})
                index=$((index-1))
            else 
                echo "${slave} not ready"
            fi; 
        done; 
        echo 'Waiting for slave readiness'
        sleep 2
    done

    printf -v SLAVE_IP_LIST '%s,' "${HOST_IP_LIST[@]}"
    LOG_FILE="${LOGS_DIR}/jmeter-master-${JMX_FOLDER}.jmx-${NOW}.log"
    mkdir -p "${RESULTS_DIR}"

    echo "Slaves IP :"
    echo "${SLAVE_IP_LIST::-1}"

    set -x
    
    ${JMETER_HOME}/bin/jmeter \
    -LINFO \
    -X \
    -d ${JMETER_HOME} \
    -n -j ${LOG_FILE} \
    -l "${RESULTS_FILE}" \
    -R ${SLAVE_IP_LIST::-1} \
    ${PARAM_HOSTS_ARGS} \
    ${PARAM_USERS_ARGS} \
    -t "${JMX_FILE_PATH}" \
    -e \
    -o "${RESULTS_DIR}/report-${JMX_FOLDER}-${NOW}"

    trap "sh ${JMETER_HOME}/bin/stoptest.sh" EXIT
fi