简易tps统计程序

import java.io.IOException;
import java.util.concurrent.atomic.AtomicInteger;

import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpException;
import org.apache.commons.httpclient.HttpStatus;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * 类Main.java的实现描述：kylin测试客户端
 * 
 * @author ke.chenk 2009-7-16 下午05:24:31
 */
public class Main {

    private static final Log log = LogFactory.getLog("client-log");

    /**
     * @param args
     */
    public static void main(String[] args) {
        int clients = Integer.valueOf(args[0]);// 并发数
        int timeWidth = Integer.valueOf(args[1]);// 时间跨度
        int totalTps = 0;
        int flag = 0;
        int totalPerTime = 0;
        final String url = args[2];
        long start = System.currentTimeMillis();
        while (System.currentTimeMillis() - start < timeWidth) {// 在指定时间跨度内，不断根据指定并发数进行请求
            final AtomicInteger atomic = new AtomicInteger();
            final AtomicInteger atomicTp = new AtomicInteger();
            final long[] times = new long[clients];
            for (int i = 0; i < clients; i++) {
                new Thread() {

                    public void run() {
                        try {
                            long tpStart = System.currentTimeMillis();
                            HttpClient httpClient = new HttpClient();
                            GetMethod method = new GetMethod(url);
                            method.setRequestHeader("Connection", "close");
                            int statusCode;
                            try {
                                statusCode = httpClient.executeMethod(method);
                                if (statusCode != HttpStatus.SC_OK) {
                                    System.err.println("Method failed: " + method.getStatusLine());
                                }
                                // System.out.println(new String(method.getResponseBody()));
                                method.releaseConnection();
                                times[atomicTp.incrementAndGet() - 1] = System.currentTimeMillis() - tpStart;
                            } catch (HttpException e) {
                                // TODO Auto-generated catch block
                                e.printStackTrace();
                            } catch (IOException e) {
                                // TODO Auto-generated catch block
                                e.printStackTrace();
                            }

                        } finally {
                            atomic.incrementAndGet();
                        }
                    }
                }.start();
            }
            // 如果还没达到指定次数，主线程等待
            while (atomic.get() < clients) {
                ;
            }

            // for (int i = 0; i < times.length; i++) {
            // System.out.println("times[" + i + "]->" + times[i]);
            // }
            // long totalTime = System.currentTimeMillis() - start;
            // System.out.println(totalTime);
            long total = 0;
            for (int i = 0; i < times.length; i++) {
                total += times[i];
            }
            long avgPerTime = total / clients;
            long thisTps = (1000 / avgPerTime) * clients;// 每秒处理tps数
            flag++;
            totalTps += thisTps;
            totalPerTime += avgPerTime;
            // System.out.println(thisTps);
        }
        long allAvgTps = totalTps / flag;// 每秒处理tps数平均值
        long allPerTime = totalPerTime / flag;// 平均响应时间
        log.info("all avg tps is:" + allAvgTps);
        log.info("all avg per time:" + allPerTime);
    }
}