path: root/java/tools/src/main/java/org/apache/qpid/tools/JVMArgConfiguration.java

/*
 *
 * 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.
 *
 */
package org.apache.qpid.tools;

import java.text.DecimalFormat;

import javax.jms.Connection;
import javax.jms.Session;

import org.apache.qpid.client.AMQConnection;

public class JVMArgConfiguration implements TestConfiguration
{
	/*
	 * By default the connection URL is used.
	 * This allows a user to easily specify a fully fledged URL any given property.
	 * Ex. SSL parameters
	 *
	 * By providing a host & port allows a user to simply override the URL.
	 * This allows to create multiple clients in test scripts easily,
	 * without having to deal with the long URL format.
	 */
	private String url = "amqp://guest:guest@clientid/testpath?brokerlist='tcp://localhost:5672'";

	private String host = "";

	private int port = -1;

	private String address = "queue; {create : always}";

	private long timeout = 0;

	private int msg_size = 1024;

	private int random_msg_size_start_from = 1;

	private boolean cacheMessage = false;

	private boolean disableMessageID = false;

	private boolean disableTimestamp = false;

	private boolean durable = false;

	private int transaction_size = 0;

	private int ack_mode = Session.AUTO_ACKNOWLEDGE;

	private int msg_count = 10;

	private int warmup_count = 1;

	private boolean random_msg_size = false;

	private String msgType = "bytes";

	private boolean printStdDev = false;

	private int sendRate = 0;

	private boolean externalController = false;

	private boolean useUniqueDest = false; // useful when using multiple connections.

	private int ackFrequency = 100;

	private DecimalFormat df = new DecimalFormat("###.##");

	private int reportEvery = 0;

	private boolean isReportTotal = false;

	private boolean isReportHeader = true;

	private int sendEOS = 0;

	private int connectionCount = 1;

	private int rollbackFrequency = 0;

	private boolean printHeaders;

	private boolean printContent;

	private long ttl;

	private int priority;

	private String readyAddress;

