[filter] demuxHTTP() now supports maxMessages options

flomesh-io · Sep 14, 2023 · 4d23f19 · 4d23f19
1 parent e2d9194
commit 4d23f19
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/filters/http.cpp b/src/filters/http.cpp
@@ -1228,6 +1228,9 @@ Demux::Options::Options(pjs::Object *options)
   Value(options, "maxHeaderSize")
     .get_binary_size(max_header_size)
     .check_nullable();
+  Value(options, "maxMessages")
+    .get(max_messages)
+    .check_nullable();
 }
 
 //
@@ -1290,6 +1293,7 @@ void Demux::reset() {
   Decoder::chain(DemuxQueue::input());
   m_request_queue.reset();
   m_eos = nullptr;
+  m_message_count = 0;
   m_http2 = false;
   m_shutdown = false;
 }
@@ -1366,7 +1370,13 @@ auto Demux::on_encode_response(ResponseHead *head) -> RequestQueue::Request* {
     return nullptr;
   } else {
     auto req = m_request_queue.shift();
-    if (m_shutdown && m_request_queue.empty()) req->is_final = true;
+    m_message_count++;
+    if (
+      (m_options.max_messages > 0 && m_message_count >= m_options.max_messages) ||
+      (m_shutdown && m_request_queue.empty())
+    ) {
+      req->is_final = true;
+    }
     return req;
   }
 }

diff --git a/src/filters/http.hpp b/src/filters/http.hpp
@@ -294,6 +294,7 @@ class Demux :
   struct Options : public http2::Endpoint::Options {
     size_t buffer_size = DATA_CHUNK_SIZE;
     size_t max_header_size = DATA_CHUNK_SIZE;
+    int max_messages = 0;
     Options() {}
     Options(pjs::Object *options);
   };
@@ -314,6 +315,7 @@ class Demux :
   Options m_options;
   RequestQueue m_request_queue;
   pjs::Ref<StreamEnd> m_eos;
+  int m_message_count = 0;
   bool m_http2 = false;
   bool m_shutdown = false;