reset response builder on disconnection #150

Author: Cylix

includes/cpp_redis/builders/reply_builder.hpp

@@ -81,6 +81,11 @@ class reply_builder {
   //!
   bool reply_available(void) const;
 
+  //!
+  //! reset the reply builder to its initial state (clear internal buffer and stages)
+  //!
+  void reset(void);
+
 private:
   //!
   //! build reply using m_buffer content

sources/builders/reply_builder.cpp

@@ -41,6 +41,12 @@ reply_builder::operator<<(const std::string& data) {
   return *this;
 }
 
+void
+reply_builder::reset(void) {
+  m_builder = nullptr;
+  m_buffer.clear();
+}
+
 bool
 reply_builder::build_reply(void) {
   if (!m_buffer.size())

sources/network/redis_connection.cpp

@@ -86,6 +86,8 @@ redis_connection::disconnect(bool wait_for_removal) {
 
   //! clear buffer
   m_buffer.clear();
+  //! clear builder
+  m_builder.reset();
 
   __CPP_REDIS_LOG(debug, "cpp_redis::network::redis_connection disconnected");
 }
@@ -186,6 +188,8 @@ redis_connection::tcp_client_disconnection_handler(void) {
   __CPP_REDIS_LOG(debug, "cpp_redis::network::redis_connection has been disconnected");
   //! clear buffer
   m_buffer.clear();
+  //! clear builder
+  m_builder.reset();
   //! call disconnection handler
   call_disconnection_handler();
 }