[coro_http_client][feat]add user data (#871)

Author: qicosmos

include/ylt/standalone/cinatra/coro_http_request.hpp

@@ -248,6 +248,10 @@ class coro_http_request {
     (aspect_data_.push_back(std::move(args)), ...);
   }
 
+  void set_user_data(std::any data) { user_data_ = std::move(data); }
+
+  std::any get_user_data() { return user_data_; }
+
   std::vector<std::string> &get_aspect_data() { return aspect_data_; }
 
   std::unordered_map<std::string_view, std::string_view> get_cookies(
@@ -288,6 +292,9 @@ class coro_http_request {
     if (!aspect_data_.empty()) {
       aspect_data_.clear();
     }
+    if (user_data_.has_value()) {
+      user_data_.reset();
+    }
   }
 
   std::unordered_map<std::string, std::string> params_;
@@ -300,5 +307,6 @@ class coro_http_request {
   bool is_websocket_ = false;
   std::vector<std::string> aspect_data_;
   std::string cached_session_id_;
+  std::any user_data_;
 };
 }  // namespace cinatra

src/coro_http/tests/test_cinatra.cpp

@@ -538,13 +538,22 @@ TEST_CASE("test request https without init_ssl") {
 struct add_data {
   bool before(coro_http_request &req, coro_http_response &res) {
     req.set_aspect_data("hello world");
+    auto val = std::make_shared<int>(42);
+    req.set_user_data(val);
     return true;
   }
 };
 
 struct add_more_data {
   bool before(coro_http_request &req, coro_http_response &res) {
     req.set_aspect_data(std::vector<std::string>{"test", "aspect"});
+    auto user_data = req.get_user_data();
+    CHECK(user_data.has_value());
+    auto val = std::any_cast<std::shared_ptr<int>>(user_data);
+    CHECK(*val == 42);
+    auto data = req.get_user_data();
+    val = std::any_cast<std::shared_ptr<int>>(data);
+    *val = 43;
     return true;
   }
 };
@@ -561,6 +570,8 @@ struct auth_t {
 
 struct dely_t {
   bool before(coro_http_request &req, coro_http_response &res) {
+    auto user_data = req.get_user_data();
+    CHECK(!user_data.has_value());
     res.set_status_and_content(status_type::unauthorized, "unauthorized");
     return false;
   }
@@ -594,9 +605,13 @@ TEST_CASE("test aspect") {
         CHECK(val[0] == "test");
         CHECK(val[1] == "aspect");
         CHECK(!req.is_upgrade());
+        auto user_data = req.get_user_data();
+        CHECK(user_data.has_value());
+        auto val1 = std::any_cast<std::shared_ptr<int>>(user_data);
+        CHECK(*val1 == 43);
         resp.set_status_and_content(status_type::ok, "ok");
       },
-      add_more_data{});
+      add_data{}, add_more_data{});
   server.set_http_handler<GET>(
       "/auth",
       [](coro_http_request &req, coro_http_response &resp) {

src/metric/tests/test_metric.cpp

@@ -1964,9 +1964,12 @@ TEST_CASE("test metric manager clean expired label") {
   CHECK(c->label_value_count() == 2);
   CHECK(summary->label_value_count() == 1);
   CHECK(h->label_value_count() == 1);
-  std::this_thread::sleep_for(std::chrono::seconds(2));
+  std::this_thread::sleep_for(std::chrono::seconds(3));
   c->inc({"/index"});
   size_t count = c->label_value_count();
+  if (count != 1) {
+    std::this_thread::sleep_for(std::chrono::seconds(2));
+  }
   CHECK(count == 1);
   auto ct1 = summary->label_value_count();
   CHECK(ct1 == 0);