Can now modify response header during handshake in SocketServer::on_handshake handler. Also corrected status_code needed for upgrade in on_upgrade.

1 files changed, 5 insertions, 0 deletions

diff --git a/ws_examples.cpp b/ws_examples.cpp
index 828fec8..b858048 100644
--- a/ws_examples.cpp
+++ b/ws_examples.cpp
@@ -50,6 +50,11 @@ int main() {
     cout << "Server: Closed connection " << connection.get() << " with status code " << status << endl;
   };
 
+  // Can modify handshake response headers here if needed
+  echo.on_handshake = [](shared_ptr<WsServer::Connection> /*connection*/, SimpleWeb::CaseInsensitiveMultimap & /*response_header*/) {
+    return SimpleWeb::StatusCode::information_switching_protocols; // Upgrade to websocket
+  };
+
   // See http://www.boost.org/doc/libs/1_55_0/doc/html/boost_asio/reference.html, Error Codes for error code meanings
   echo.on_error = [](shared_ptr<WsServer::Connection> connection, const SimpleWeb::error_code &ec) {
     cout << "Server: Error in connection " << connection.get() << ". "