1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
package com.vaadin.tests.application;
import java.io.IOException;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import com.vaadin.server.DeploymentConfiguration;
import com.vaadin.server.RequestHandler;
import com.vaadin.server.ServiceException;
import com.vaadin.server.VaadinRequest;
import com.vaadin.server.VaadinResponse;
import com.vaadin.server.VaadinServlet;
import com.vaadin.server.VaadinServletService;
import com.vaadin.server.VaadinSession;
import com.vaadin.server.communication.HeartbeatHandler;
import com.vaadin.server.communication.UidlRequestHandler;
import com.vaadin.ui.UI;
public class CommErrorEmulatorServlet extends VaadinServlet {
private Map<UI, Integer> uidlResponseCode = Collections
.synchronizedMap(new HashMap<UI, Integer>());
private Map<UI, Integer> heartbeatResponseCode = Collections
.synchronizedMap(new HashMap<UI, Integer>());
private final CommErrorUIDLRequestHandler uidlHandler = new CommErrorUIDLRequestHandler();
private final CommErrorHeartbeatHandler heartbeatHandler = new CommErrorHeartbeatHandler();
public class CommErrorUIDLRequestHandler extends UidlRequestHandler {
@Override
public boolean synchronizedHandleRequest(VaadinSession session,
VaadinRequest request, VaadinResponse response)
throws IOException {
UI ui = session.getService().findUI(request);
if (ui != null && uidlResponseCode.containsKey(ui)) {
response.sendError(uidlResponseCode.get(ui), "Error set in UI");
return true;
}
return super.synchronizedHandleRequest(session, request, response);
}
}
public class CommErrorHeartbeatHandler extends HeartbeatHandler {
@Override
public boolean synchronizedHandleRequest(VaadinSession session,
VaadinRequest request, VaadinResponse response)
throws IOException {
UI ui = session.getService().findUI(request);
if (ui != null && heartbeatResponseCode.containsKey(ui)) {
response.sendError(heartbeatResponseCode.get(ui),
"Error set in UI");
return true;
}
return super.synchronizedHandleRequest(session, request, response);
}
}
public class CommErrorEmulatorService extends VaadinServletService {
public CommErrorEmulatorService(VaadinServlet servlet,
DeploymentConfiguration deploymentConfiguration)
throws ServiceException {
super(servlet, deploymentConfiguration);
}
@Override
protected List<RequestHandler> createRequestHandlers()
throws ServiceException {
List<RequestHandler> handlers = super.createRequestHandlers();
handlers.add(uidlHandler);
handlers.add(heartbeatHandler);
return handlers;
}
}
@Override
protected VaadinServletService createServletService(
DeploymentConfiguration deploymentConfiguration)
throws ServiceException {
CommErrorEmulatorService s = new CommErrorEmulatorService(this,
deploymentConfiguration);
s.init();
return s;
}
public void setUIDLResponseCode(final UI ui, int responseCode,
final int delay) {
uidlResponseCode.put(ui, responseCode);
System.out.println(
"Responding with " + responseCode + " to UIDL requests for "
+ ui + " for the next " + delay + "s");
new Thread(() -> {
try {
Thread.sleep(delay * 1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("Handing UIDL requests normally again");
uidlResponseCode.remove(ui);
}).start();
}
public void setHeartbeatResponseCode(final UI ui, int responseCode,
final int delay) {
heartbeatResponseCode.put(ui, responseCode);
System.out.println("Responding with " + responseCode
+ " to heartbeat requests for " + ui + " for the next " + delay
+ "s");
new Thread(() -> {
try {
Thread.sleep(delay * 1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("Handing heartbeat requests normally again");
heartbeatResponseCode.remove(ui);
}).start();
}
}
|
