add some methods (qicosmos#489)

helintongh · Jan 15, 2024 · 595a530 · 595a530
1 parent ec7f8d6
commit 595a530
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/include/cinatra/coro_http_connection.hpp b/include/cinatra/coro_http_connection.hpp
@@ -300,6 +300,36 @@ class coro_http_connection
     co_return true;
   }
 
+  std::string local_address() {
+    if (has_closed_) {
+      return "";
+    }
+
+    std::stringstream ss;
+    std::error_code ec;
+    ss << socket_.local_endpoint(ec);
+    if (ec) {
+      return "";
+    }
+    return ss.str();
+  }
+
+  std::string remote_address() {
+    static std::string remote_addr;
+    if (has_closed_) {
+      return remote_addr;
+    }
+
+    std::stringstream ss;
+    std::error_code ec;
+    ss << socket_.remote_endpoint(ec);
+    if (ec) {
+      return remote_addr;
+    }
+    remote_addr = ss.str();
+    return ss.str();
+  }
+
   async_simple::coro::Lazy<bool> write_data(std::string_view message) {
     std::vector<asio::const_buffer> buffers;
     buffers.push_back(asio::buffer(message));

diff --git a/include/cinatra/coro_http_request.hpp b/include/cinatra/coro_http_request.hpp
@@ -169,6 +169,10 @@ class coro_http_request {
     return content_type::unknown;
   }
 
+  std::string_view get_url() { return parser_.url(); }
+
+  std::string_view get_method() { return parser_.method(); }
+
   std::string_view get_boundary() {
     auto content_type = get_header_value("content-type");
     if (content_type.empty()) {

diff --git a/tests/test_coro_http_server.cpp b/tests/test_coro_http_server.cpp
@@ -389,6 +389,11 @@ TEST_CASE("get post") {
 
   server.set_http_handler<cinatra::GET, cinatra::POST>(
       "/test1", [](coro_http_request &req, coro_http_response &resp) {
+        CHECK(req.get_method() == "POST");
+        CHECK(req.get_url() == "/test1");
+        CHECK(req.get_conn()->local_address() == "127.0.0.1:9001");
+        CHECK(req.get_conn()->remote_address().find("127.0.0.1:") !=
+              std::string::npos);
         resp.add_header("Host", "Cinatra");
         resp.set_status_and_content(cinatra::status_type::ok, "hello world");
       });