feat: write_end_callback for stream and ws connection

src/connection/stream_connection.cpp

@@ -143,6 +143,10 @@ bool stream_connection::buf2sock(bool &closed)
                     }
                     else
                     {
+                        if (this->write_end_callback)
+                        {
+                            return this->write_end_callback(*this);
+                        }
                         return false; // nothing m_waiting_send_pack to m_send_buffer
                     }
                 }
@@ -250,4 +254,6 @@ void stream_connection::reuse()
     this->should_send_idx = -1;
     this->should_send_size = 0;
     this->sock2buf_data_len = 0;
+
+    write_end_callback = nullptr;
 }

src/connection/stream_connection.h

@@ -43,6 +43,7 @@ namespace tubekit
             buffer::buffer m_send_buffer;
             buffer::buffer m_recv_buffer;
             buffer::buffer m_wating_send_pack;
+            std::function<bool(stream_connection &connection)> write_end_callback{nullptr};
 
         private:
             int should_send_idx{-1};

src/connection/websocket_connection.cpp

@@ -58,6 +58,7 @@ void websocket_connection::reuse()
     this->m_wating_send_pack.set_limit_max(mem_buffer_size_max);
 
     this->destory_callback = nullptr;
+    this->write_end_callback = nullptr;
 
     http_parser_init(&m_http_parser, HTTP_REQUEST);
     m_http_parser.data = this;
@@ -209,6 +210,10 @@ bool websocket_connection::buf2sock(bool &closed)
                     }
                     else
                     {
+                        if (this->write_end_callback)
+                        {
+                            return this->write_end_callback(*this);
+                        }
                         return false; // noting m_waiting_send_pack to m_send_buffer
                     }
                 }

src/connection/websocket_connection.h

@@ -78,7 +78,8 @@ namespace tubekit
             buffer::buffer m_send_buffer;
             buffer::buffer m_wating_send_pack;
 
-            std::function<void(websocket_connection &connection)> destory_callback;
+            std::function<void(websocket_connection &connection)> destory_callback{nullptr};
+            std::function<bool(websocket_connection &connection)> write_end_callback{nullptr};
 
         private:
             int should_send_idx{-1};