不受控制的客户端内存使用量过大(websocketx)

Question

当向简单的Netty WebSocket echo服务器实现(示例包中找到的实现的稍微修改版本)发送不受控制的大量TextWebSocketFrames时，无需等待ChannelFuture上的客户端sync()，服务器上的堆内存使用量将呈指数级增长，直到最终耗尽内存。

Testcase:在写入下一个文本帧之前，客户端不等待实际的字节被写入，也不等待服务器将文本“回显”给客户端

for (int i = 0; i < 10000000; i++) {
    ch.write(new TextWebSocketFrame("Message #" + i));
}

当观察yourkit内存探查器时（测试开始15秒，写入约20.000帧），BigEndianHeapChannelBuffer对象的数量增长过快。

BigEndianHeapChannelBuffer 284,509 (objects) 9,104,288 (shallow size) (after ~30.000 frames sent within 10 second window)

在服务器端，主要从 BigEndianHeapChannelBuffers 和复合通道缓冲器对象中可以观察到一大堆堆积，这些对象永远不会被清理，也不会被垃圾回收（当引用被保存时，这可能是不可能的）。我猜这与（单个）工作线程无法将下游“echo”响应写入客户端通道有关，因为它正忙于处理来自客户端的快速传入请求

有没有办法在服务器端防止/限制这种情况（意外拒绝服务）？

Answer 1

在Netty中，大多数I/O操作都是异步的。因此，在不等待先前写入完成的情况下写入数千条消息将为您带来OutOfMemoryError。为了避免这种情况，我更喜欢使用计数器变量来计算挂起的写入次数。例如：

private final AtomicInteger pendingWrites = new AtomicInteger();

...
while (pendingWrites.get() < MAX_PENDING_WRITES) {
  pendingWrites.incrementAndGet();
  ch.write(msg).addListener(new ChannelFutureListener() {
    ...
    pendingWrites.decrementAndGet();
    if (pendingWrites.get() < MAX_PENDING_WRITES) {
      // resume writing here
    }
  }
}

或者，您可以使用分块写手，它基本上可以完成相同的工作。