使用HttpClient对web应用进行测试

在几天程序突然报出了数据库连接被管理员销毁的情况! 一时之间也毫无头绪, 甚至怀疑我们的程序能否通过压力测试, 所以就使用了HttpClient整了一段测试跑了来测试一下, 最后虽然找到了问题是因为ArcGis占有了很多连接没有释放导致的, 呵呵， 既然写了, 还是拿出来试试:
[code]
package com.gomt.httpclient;

import java.awt.BorderLayout;
import java.awt.FlowLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;

import javax.swing.JButton;
import javax.swing.JComboBox;
import javax.swing.JEditorPane;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JSplitPane;
import javax.swing.JTextArea;
import javax.swing.JTextField;
import javax.swing.SwingUtilities;
import javax.swing.text.BadLocationException;
import javax.swing.text.html.HTMLDocument;

import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.MultiThreadedHttpConnectionManager;
import org.apache.commons.httpclient.NameValuePair;
import org.apache.commons.httpclient.methods.PostMethod;


public class AutoTest {

static final String LOGON_SITE = "192.168.0.93";
static final int LOGON_PORT = 7001;

public AutoTest() {

}

/**
* @param args
*/
public static void main(String[] args) {
HttpClientFrame f = new HttpClientFrame();
f.setTitle("HttpClient测试");
f.setSize(700, 500);
f.addWindowListener(
new WindowAdapter() {
public void windowClosing(WindowEvent e) {
System.exit(0);
}
}
);
f.setVisible(true);
}

public static class HttpClientFrame extends JFrame {
private static final long serialVersionUID = 3348783285573305436L;
private JComboBox cmbURL;
private JComboBox runcount;
private JTextField cmbPort;
private JTextArea taTextResponse;
private JEditorPane htmlPane;

private HttpClient client;
String[] userArr = new String[]{"admin", "tianyy", "jiangzy", "caoxd", "changam", "chejh", "chench", "chengh", "chenhr", "guzm", "hesb", "hep", "laijw"};
String[] passArr = new String[]{"admin", "tian", "1", "1", "1", "1", "1", "1", "1", "1", "1", "1", "1"};
long begin = 0;

public HttpClientFrame() {

begin = System.currentTimeMillis();
System.out.println("start : " + begin);
//HttpClient
client = new HttpClient(new MultiThreadedHttpConnectionManager());
//client.getHostConfiguration().setHost(LOGON_SITE, LOGON_PORT, "http");
client.getHttpConnectionManager().getParams().setConnectionTimeout(30000);

JPanel panInput = new JPanel(new FlowLayout());

String[] aURLs = {
"192.168.0.9",
"192.168.0.10",
"192.168.0.93"//,
//"http://www.anybrowser.org/",
//"http://jakarta.apache.org/",
//"http://www.w3.org/"
};

final JButton btnGET = new JButton("测试");
btnGET.addActionListener(
new ActionListener() {
public void actionPerformed(ActionEvent ae) {
String url = (String) cmbURL.getSelectedItem();
if (url != null && url.length() > 0) {
//loadPage(url);
int runtimes = 1;
int port = 80;
try {
runtimes = Integer.parseInt((String)runcount.getSelectedItem());


} catch(Exception eee) {
runtimes = 1;
}
try {
port = Integer.parseInt(cmbPort.getText());
} catch(Exception ee) {
port = 80;
}

java.util.Random rm = new java.util.Random();
for (int i = 0; i < runtimes; i++) {
int rmi = rm.nextInt();
int arryp = Math.abs(rmi % 13);
System.out.println("arryp = " + arryp);
loadPage(url, i+1, begin, userArr[arryp], passArr[arryp], port);
// PostMethod post = new PostMethod("http://" + url + ":" + LOGON_PORT + "/szsf/loginaction.action");
// NameValuePair username = new NameValuePair("user", userArr[arryp]);
// NameValuePair password = new NameValuePair("pwd", passArr[arryp]);
// NameValuePair method = new NameValuePair("method", "b");
// post.setRequestBody(new NameValuePair[]{username, password, method});
//get.setFollowRedirects(true);

}
}
}
}
);

cmbURL = new JComboBox(aURLs);
cmbURL.setToolTipText("输入一个地址！");
cmbURL.setEditable(true);
cmbURL.setSelectedIndex(0);

cmbPort = new JTextField();
cmbPort.setToolTipText("请输入端口号！");
cmbPort.setEditable(true);
cmbPort.setText("7001");

JLabel lblURL = new JLabel("地址:");

runcount = new JComboBox(new String[]{"1","10","100","1000"});
runcount.setToolTipText("运行次数");
runcount.setEditable(true);
runcount.setSelectedIndex(0);

panInput.add(lblURL);
panInput.add(cmbURL);
panInput.add(cmbPort);
panInput.add(btnGET);
panInput.add(runcount);
panInput.add(new JLabel("次"));

taTextResponse = new JTextArea();
taTextResponse.setEditable(false);
taTextResponse.setCaretPosition(0);

htmlPane = new JEditorPane();
htmlPane.setContentType("text/html;charset=UTF-8");
htmlPane.setEditable(false);

JSplitPane splitResponsePane = new JSplitPane(
JSplitPane.HORIZONTAL_SPLIT,
new JScrollPane(taTextResponse),
new JScrollPane(htmlPane)
);
splitResponsePane.setOneTouchExpandable(false);
splitResponsePane.setDividerLocation(350);
// it would be better to set resizeWeight, but this method does
// not exist in JRE 1.2.2
// splitResponsePane.setResizeWeight(0.5);

this.getContentPane().setLayout(new BorderLayout());
this.getContentPane().add(panInput, BorderLayout.NORTH);
this.getContentPane().add(splitResponsePane, BorderLayout.CENTER);
}

/**
* Sets the HTML content to be displayed.
*
* @param content an HTML document
* @throws UnsupportedEncodingException
*/
private void setDocumentContent(String content, InputStream in) throws UnsupportedEncodingException {

HTMLDocument doc = new HTMLDocument();

// try {
// doc.remove(0, doc.getLength());
// } catch (BadLocationException e) {
// e.printStackTrace();
// }
doc.putProperty("IgnoreCharsetDirective", Boolean.TRUE);

taTextResponse.setText(new String(content.getBytes("GBK"), "UTF-8"));
taTextResponse.setCaretPosition(0);
try {
htmlPane.read(new java.io.ByteArrayInputStream(taTextResponse.getText().getBytes()), doc);
} catch (IOException e) {
e.printStackTrace();
}
if(doc.getLength() != 0) {
htmlPane.setDocument(doc);
htmlPane.setCaretPosition(0);
}
taTextResponse.requestFocus();

}

/**
* Loads the page at the given URL from a separate thread.
* @param url
*/
private void loadPage(final String url, final int id, final long begin,
final String usernames, final String passwords, final int port) {
// create a new thread to load the URL from
new Thread() {
public void run() {
PostMethod post = new PostMethod("http://" + url + ":" + port + "/szsf/loginaction.action");
NameValuePair username = new NameValuePair("user", usernames);
NameValuePair password = new NameValuePair("pwd", passwords);
NameValuePair method = new NameValuePair("method", "b");
post.setRequestBody(new NameValuePair[]{username, password, method});

try {
int iGetResultCode = client.executeMethod(post);
final InputStream in = post.getResponseBodyAsStream();
java.io.ByteArrayOutputStream byteout = new java.io.ByteArrayOutputStream(5120);
byte[] b = new byte[5120];
int len = 0;
while((len =in.read(b)) > -1) {
byteout.write(b, 0, len);
}

final String strGetResponseBody = new String(byteout.toByteArray());//post.getResponseBodyAsString();
System.out.println("返回状态为：" + iGetResultCode);

if (strGetResponseBody != null) {
// set the HTML on the UI thread
SwingUtilities.invokeLater(
new Runnable() {
public void run() {
try {
setDocumentContent(strGetResponseBody, in);
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
}
}
}
);
}
} catch (Exception ex) {
ex.printStackTrace();
} finally {
post.releaseConnection();
}
}
}.start();
}

}

/**
* A thread that performs a GET.
*/
// static class PostThread extends Thread {
//
// private HttpClient httpClient;
// private PostMethod method;
// private int id;
// private long begin;
//
// public PostThread(HttpClient httpClient, PostMethod method, int id, long begin) {
// this.httpClient = httpClient;
// this.method = method;
// this.id = id;
// this.begin = begin;
//
// }
//
// /**
// * Executes the GetMethod and prints some satus information.
// */
// public void run() {
//
// try {
//
// System.out.println(id + " - about to get something from " + method.getURI());
// // execute the method
// httpClient.executeMethod(method);
//
// //System.out.println(id + " - get executed");
// // get the response body as an array of bytes
// ///byte[] bytes = method.getResponseBody();
//
// //System.out.println(id + " - " + bytes.length + " bytes read");
// final String resString = method.getResponseBodyAsString();
// if (resString != null) {
// // set the HTML on the UI thread
// SwingUtilities.invokeLater(
// new Runnable() {
// public void run() {
// setDocumentContent(resString);
// }
// }
// );
// }
// //System.out.println("返回字串为: " + resString);
//
// } catch (Exception e) {
// System.out.println(id + " - error: " + e);
// } finally {
// // always release the connection after we're done
// method.releaseConnection();
// //System.out.println(id + " - connection released");
// System.out.println(id + " begin: " + begin + " end : "+ System.currentTimeMillis() + " cost : " + (System.currentTimeMillis() - begin)/1000.0);
// }
// }
//
// }

}

[/code]