blob: badd7ee2274512ddb2661d22da6a11edbc498747 (
plain)
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
|
package fi.iki.elonen.samples.echo;
import fi.iki.elonen.NanoHTTPD;
import fi.iki.elonen.WebSocket;
import fi.iki.elonen.WebSocketFrame;
import java.io.IOException;
/**
* @author Paul S. Hawke (paul.hawke@gmail.com)
* On: 4/23/14 at 10:34 PM
*/
class DebugWebSocket extends WebSocket {
private final boolean debug;
public DebugWebSocket(NanoHTTPD.IHTTPSession handshake, boolean debug) {
super(handshake);
this.debug = debug;
}
@Override
protected void onPong(WebSocketFrame pongFrame) {
if (debug) {
System.out.println("P " + pongFrame);
}
}
@Override
protected void onMessage(WebSocketFrame messageFrame) {
try {
messageFrame.setUnmasked();
sendFrame(messageFrame);
} catch (IOException e) {
throw new RuntimeException(e);
}
}
@Override
protected void onClose(WebSocketFrame.CloseCode code, String reason, boolean initiatedByRemote) {
if (debug) {
System.out.println("C [" + (initiatedByRemote ? "Remote" : "Self") + "] " +
(code != null ? code : "UnknownCloseCode[" + code + "]") +
(reason != null && !reason.isEmpty() ? ": " + reason : ""));
}
}
@Override
protected void onException(IOException e) {
e.printStackTrace();
}
@Override
protected void handleWebsocketFrame(WebSocketFrame frame) throws IOException {
if (debug) {
System.out.println("R " + frame);
}
super.handleWebsocketFrame(frame);
}
@Override
public synchronized void sendFrame(WebSocketFrame frame) throws IOException {
if (debug) {
System.out.println("S " + frame);
}
super.sendFrame(frame);
}
}
|