Allow having no close handler

yongju-hong · Jan 19, 2020 · 0faebc7 · 0faebc7
1 parent 61b6916
commit 0faebc7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/App.h b/src/App.h
@@ -138,7 +138,9 @@ struct TemplatedApp {
         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();