org.springframework.benchmark.cmt.web.AbstractBenchmarkController.java Source code

Introduction

Here is the source code for org.springframework.benchmark.cmt.web.AbstractBenchmarkController.java

Source

/*
 * The Spring Framework is published under the terms
 * of the Apache Software License.
 */

package org.springframework.benchmark.cmt.web;

import java.io.IOException;
import java.util.HashMap;
import java.util.Random;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.BeansException;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.AbstractController;

import org.springframework.benchmark.cmt.client.BenchmarkFactory;
import org.springframework.benchmark.cmt.server.Benchmark;

/**
 * 
 * @author Rod Johnson
 */
public abstract class AbstractBenchmarkController extends AbstractController {

    Log log = LogFactory.getLog(getClass());

    // Could allow override
    public final static int USERS = 99; // indexed from 1 to 100

    public final static int ITEMS = 999;

    private Random rand = new Random();

    /** Map from name to benchmarkfactory */
    private HashMap benchmarkFactories = new HashMap();

    private Config config;

    private String[] benchmarkNames;

    public void setConfig(Config config) {
        this.config = config;
    }

    /** 
     * Convenience method for subclasses
     * @param sz size of array to index
     * @return a random array of list index from 0 up to sz-1
     */
    protected int randomIndex(int sz) {
        return Math.abs(rand.nextInt(sz));
    }

    private Benchmark getBenchmark(String name) throws Exception {
        BenchmarkFactory bf = (BenchmarkFactory) benchmarkFactories.get(name);
        if (bf == null)
            throw new ServletException("No benchmark factory with name '" + name + "': " + "valid values are "
                    + StringUtils.arrayToCommaDelimitedString(this.benchmarkNames));
        return bf.getBenchmark();
    }

    /**
     * @see org.springframework.web.servlet.mvc.Controller#handleRequest(javax.servlet.http.HttpServletRequest,
     *      javax.servlet.http.HttpServletResponse)
     */
    protected final ModelAndView handleRequestInternal(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {

        //String name = request.getParameter("bm");
        //if (name == null)
        //   throw new ServletException("'bm' parameter may not be null: " +
        //         "valid values are " + StringUtils.arrayToCommaDelimitedString(this.benchmarkNames));

        String name = config.getMode();

        // Could add timing to model
        Benchmark benchmark;
        try {
            long st = System.currentTimeMillis();
            benchmark = getBenchmark(name);
            ModelAndView mv = run(benchmark);
            long et = System.currentTimeMillis() - st;
            mv.getModel().put("time", "" + et);
            return mv;
        } catch (Exception ex) {
            throw new ServletException(ex);
        }

    }

    protected abstract ModelAndView run(Benchmark benchmark) throws Exception;

    /**
     * @see org.springframework.context.support.ApplicationObjectSupport#initApplicationContext()
     */
    protected void initApplicationContext() throws BeansException {
        super.initApplicationContext();

        String[] names = getApplicationContext().getBeanNamesForType(BenchmarkFactory.class);
        for (int i = 0; i < names.length; i++) {
            BenchmarkFactory bmf = (BenchmarkFactory) getApplicationContext().getBean(names[i]);
            this.benchmarkFactories.put(names[i], bmf);
            logger.info("Mapping from " + names[i] + " to " + bmf);
        }
        this.benchmarkNames = names;
    }
}