#!/bin/sh
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
#

# Benchmark script for comparing cluster performance.

# Default options
MESSAGES="-m 10000"
REPEAT="--repeat 10"
QUEUES="-q 6"
SENDERS="-s 3"
RECEIVERS="-r 3"
BROKERS= # Local broker
CLIENT_HOSTS= # No ssh, all clients are local
# Connection options
TCP_NODELAY=false
RECONNECT=true
HEARTBEAT=1

while getopts "m:f:n:b:q:s:r:c:h:i:txyv-" opt; do
    case $opt in
	b) BROKERS="-b $OPTARG";;
	c) CLIENT_HOSTS="-c $OPTARG";;
	h) HEARTBEAT=$OPTARG;;
	i) RECONNECT=$OPTARG;;
	m) MESSAGES="-m $OPTARG";;
	n) REPEAT="--repeat $OPTARG";;
	q) QUEUES="-q $OPTARG";;
	r) RECEIVERS="-r $OPTARG";;
	s) SENDERS="-s $OPTARG";;
	t) TCP_NODELAY=true;;
	v) OPTS="--verbose";;
	x) SAVE_RECEIVED="--save-received";;
	y) NO_DELETE="--no-delete";;
	-) break ;;
	*) echo "Unknown option"; exit 1;;
    esac
done
shift $(($OPTIND-1))

CONNECTION_OPTIONS="--connection-options {tcp-nodelay:$TCP_NODELAY,reconnect:$RECONNECT,heartbeat:$HEARTBEAT}"

BROKER=$(echo $BROKERS | sed s/,.*//)
run_test() { echo $*; shift; "$@"; echo; echo; echo; }

OPTS="$OPTS $REPEAT $BROKERS --summarize $QUEUES $SENDERS $RECEIVERS $MESSAGES $CLIENT_HOSTS $SAVE_RECEIVED $CONNECTION_OPTIONS $NO_DELETE"

run_test "Benchmark:" qpid-cpp-benchmark $OPTS "$@"