set_ws_deflate

helintongh · Apr 19, 2024 · e001f60 · e001f60
1 parent 3819115
commit e001f60
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/include/cinatra/coro_http_client.hpp b/include/cinatra/coro_http_client.hpp
@@ -276,18 +276,21 @@ class coro_http_client : public std::enable_shared_from_this<coro_http_client> {
     return std::move(body_);
   }
 
+#ifdef CINATRA_ENABLE_GZIP
+  void set_ws_deflate(bool enable_ws_deflate) {
+    enable_ws_deflate_ = enable_ws_deflate;
+  }
+#endif
+
   // only make socket connet(or handshake) to the host
-  async_simple::coro::Lazy<resp_data> connect(std::string uri,
-                                              bool enable_ws_deflate = false) {
+  async_simple::coro::Lazy<resp_data> connect(std::string uri) {
     resp_data data{};
     bool no_schema = !has_schema(uri);
     std::string append_uri;
     if (no_schema) {
       append_uri.append("http://").append(uri);
     }
 
-    enable_ws_deflate_ = enable_ws_deflate;
-
     auto [ok, u] = handle_uri(data, no_schema ? append_uri : uri);
     if (!ok) {
       co_return resp_data{std::make_error_code(std::errc::protocol_error), 404};
@@ -2123,8 +2126,8 @@ class coro_http_client : public std::enable_shared_from_this<coro_http_client> {
   std::string resp_chunk_str_;
   std::span<char> out_buf_;
 
-  bool enable_ws_deflate_ = false;
 #ifdef CINATRA_ENABLE_GZIP
+  bool enable_ws_deflate_ = false;
   bool is_server_support_ws_deflate_ = false;
   std::string inflate_str_;
 #endif

diff --git a/tests/test_cinatra_websocket.cpp b/tests/test_cinatra_websocket.cpp
@@ -325,8 +325,9 @@ TEST_CASE("test websocket permessage defalte") {
   std::this_thread::sleep_for(std::chrono::milliseconds(300));
 
   coro_http_client client{};
+  client.set_ws_deflate(true);
   async_simple::coro::syncAwait(
-      client.connect("ws://localhost:8090/ws_extesion", true));
+      client.connect("ws://localhost:8090/ws_extesion"));
 
   std::string send_str("test");