From 66e1f58879dfd9f73bfc6857b9f88b9d2defd9dc Mon Sep 17 00:00:00 2001
From: Redkale <8730487+redkale@users.noreply.github.com>
Date: Wed, 8 Jan 2020 10:42:08 +0800
Subject: [PATCH]

---
 test/org/redkale/test/websocket/VideoWebSocketServlet.java | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/test/org/redkale/test/websocket/VideoWebSocketServlet.java b/test/org/redkale/test/websocket/VideoWebSocketServlet.java
index 9d1a298d1..33a6037ba 100644
--- a/test/org/redkale/test/websocket/VideoWebSocketServlet.java
+++ b/test/org/redkale/test/websocket/VideoWebSocketServlet.java
@@ -65,7 +65,7 @@ public class VideoWebSocketServlet extends WebSocketServlet {
             }
 
             @Override
-            public void onConnected() {
+            public CompletableFuture onConnected() {
                 if (repeat) {
                     super.close();
                 } else {
@@ -82,6 +82,7 @@ public class VideoWebSocketServlet extends WebSocketServlet {
                     String msg = ("{'type':'discover_user','user':{'userid':'" + this.getSessionid() + "','username':'" + users.get(this.getSessionid()) + "'}}").replace('\'', '"');
                     super.broadcastMessage(msg);
                 }
+                return null;
             }
 
             @Override
@@ -91,10 +92,10 @@ public class VideoWebSocketServlet extends WebSocketServlet {
             }
 
             @Override
-            public void onClose(int code, String reason) {
+            public CompletableFuture onClose(int code, String reason) {
                 sessions.remove(this.getSessionid());
                 String msg = ("{'type':'remove_user','user':{'userid':'" + this.getSessionid() + "','username':'" + users.get(this.getSessionid()) + "'}}").replace('\'', '"');
-                super.broadcastMessage(msg);
+                return super.broadcastMessage(msg);
             } 
 
             @Override