Skip to content

Commit f822b0a

Browse files
author
Stephane Landelle
committed
minor clean up
1 parent 613a83a commit f822b0a

File tree

1 file changed

+28
-30
lines changed

1 file changed

+28
-30
lines changed

providers/netty/src/main/java/org/asynchttpclient/providers/netty/ws/NettyWebSocket.java

Lines changed: 28 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,8 @@
3434
import java.util.concurrent.ConcurrentLinkedQueue;
3535

3636
public class NettyWebSocket implements WebSocket {
37-
private final static Logger logger = LoggerFactory.getLogger(NettyWebSocket.class);
37+
38+
private static final Logger LOGGER = LoggerFactory.getLogger(NettyWebSocket.class);
3839

3940
private final Channel channel;
4041
private final ConcurrentLinkedQueue<WebSocketListener> listeners = new ConcurrentLinkedQueue<WebSocketListener>();
@@ -150,29 +151,28 @@ public void onBinaryFragment(byte[] message, boolean last) {
150151

151152
if (byteBuffer.size() > maxBufferSize) {
152153
byteBuffer.reset();
153-
Exception e = new Exception("Exceeded Netty Web Socket maximum buffer size of " + getMaxBufferSize());
154+
Exception e = new Exception("Exceeded Netty Web Socket maximum buffer size of " + maxBufferSize);
154155
onError(e);
155-
this.close();
156+
close();
156157
return;
157158
}
158159
}
159160

160-
for (WebSocketListener l : listeners) {
161-
if (l instanceof WebSocketByteListener) {
161+
for (WebSocketListener listener : listeners) {
162+
if (listener instanceof WebSocketByteListener) {
163+
WebSocketByteListener byteListener = (WebSocketByteListener) listener;
162164
try {
163165
if (!last) {
164-
WebSocketByteListener.class.cast(l).onFragment(message, last);
166+
byteListener.onFragment(message, last);
167+
} else if (byteBuffer.size() > 0) {
168+
byteBuffer.write(message);
169+
byteListener.onFragment(message, last);
170+
byteListener.onMessage(byteBuffer.toByteArray());
165171
} else {
166-
if (byteBuffer.size() > 0) {
167-
byteBuffer.write(message);
168-
WebSocketByteListener.class.cast(l).onFragment(message, last);
169-
WebSocketByteListener.class.cast(l).onMessage(byteBuffer.toByteArray());
170-
} else {
171-
WebSocketByteListener.class.cast(l).onMessage(message);
172-
}
172+
byteListener.onMessage(message);
173173
}
174174
} catch (Exception ex) {
175-
l.onError(ex);
175+
listener.onError(ex);
176176
}
177177
}
178178
}
@@ -189,9 +189,9 @@ public void onTextFragment(String message, boolean last) {
189189

190190
if (textBuffer.length() > maxBufferSize) {
191191
textBuffer.setLength(0);
192-
Exception e = new Exception("Exceeded Netty Web Socket maximum buffer size of " + getMaxBufferSize());
192+
Exception e = new Exception("Exceeded Netty Web Socket maximum buffer size of " + maxBufferSize);
193193
onError(e);
194-
this.close();
194+
close();
195195
return;
196196
}
197197
}
@@ -202,13 +202,11 @@ public void onTextFragment(String message, boolean last) {
202202
try {
203203
if (!last) {
204204
textListener.onFragment(message, last);
205+
} else if (textBuffer.length() > 0) {
206+
textListener.onFragment(message, last);
207+
textListener.onMessage(textBuffer.append(message).toString());
205208
} else {
206-
if (textBuffer.length() > 0) {
207-
textListener.onFragment(message, last);
208-
textListener.onMessage(textBuffer.append(message).toString());
209-
} else {
210-
textListener.onMessage(message);
211-
}
209+
textListener.onMessage(message);
212210
}
213211
} catch (Exception ex) {
214212
listener.onError(ex);
@@ -221,11 +219,11 @@ public void onTextFragment(String message, boolean last) {
221219
}
222220

223221
public void onError(Throwable t) {
224-
for (WebSocketListener l : listeners) {
222+
for (WebSocketListener listener : listeners) {
225223
try {
226-
l.onError(t);
224+
listener.onError(t);
227225
} catch (Throwable t2) {
228-
logger.error("", t2);
226+
LOGGER.error("", t2);
229227
}
230228

231229
}
@@ -236,14 +234,14 @@ public void onClose() {
236234
}
237235

238236
public void onClose(int code, String reason) {
239-
for (WebSocketListener l : listeners) {
237+
for (WebSocketListener listener : listeners) {
240238
try {
241-
if (l instanceof WebSocketCloseCodeReasonListener) {
242-
WebSocketCloseCodeReasonListener.class.cast(l).onClose(this, code, reason);
239+
if (listener instanceof WebSocketCloseCodeReasonListener) {
240+
WebSocketCloseCodeReasonListener.class.cast(listener).onClose(this, code, reason);
243241
}
244-
l.onClose(this);
242+
listener.onClose(this);
245243
} catch (Throwable t) {
246-
l.onError(t);
244+
listener.onError(t);
247245
}
248246
}
249247
}

0 commit comments

Comments
 (0)