diff --git a/src/App.h b/src/App.h index d6b1b06..526dec1 100644 --- a/src/App.h +++ b/src/App.h @@ -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 *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();