add endpoint print functions

2020-03-02 09:41:06 +08:00
parent 57f069135b
commit 2cf48aed35
3 changed files with 30 additions and 0 deletions
@@ -5,6 +5,7 @@
 #include <filesystem>
 #include <fstream>
 #include <iomanip>  // for put_time
+#include <iostream>
 #include <sstream>

 #include "webcc/string.h"
@@ -72,5 +73,21 @@ void DumpByLine(const std::string& data, std::ostream& os,
  }
 }

+void PrintEndpoint(std::ostream& ostream,
+                   const asio::ip::tcp::endpoint& endpoint) {
+  ostream << endpoint;
+  if (endpoint.protocol() == asio::ip::tcp::v4()) {
+    ostream << ", v4";
+  } else if (endpoint.protocol() == asio::ip::tcp::v6()) {
+    ostream << ", v6";
+  }
+}
+
+std::string EndpointToString(const asio::ip::tcp::endpoint& endpoint) {
+  std::stringstream ss;
+  PrintEndpoint(ss, endpoint);
+  return ss.str();
+}
+
 }  // namespace utility
 }  // namespace webcc