move endpoint initialization into constructor #2

1 files changed, 7 insertions, 1 deletions

diff --git a/server_ws.hpp b/server_ws.hpp
index 6674e48..955975c 100644
--- a/server_ws.hpp
+++ b/server_ws.hpp
@@ -121,7 +121,13 @@ namespace SimpleWeb {
       /// Used to call Server::upgrade.
       template <typename... Args>
       Connection(std::shared_ptr<ScopeRunner> handler_runner_, long timeout_idle, Args &&... args) noexcept
-          : handler_runner(std::move(handler_runner_)), socket(new socket_type(std::forward<Args>(args)...)), timeout_idle(timeout_idle), closed(false) {}
+          : handler_runner(std::move(handler_runner_)), socket(new socket_type(std::forward<Args>(args)...)), timeout_idle(timeout_idle), closed(false) {
+          try {
+              endpoint = socket->lowest_layer().remote_endpoint();
+          }
+          catch (...) {
+          }
+      }
 
       std::shared_ptr<ScopeRunner> handler_runner;