|
| 1 | +#!/bin/sh |
| 2 | +# |
| 3 | +# Main entrypoint for our Docker image - runs Gru, Minions or other commands |
| 4 | + |
| 5 | +# any .jmx file passed in the command line we act as 'Gru' |
| 6 | +if [ ${1##*.} = 'jmx' ]; then |
| 7 | + |
| 8 | + if [ "$MINION_HOSTS" = '' ]; then |
| 9 | + echo "MINION_HOSTS must be specified - a command separated list of hostnames or IP addresses" |
| 10 | + exit 1 |
| 11 | + fi |
| 12 | + echo "Connecting to $MINION_HOSTS" |
| 13 | + |
| 14 | + # AWS Public HOSTNAME API |
| 15 | + echo "Detecting an AWS Environment" |
| 16 | + PUBLIC_HOSTNAME=$(curl -s --max-time 5 http://169.254.169.254/latest/meta-data/public-hostname) |
| 17 | + |
| 18 | + if [ "$PUBLIC_HOSTNAME" = '' ]; then |
| 19 | + echo "Not running in AWS. Using Gru HOSTNAME $HOSTNAME" |
| 20 | + else |
| 21 | + HOSTNAME=$PUBLIC_HOSTNAME |
| 22 | + echo "Using Gru AWS Public HOSTNAME $HOSTNAME" |
| 23 | + fi |
| 24 | + # empty the logs directory, or jmeter may fail |
| 25 | + rm -rf /logs/report /logs/*.log |
| 26 | + # run jmeter in client (gru) mode |
| 27 | + exec jmeter -n $JMETER_FLAGS \ |
| 28 | + -R $MINION_HOSTS \ |
| 29 | + -Dclient.rmi.localport=51000 \ |
| 30 | + -Djava.rmi.server.hostname=${PUBLIC_HOSTNAME} \ |
| 31 | + -l $RESULTS_LOG \ |
| 32 | + -t $1 \ |
| 33 | + -e -o /logs/report |
| 34 | + |
| 35 | +fi |
| 36 | + |
| 37 | +# act as a 'Minion' |
| 38 | +if [ "$1" = 'minion' ]; then |
| 39 | + |
| 40 | + # AWS Public HOSTNAME API |
| 41 | + echo "Detecting an AWS Environment" |
| 42 | + PUBLIC_HOSTNAME=$(curl -s --max-time 5 http://169.254.169.254/latest/meta-data/public-hostname) |
| 43 | + |
| 44 | + if [ "$PUBLIC_HOSTNAME" = '' ]; then |
| 45 | + echo "Not running in AWS. Using Minion HOSTNAME $HOSTNAME" |
| 46 | + else |
| 47 | + HOSTNAME=$PUBLIC_HOSTNAME |
| 48 | + echo "Using Minion AWS Public HOSTNAME $HOSTNAME" |
| 49 | + fi |
| 50 | + # run jmeter in server (minion) mode |
| 51 | + exec jmeter-server -n \ |
| 52 | + -Dserver.rmi.localport=50000 \ |
| 53 | + -Djava.rmi.server.hostname=${HOSTNAME} |
| 54 | + |
| 55 | +fi |
| 56 | + |
| 57 | +exec "$@" |
0 commit comments