Assume WsConfig will always be present with WebSocket support on the …

…classpath. Signed-off-by: Santiago Pericasgeertsen <[email protected]>
helidon-io · Nov 20, 2023 · 36e579c · 36e579c
1 parent 4c47c47
commit 36e579c
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/webserver/websocket/src/main/java/io/helidon/webserver/websocket/WsConnection.java b/webserver/websocket/src/main/java/io/helidon/webserver/websocket/WsConnection.java
@@ -84,7 +84,7 @@ private WsConnection(ConnectionContext ctx,
                                       .stream()
                                       .filter(p -> p instanceof WsConfig)
                                       .findFirst()
-                                      .orElse(null);
+                                      .orElseThrow(() -> new InternalError("Unable to find WebSocket config"));
     }
 
     /**
@@ -253,9 +253,7 @@ private boolean processFrame(ClientWsFrame frame) {
 
     private ClientWsFrame readFrame() {
         try {
-            int maxFrameLength = wsConfig != null ? wsConfig.maxFrameLength()
-                                                  : Integer.parseInt(MAX_FRAME_LENGTH);
-            return ClientWsFrame.read(ctx, dataReader, maxFrameLength);
+            return ClientWsFrame.read(ctx, dataReader, wsConfig.maxFrameLength());
         } catch (DataReader.InsufficientDataAvailableException e) {
             throw new CloseConnectionException("Socket closed by the other side", e);
         } catch (WsCloseException e) {