apiVersion: gatling-operator.tech.zozo.com/v1alpha1
kind: Gatling
metadata:
  name: gatling-sample01
spec:
  generateReport: true
  generateLocalReport: false
  notifyReport: false
  cleanupAfterJobDone: false

  initContainers:
  - args:
    - |
      PARALLELISM=1
      NAMESPACE=default
      JOB_NAME=gatling-sample02
      POD_NAME=$(cat /etc/pod-info/name)

      kubectl label pods -n $NAMESPACE $POD_NAME gatling-waiter=initialized

      while true; do
        READY_PODS=$(kubectl get pods -n $NAMESPACE --selector=job-name=$JOB_NAME-runner,gatling-waiter=initialized --no-headers | grep -c ".*");
        echo "$READY_PODS/$PARALLELISM pods are ready";
        if  [ $READY_PODS -eq $PARALLELISM ]; then
          break;
        fi;
        sleep 1;
      done
    command:
    - /bin/sh
    - -c
    image: bitnami/kubectl:1.21.8
    ...omit...

 containers:
  - args:  ############## gatling-runner
    - |
      SIMULATIONS_DIR_PATH=/opt/gatling/user-files/simulations
      TEMP_SIMULATIONS_DIR_PATH=/opt/gatling/user-files/simulations-temp
      RESOURCES_DIR_PATH=/opt/gatling/user-files/resources
      RESULTS_DIR_PATH=/opt/gatling/results
      START_TIME=""
      if [ -z "${START_TIME}" ]; then
        START_TIME=$(date +"%Y-%m-%d %H:%M:%S" --utc)
      fi
      start_time_stamp=$(date -d "${START_TIME}" +"%s")
      current_time_stamp=$(date +"%s")
      echo "Wait until ${START_TIME}"
      until [ ${current_time_stamp} -ge ${start_time_stamp} ];
      do
        current_time_stamp=$(date +"%s")
        echo "it's ${current_time_stamp} now and waiting until ${start_time_stamp} ..."
        sleep 1;
      done
      if [ ! -d ${SIMULATIONS_DIR_PATH} ]; then
        mkdir -p ${SIMULATIONS_DIR_PATH}
      fi
      if [ -d ${TEMP_SIMULATIONS_DIR_PATH} ]; then
        cp -p ${TEMP_SIMULATIONS_DIR_PATH}/*.scala ${SIMULATIONS_DIR_PATH}
      fi
      if [ ! -d ${RESOURCES_DIR_PATH} ]; then
        mkdir -p ${RESOURCES_DIR_PATH}
      fi
      if [ ! -d ${RESULTS_DIR_PATH} ]; then
        mkdir -p ${RESULTS_DIR_PATH}
      fi
      gatling.sh -sf ${SIMULATIONS_DIR_PATH} -s MyBasicSimulation -rsf ${RESOURCES_DIR_PATH} -rf ${RESULTS_DIR_PATH} -nr

      if [ $? -eq 0 ]; then
        touch ${RESULTS_DIR_PATH}/COMPLETED
      fi
    command:
    - /bin/sh
    - -c
    image: ghcr.io/st-tech/gatling:latest
    ...omit...
  - args: ############## gatling-result-transferer
    - |
      RESULTS_DIR_PATH=/opt/gatling/results
      rclone config create s3 s3 env_auth=true region ap-northeast-1
      while true; do
        if [ -f "${RESULTS_DIR_PATH}/COMPLETED" ]; then
          for source in $(find ${RESULTS_DIR_PATH} -type f -name *.log)
          do
            rclone copyto ${source} --s3-no-check-bucket --s3-env-auth s3:my-gatling-reports-0001/gatling-sample02/2391946291/${HOSTNAME}.log
          done
          break
        fi
        sleep 1;
      done
    command:
    - /bin/sh
    - -c
    image: rclone/rclone
    ...omit...