com.fusesource.forge.jmstest.executor.AbstractBenchmarkExecutor.java Source code

Introduction

Here is the source code for com.fusesource.forge.jmstest.executor.AbstractBenchmarkExecutor.java

Source

/*
 * Copyright (C) 2009, Progress Software Corporation and/or its
 * subsidiaries or affiliates.  All rights reserved.
 *
 * Licensed 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 com.fusesource.forge.jmstest.executor;

import java.lang.reflect.Method;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.context.ApplicationContext;

import com.fusesource.forge.jmstest.benchmark.command.BenchmarkCommand;
import com.fusesource.forge.jmstest.benchmark.command.transport.CommandTransport;
import com.fusesource.forge.jmstest.config.SpringConfigHelper;
import com.fusesource.forge.jmstest.frontend.BenchmarkConnector;

public abstract class AbstractBenchmarkExecutor {

    private BenchmarkConnector benchmarkConnector = new BenchmarkConnector();
    private Log log = null;
    private SpringConfigHelper springConfigHelper = new SpringConfigHelper();

    protected BenchmarkConnector getConnector() {
        return benchmarkConnector;
    }

    public CommandTransport getCmdTransport() {
        return benchmarkConnector.getCmdTransport();
    }

    public final void sendCommand(BenchmarkCommand command) {
        getCmdTransport().sendCommand(command);
    }

    void setJmsPort(int jmsPort) {
        getConnector().setPort(jmsPort);
    }

    int getJmsPort() {
        return getConnector().getPort();
    }

    public void setHostname(String hostname) {
        getConnector().setHostName(hostname);
    }

    public String getHostname() {
        return getConnector().getHostname();
    }

    public String getDestinationName() {
        return getConnector().getDestinationName();
    }

    public void setDestinationName(String destinationName) {
        getConnector().setDestinationName(destinationName);
    }

    public void setSpringConfigLocations(String springConfigDirs) {
        log().debug("Setting a location for the config files");
        this.springConfigHelper.setSpringConfigLocations(springConfigDirs);
    }

    protected ApplicationContext getApplicationContext() {
        return springConfigHelper.getApplicationContext();
    }

    protected ApplicationContext getApplicationContext(String springConfigDirs) {
        SpringConfigHelper sch = new SpringConfigHelper();
        sch.setSpringConfigLocations(springConfigDirs);
        return sch.getApplicationContext();
    }

    protected Method getSetterByName(String propName) {

        log().debug("Trying to find setter method for: " + propName);
        String methodName = "set" + propName.substring(0, 1).toUpperCase() + propName.substring(1);

        for (Method m : this.getClass().getMethods()) {
            if (m.getName().equals(methodName)) {
                if (m.getParameterTypes().length == 1) {
                    return m;
                }
            }
        }

        return null;
    }

    protected void handleArguments(String[] args) {
        for (int i = 0; i < args.length; i++) {
            if (args[i].startsWith("-")) {
                String propName = args[i].substring(1);
                String value = null;
                if (args.length > (i + 1) && !(args[i + 1].startsWith("-"))) {
                    value = args[++i];
                } else {
                    value = Boolean.TRUE.toString();
                }
                try {
                    Method m = getSetterByName(propName);
                    if (m != null) {
                        log().debug("Setting property: " + propName + "=" + value);
                        m.invoke(this, value);
                    }
                } catch (Exception e) {
                    log().warn("Property " + propName + " not set.", e);
                }
            } else {
                log().warn("Ignoring command line argument: " + args[i]);
            }
        }
    }

    private Log log() {
        if (log == null) {
            log = LogFactory.getLog(this.getClass());
        }
        return log;
    }
}