#!/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"
FLOW="--flow-control 100"	      # Flow control limit on queue depth for latency.
REPEAT="--repeat 10"
QUEUES="-q 6"
SENDERS="-s 3"
RECEIVERS="-r 3"
BROKERS=			# Local broker
CLIENT_HOSTS=			# No ssh, all clients are local

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

REPLICATE="node:{x-declare:{arguments:{'qpid.replicate':all}}}"
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 $TCP_NODELAY $NO_DELETE"
OPTS="$OPTS --create-option $REPLICATE"
run_test "Benchmark:" qpid-cpp-benchmark $OPTS "$@"
