public class GUIInterna {
private static Boolean allowed = Settings.isAllowLoadtestMode();
private static ThreadLocalData tld = new ThreadLocalData();
private static Map<String, Map<Object, Object>> loadtestClients =
new HashMap<String, Map<Object, Object>>();
public static void begin(HttpServletRequest req) {
if (!allowed) return;
if (req == null) return;
String key = getCookie(req);
if (key == null) return;
String q = req.getQueryString();
synchronized (loadtestClients) {
if (q != null) {
if (q.endsWith("noloadtest")) {
if (loadtestClients.containsKey(key)) {
loadtestClients.remove(key);
}
} else if (q.endsWith("loadtest")) {
if (loadtestClients.containsKey(key)) {
loadtestClients.remove(key);
}
loadtestClients.put(key, new HashMap<Object, Object>());
} else if (q.equals("clearloadtests")) {
loadtestClients.clear();
}
}
if (loadtestClients.containsKey(key)) {
tld.setBoolean(Boolean.TRUE);
tld.setMap(loadtestClients.get(key));
} else {
tld.setBoolean(Boolean.FALSE);
}
}
}
public static boolean isLoadPerformanceMode() {
return allowed && tld.getBoolean();
}
public static void end(HttpServletRequest req) {
if (!allowed) return;
if (req == null) return;
if (tld.getBoolean()) {
String c = getCookie(req);
if (c != null) {
synchronized (loadtestClients) {
loadtestClients.put(c, tld.getMap());
}
}
}
tld.remove();
}
private static String getCookie(HttpServletRequest req) {
Cookie c[] = req.getCookies();
for (int i = 0; c != null && i < c.length; i++) {
if (c[i].getName().equals("JSESSIONID")) {
return c[i].getValue();
}
}
return null;
}
public static Map<Object, Object> getReplayModeData() {
return tld.getMap();
}
private static class ValueContainer {
Boolean b = null;
Map<Object, Object> m = null;
ValueContainer() {
// nothing to be done here
}
}
private static class ThreadLocalData extends ThreadLocal<ValueContainer> {
Boolean getBoolean() {
ValueContainer vc = get();
if (vc != null && vc.b != null) {
return vc.b;
}
return Boolean.FALSE;
}
Map<Object, Object> getMap() {
ValueContainer vc = get();
if (vc != null && vc.m != null) {
return vc.m;
}
return null;
}
void setBoolean(Boolean b) {
ValueContainer vc = get();
if (vc == null) {
vc = new ValueContainer();
set(vc);
}
vc.b = b;
}
void setMap(Map<Object, Object> m) {
ValueContainer vc = get();
if (vc == null) {
vc = new ValueContainer();
set(vc);
}
vc.m = m;
}
}
}