	public JVMArgConfiguration()
	{

		url = System.getProperty("url",url);
		host = System.getProperty("host","");
		port = Integer.getInteger("port", -1);
		address = System.getProperty("address",address);

		timeout = Long.getLong("timeout",0);
		msg_size  = Integer.getInteger("msg-size", 0);
		cacheMessage = true; //Boolean.getBoolean("cache-msg");
		disableMessageID = Boolean.getBoolean("disable-message-id");
		disableTimestamp = Boolean.getBoolean("disable-timestamp");
		durable = Boolean.getBoolean("durable");
		transaction_size = Integer.getInteger("tx",1000);
		ack_mode = Integer.getInteger("ack-mode",Session.AUTO_ACKNOWLEDGE);
		msg_count = Integer.getInteger("msg-count",msg_count);
		warmup_count = Integer.getInteger("warmup-count",warmup_count);
		random_msg_size = Boolean.getBoolean("random-msg-size");
		msgType = System.getProperty("msg-type","bytes");
		printStdDev = Boolean.getBoolean("print-std-dev");
		sendRate = Integer.getInteger("rate",0);
		externalController = Boolean.getBoolean("ext-controller");
		useUniqueDest = Boolean.getBoolean("use-unique-dest");
		random_msg_size_start_from = Integer.getInteger("random-msg-size-start-from", 1);
		reportEvery = Integer.getInteger("report-every",0);
		isReportTotal = Boolean.getBoolean("report-total");
		isReportHeader = (System.getProperty("report-header") == null) ? true : Boolean.getBoolean("report-header");
		sendEOS = Integer.getInteger("send-eos",1);
		connectionCount = Integer.getInteger("con_count",1);
		ackFrequency = Integer.getInteger("ack-frequency",100);
		rollbackFrequency = Integer.getInteger("rollback-frequency",0);
		printHeaders = Boolean.getBoolean("print-headers");
		printContent = Boolean.getBoolean("print-content");
		ttl = Long.getLong("ttl", 0);
		priority = Integer.getInteger("priority", 0);
		readyAddress = System.getProperty("ready-address");
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getUrl()
	 */
	@Override
	public String getUrl()
	{
		return url;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getHost()
	 */
	@Override
	public String getHost()
	{
		return host;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getPort()
	 */
	@Override
	public int getPort()
	{
		return port;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getAddress()
	 */
	@Override
	public String getAddress()
	{
		return address;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getTimeout()
	 */
	@Override
	public long getTimeout()
	{
		return timeout;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getAckMode()
	 */
	@Override
	public int getAckMode()
	{
		return ack_mode;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getMsgCount()
	 */
	@Override
	public int getMsgCount()
	{
		return msg_count;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getMsgSize()
	 */
	@Override
	public int getMsgSize()
	{
		return msg_size;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getRandomMsgSizeStartFrom()
	 */
	@Override
	public int getRandomMsgSizeStartFrom()
	{
		return random_msg_size_start_from;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#isDurable()
	 */
	@Override
	public boolean isDurable()
	{
		return durable;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#isTransacted()
	 */
	@Override
	public boolean isTransacted()
	{
		return transaction_size > 0;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getTransactionSize()
	 */
	@Override
	public int getTransactionSize()
	{
		return transaction_size;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getWarmupCount()
	 */
	@Override
	public int getWarmupCount()
	{
		return warmup_count;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#isCacheMessage()
	 */
	@Override
	public boolean isCacheMessage()
	{
		return cacheMessage;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#isDisableMessageID()
	 */
	@Override
	public boolean isDisableMessageID()
	{
		return disableMessageID;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#isDisableTimestamp()
	 */
	@Override
	public boolean isDisableTimestamp()
	{
		return disableTimestamp;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#isRandomMsgSize()
	 */
	@Override
	public boolean isRandomMsgSize()
	{
		return random_msg_size;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getMessageType()
	 */
	@Override
	public String getMessageType()
	{
		return msgType;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#isPrintStdDev()
	 */
	@Override
	public boolean isPrintStdDev()
	{
		return printStdDev;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getSendRate()
	 */
	@Override
	public int getSendRate()
	{
		return sendRate;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#isExternalController()
	 */
	@Override
	public boolean isExternalController()
	{
		return externalController;
	}

	public void setAddress(String addr)
	{
		address = addr;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#isUseUniqueDests()
	 */
	@Override
	public boolean isUseUniqueDests()
	{
		return useUniqueDest;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getAckFrequency()
	 */
	@Override
	public int getAckFrequency()
	{
		return ackFrequency;
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#createConnection()
	 */
	@Override
	public Connection createConnection() throws Exception
	{
		if (getHost().equals("") || getPort() == -1)
		{
			return new AMQConnection(getUrl());
		}
		else
		{
			return new AMQConnection(getHost(),getPort(),"guest","guest","test","test");
		}
	}

	/* (non-Javadoc)
	 * @see org.apache.qpid.tools.TestConfiguration#getDecimalFormat()
	 */
	@Override
	public DecimalFormat getDecimalFormat()
	{
		return df;
	}

	@Override
	public int reportEvery()
	{
		return reportEvery;
	}

	@Override
	public boolean isReportTotal()
	{
		return isReportTotal;
	}

	@Override
	public boolean isReportHeader()
	{
		return isReportHeader;
	}

	@Override
	public int getSendEOS()
	{
		return sendEOS;
	}

	@Override
	public int getConnectionCount()
	{
		return connectionCount;
	}

	@Override
	public int getRollbackFrequency()
	{
		return rollbackFrequency;
	}

	@Override
	public boolean isPrintHeaders()
	{
		return printHeaders;
	}

	@Override
	public boolean isPrintContent()
	{
		return printContent;
	}

	@Override
	public long getTTL()
	{
		return ttl;
	}

	@Override
	public int getPriority()
	{
		return priority;
	}

	@Override
	public String getReadyAddress()
	{
		return readyAddress;
	}
}