Allow having no close handler

2020-01-19 19:46:51 +01:00
parent 61b69169c7
commit 0faebc7590
1 changed files with 3 additions and 1 deletions
@@ -138,7 +138,9 @@ public:
        webSocketContext->getExt()->messageHandler = std::move(behavior.message);
        webSocketContext->getExt()->drainHandler = std::move(behavior.drain);
        webSocketContext->getExt()->closeHandler = std::move([closeHandler = std::move(behavior.close)](WebSocket<SSL, true> *ws, int code, std::string_view message) mutable {
-            closeHandler(ws, code, message);
+            if (closeHandler) {
+                closeHandler(ws, code, message);
+            }

            /* Destruct user data after returning from close handler */
            ((UserData *) ws->getUserData())->~UserData();