From 0ee5b911d28157cebd8562d73332c7575234973c Mon Sep 17 00:00:00 2001
From: Redkale <22250530@qq.com>
Date: Mon, 15 Jan 2018 14:50:03 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=8D=95=E4=B8=AAWebSocket?=
 =?UTF-8?q?=E5=B9=B6=E5=8F=91=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/org/redkale/net/http/WebSocketRunner.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/org/redkale/net/http/WebSocketRunner.java b/src/org/redkale/net/http/WebSocketRunner.java
index fbf389afc..52aaffd43 100644
--- a/src/org/redkale/net/http/WebSocketRunner.java
+++ b/src/org/redkale/net/http/WebSocketRunner.java
@@ -269,12 +269,13 @@ class WebSocketRunner implements Runnable {
                             ByteBuffer[] buffers = entry.packet.sendBuffers != null ? entry.packet.duplicateSendBuffers() : entry.packet.encode(context.getBufferSupplier());
                             lastSendTime = System.currentTimeMillis();
                             channel.write(buffers, buffers, this);
+                        } else {
+                            writing.set(false);
                         }
                     } catch (Exception e) {
                         context.getLogger().log(Level.WARNING, "WebSocket sendMessage abort on rewrite, force to close channel, live " + (System.currentTimeMillis() - webSocket.getCreatetime()) / 1000 + " seconds", e);
                         closeRunner(RETCODE_SENDEXCEPTION, "websocket send message failed on rewrite");
                     }
-                    writing.set(false);
                 }
 
                 @Override