瀏覽代碼

Adding removeWebSocketHandler

Philipp Hasper 9 年之前
父節點
當前提交
a9c11e10be
共有 2 個文件被更改,包括 15 次插入0 次删除
  1. 9 0
      include/CivetServer.h
  2. 6 0
      src/CivetServer.cpp

+ 9 - 0
include/CivetServer.h

@@ -243,6 +243,15 @@ class CIVETWEB_API CivetServer
 	 */
 	void removeHandler(const std::string &uri);
 
+    /**
+     * removeWebSocketHandler(const std::string &)
+     *
+     * Removes a web socket handler.
+     *
+     * @param uri - the exact URL used in addWebSocketHandler().
+     */
+    void removeWebSocketHandler(const std::string &uri);
+
 	/**
 	 * getListeningPorts()
 	 *

+ 6 - 0
src/CivetServer.cpp

@@ -285,6 +285,12 @@ void
 CivetServer::removeHandler(const std::string &uri)
 {
 	mg_set_request_handler(context, uri.c_str(), NULL, NULL);
+}
+
+void
+CivetServer::removeWebSocketHandler(const std::string &uri)
+{
+    mg_set_websocket_handler(context, uri.c_str(), NULL, NULL, NULL, NULL, NULL);
 }
 
 void