|
1
23
24
25 import org.objectweb.clif.scenario.util.multithread.MTScenario;
26 import org.objectweb.clif.scenario.util.multithread.MTScenarioSession;
27 import org.objectweb.clif.storage.api.ActionEvent;
28 import org.objectweb.clif.util.ClifClassLoader;
29 import org.objectweb.clif.supervisor.api.ClifException;
30 import org.apache.commons.httpclient.HttpClient;
31 import org.apache.commons.httpclient.HttpMethod;
32 import org.apache.commons.httpclient.cookie.CookiePolicy;
33 import org.apache.commons.httpclient.methods.GetMethod;
34 import org.apache.commons.httpclient.methods.PostMethod;
35 import org.apache.log4j.BasicConfigurator;
36 import org.apache.log4j.varia.NullAppender;
37 import java.util.StringTokenizer  ;
38 import java.util.Map ;
39 import java.util.HashMap ;
40 import java.util.List ;
41 import java.util.ArrayList ;
42 import java.io.BufferedReader ;
43 import java.io.InputStreamReader ;
44 import java.io.Serializable ;
45
46
47
64 public class Webtest extends MTScenario
65 {
66 static private Map urlsCache = new HashMap ();
67
68
69 static {
71 BasicConfigurator.configure(new NullAppender());
72 }
73
74
75 static protected synchronized void clearURLs()
76 {
77 urlsCache.clear();
78 }
79
80
81 static protected String [] initURLs(String filename)
82 {
83 String [] urls;
84 if (urlsCache.containsKey(filename))
85 {
86 urls = (String [])urlsCache.get(filename);
87 }
88 else
89 {
90 synchronized(urlsCache)
91 {
92 if (urlsCache.containsKey(filename))
93 {
94 urls = (String [])urlsCache.get(filename);
95 }
96 else
97 {
98 try
99 {
100 List urlList = new ArrayList ();
101 BufferedReader br = new BufferedReader (
102 new InputStreamReader (ClifClassLoader.getClassLoader().getResourceAsStream(filename)));
103 String line = null;
104 while ((line = br.readLine()) != null)
105 {
106 line = line.trim();
107 if (! line.startsWith("#"))
108 {
109 urlList.add(line);
110 }
111 }
112 br.close();
113 urls = (String [])urlList.toArray(new String [urlList.size()]);
114 urlsCache.put(filename, urls);
115 }
116 catch (Exception ex)
117 {
118 ex.printStackTrace(System.err);
119 urls = null;
120 }
121 }
122 }
123 }
124 return urls;
125 }
126
127
128 long arg_sleep_ms = 0;
129
130
131 public void init(Serializable testId)
132 throws ClifException
133 {
134 clearURLs();
135 super.init(testId);
136 }
137
138
139
149 public MTScenarioSession newSession(int sessionId, String arg)
150 throws ClifException
151 {
152 MTScenarioSession session = null;
153
154 try
155 {
156 StringTokenizer parser = new StringTokenizer (arg);
157 arg_sleep_ms = Integer.parseInt(parser.nextToken()) * 1000;
158 session = new Session(sessionId, initURLs(parser.nextToken()));
159 }
160 catch (Exception ex)
161 {
162 throw new ClifException(
163 "Webtest requires 5 arguments:\n\t<number of concurrent threads>\n\t<test duration in seconds>\n\t<ramp-up duration in seconds>\n\t<think time in s>\n\t<file containing URL list>",
164 ex);
165 }
166 return session;
167 }
168
169
170 class Session extends HttpClient implements MTScenarioSession
171 {
172 int index;
173 int id;
174 HttpMethod[] methods;
175
176 public Session(int id, String [] urls)
177 {
178 super();
179 this.id = id;
180 index = 0;
181 getState().setCookiePolicy(CookiePolicy.COMPATIBILITY);
182 methods = new HttpMethod[urls.length];
183 for (int i=0 ; i<urls.length ; ++i)
184 {
185 if (urls[i].charAt(0) == 'p' || urls[i].charAt(0) == 'P')
186 {
187 methods[i] = new PostMethod(urls[i].substring(urls[i].indexOf("http")));
188 }
189 else
190 {
191 methods[i] = new GetMethod(urls[i]);
192 }
193 methods[i].setFollowRedirects(true);
194 methods[i].setStrictMode(false);
195 }
196 }
197
198 public ActionEvent action(ActionEvent report)
199 {
200 try
201 {
202 if (methods[index] instanceof PostMethod)
203 {
204 report.type = "HTTP POST";
205 }
206 else
207 {
208 report.type = "HTTP GET";
209 }
210 report.setDate(System.currentTimeMillis());
211 try
212 {
213 executeMethod(methods[index]);
214 report.result = methods[index].getStatusCode() + " - " + methods[index].getStatusText();
215 }
216 catch (Exception ex)
217 {
218 report.result = ex.toString();
219 report.successful = false;
220 }
221 methods[index].releaseConnection();
222 report.comment = methods[index].getURI().toString();
223 String path = methods[index].getPath();
224 methods[index].recycle();
225 methods[index].setPath(path);
226 report.duration = (int) (System.currentTimeMillis() - report.getDate());
227 report.sessionId = id;
228 if (++index == methods.length)
229 {
230 index = 0;
231 }
232 Thread.sleep(arg_sleep_ms);
233 }
234 catch (Exception ex)
235 {
236 ex.printStackTrace(System.err);
237 }
238 return report;
239 }
240 }
241 }
242
|
Popular Tags
|
Java API By Example, From Geeks To Geeks.