From d9bf5c8412db0df4b0d1ed2144225c9fb163d507 Mon Sep 17 00:00:00 2001
From: Redkale <22250530@qq.com>
Date: Thu, 25 May 2017 17:54:14 +0800
Subject: [PATCH]

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

diff --git a/src/org/redkale/net/http/WebSocketEngine.java b/src/org/redkale/net/http/WebSocketEngine.java
index dc6eed82a..a43ce173a 100644
--- a/src/org/redkale/net/http/WebSocketEngine.java
+++ b/src/org/redkale/net/http/WebSocketEngine.java
@@ -103,6 +103,9 @@ public final class WebSocketEngine {
     }
 
     CompletableFuture<Integer> sendMessage(final boolean recent, final Object message, final boolean last, final Serializable... groupids) {
+        if (message instanceof CompletableFuture) {
+            return ((CompletableFuture) message).thenCompose((json) -> sendMessage(recent, json, last, groupids));
+        }
         CompletableFuture<Integer> future = null;
         for (Serializable groupid : groupids) {
             WebSocketGroup group = getWebSocketGroup(groupid);