diff --git a/src/org/redkale/net/http/WebSocket.java b/src/org/redkale/net/http/WebSocket.java index 8f3c3ac19..fb09e8098 100644 --- a/src/org/redkale/net/http/WebSocket.java +++ b/src/org/redkale/net/http/WebSocket.java @@ -8,8 +8,10 @@ package org.redkale.net.http; import org.redkale.net.http.WebSocketPacket.FrameType; import java.io.*; import java.net.*; +import java.nio.ByteBuffer; import java.util.*; import java.util.concurrent.*; +import java.util.function.Supplier; import java.util.stream.Stream; import org.redkale.convert.json.JsonConvert; import org.redkale.net.*; @@ -382,6 +384,15 @@ public abstract class WebSocket { return _engine.getLocalWebSockets(); } + /** + * 获取ByteBuffer资源池 + * + * @return Supplier + */ + protected Supplier getByteBufferSupplier() { + return this._runner.context.getBufferSupplier(); + } + //------------------------------------------------------------------- /** * 返回sessionid, null表示连接不合法或异常,默认实现是request.sessionid(true),通常需要重写该方法