add user data

qicosmos · Dec 24, 2024 · 7419a93 · 7419a93
1 parent 22fa4d0
commit 7419a93
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/include/cinatra/coro_http_request.hpp b/include/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
diff --git a/tests/test_cinatra.cpp b/tests/test_cinatra.cpp
@@ -547,13 +547,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;
   }
 };
@@ -570,6 +579,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;
   }
@@ -603,9 +614,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) {