1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.eclipse.jetty.websocket.common.events;
20
21 import java.io.IOException;
22 import java.io.InputStream;
23 import java.io.Reader;
24 import java.nio.ByteBuffer;
25
26 import org.eclipse.jetty.util.log.Log;
27 import org.eclipse.jetty.util.log.Logger;
28 import org.eclipse.jetty.websocket.api.WebSocketListener;
29 import org.eclipse.jetty.websocket.api.WebSocketPolicy;
30 import org.eclipse.jetty.websocket.api.extensions.Frame;
31 import org.eclipse.jetty.websocket.common.CloseInfo;
32 import org.eclipse.jetty.websocket.common.message.SimpleBinaryMessage;
33 import org.eclipse.jetty.websocket.common.message.SimpleTextMessage;
34
35
36
37
38 public class JettyListenerEventDriver extends AbstractEventDriver
39 {
40 private static final Logger LOG = Log.getLogger(JettyListenerEventDriver.class);
41 private final WebSocketListener listener;
42 private boolean hasCloseBeenCalled = false;
43
44 public JettyListenerEventDriver(WebSocketPolicy policy, WebSocketListener listener)
45 {
46 super(policy,listener);
47 this.listener = listener;
48 }
49
50 @Override
51 public void onBinaryFrame(ByteBuffer buffer, boolean fin) throws IOException
52 {
53 if (activeMessage == null)
54 {
55 activeMessage = new SimpleBinaryMessage(this);
56 }
57
58 appendMessage(buffer,fin);
59 }
60
61 @Override
62 public void onBinaryMessage(byte[] data)
63 {
64 listener.onWebSocketBinary(data,0,data.length);
65 }
66
67 @Override
68 public void onClose(CloseInfo close)
69 {
70 if (hasCloseBeenCalled)
71 {
72
73 return;
74 }
75 hasCloseBeenCalled = true;
76
77 int statusCode = close.getStatusCode();
78 String reason = close.getReason();
79 listener.onWebSocketClose(statusCode,reason);
80 }
81
82 @Override
83 public void onConnect()
84 {
85 LOG.debug("onConnect()");
86 listener.onWebSocketConnect(session);
87 }
88
89 @Override
90 public void onError(Throwable cause)
91 {
92 listener.onWebSocketError(cause);
93 }
94
95 @Override
96 public void onFrame(Frame frame)
97 {
98
99 }
100
101 @Override
102 public void onInputStream(InputStream stream)
103 {
104
105 }
106
107 @Override
108 public void onReader(Reader reader)
109 {
110
111 }
112
113 @Override
114 public void onTextFrame(ByteBuffer buffer, boolean fin) throws IOException
115 {
116 if (activeMessage == null)
117 {
118 activeMessage = new SimpleTextMessage(this);
119 }
120
121 appendMessage(buffer,fin);
122 }
123
124 @Override
125 public void onTextMessage(String message)
126 {
127 listener.onWebSocketText(message);
128 }
129
130 @Override
131 public String toString()
132 {
133 return String.format("%s[%s]",JettyListenerEventDriver.class.getSimpleName(),listener.getClass().getName());
134 }
135 }