From 804e0e27e769334ee67d00f06e0ab06850d262ad Mon Sep 17 00:00:00 2001
From: Redkale <22250530@qq.com>
Date: Tue, 23 May 2017 22:14:03 +0800
Subject: [PATCH]

---
 src/org/redkale/net/http/HttpServer.java | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/org/redkale/net/http/HttpServer.java b/src/org/redkale/net/http/HttpServer.java
index abbc6d302..a66689f14 100644
--- a/src/org/redkale/net/http/HttpServer.java
+++ b/src/org/redkale/net/http/HttpServer.java
@@ -181,11 +181,28 @@ public class HttpServer extends Server<String, HttpContext, HttpRequest, HttpRes
         return addRestServlet(name, serviceType, service, userType, baseServletClass, prefix, null);
     }
 
+    /**
+     * 添加WebSocketServlet
+     *
+     * @param <S>           WebSocket
+     * @param <T>           HttpServlet
+     * @param webSocketType WebSocket的类型
+     * @param prefix        url前缀
+     * @param conf          配置信息
+     *
+     * @return RestServlet
+     */
+    public <S extends WebSocket, T extends HttpServlet> T addRestWebSocketServlet(final Class<S> webSocketType, final String prefix, final AnyValue conf) {
+        T servlet = Rest.createRestWebSocketServlet(webSocketType);
+        if (servlet != null) this.prepare.addServlet(servlet, prefix, conf);
+        return servlet;
+    }
+
     /**
      * 添加RestServlet
      *
      * @param <S>             Service
-     * @param <T>             RestServlet
+     * @param <T>             HttpServlet
      * @param name            Service的资源名
      * @param serviceType     Service的类型
      * @param service         Service对象