#!/bin/bash
export TERM=xterm
pid=0

function display_container_header {
    echo "+====================================================="
    echo "| Container   : $HOSTNAME"
    echo "| OS          : $(</etc/redhat-release)"
    echo "| kubernetes  : $KUBERNETES_SERVICE_HOST"
    echo "| user        : $(</etc/whoami)"
    echo "+====================================================="
}

function start_service {
    if [ ! -v TEST_DURATION ]; then
        TEST_DURATION=10;
        export TEST_DURATION;
    fi
    if echo $TEST_DURATION | egrep -q '^[0-9]+$'; then
        echo "" > /dev/null &
    else
        TEST_DURATION=10;
    fi
    echo "+====================================================="
    echo "| Container $HOSTNAME is now bombing CPU for $TEST_DURATION sec"
    echo "+====================================================="
    /bin/bomb > /dev/null &
    TASK_PID=$!
    sleep $TEST_DURATION
    kill $TASK_PID
    exit 0
}

display_container_header
start_service