From 0e6e225fb907ab731d35c13d98473b2bcae4ad71 Mon Sep 17 00:00:00 2001
From: rstoyanchev <rstoyanchev@vmware.com>
Date: Wed, 29 Nov 2023 18:11:50 +0000
Subject: [PATCH] Implement messageSize methods in JettyWebSocketSession

Closes gh-28325
---
 .../socket/adapter/jetty/JettyWebSocketSession.java    | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/spring-websocket/src/main/java/org/springframework/web/socket/adapter/jetty/JettyWebSocketSession.java b/spring-websocket/src/main/java/org/springframework/web/socket/adapter/jetty/JettyWebSocketSession.java
index 883f913589..f365f717f0 100644
--- a/spring-websocket/src/main/java/org/springframework/web/socket/adapter/jetty/JettyWebSocketSession.java
+++ b/spring-websocket/src/main/java/org/springframework/web/socket/adapter/jetty/JettyWebSocketSession.java
@@ -142,11 +142,10 @@ public class JettyWebSocketSession extends AbstractWebSocketSession<Session> {
 		return (InetSocketAddress) getNativeSession().getRemoteSocketAddress();
 	}
 
-	/**
-	 * This method is a no-op for Jetty.
-	 */
 	@Override
 	public void setTextMessageSizeLimit(int messageSizeLimit) {
+		checkNativeSessionInitialized();
+		getNativeSession().setMaxTextMessageSize(messageSizeLimit);
 	}
 
 	@Override
@@ -155,11 +154,10 @@ public class JettyWebSocketSession extends AbstractWebSocketSession<Session> {
 		return (int) getNativeSession().getMaxTextMessageSize();
 	}
 
-	/**
-	 * This method is a no-op for Jetty.
-	 */
 	@Override
 	public void setBinaryMessageSizeLimit(int messageSizeLimit) {
+		checkNativeSessionInitialized();
+		getNativeSession().setMaxBinaryMessageSize(messageSizeLimit);
 	}
 
 	@Override