O(1) socket removal

1 files changed, 7 insertions, 0 deletions

diff --git a/src/socket_base.hpp b/src/socket_base.hpp
index 284d2c4..3f5774f 100644
--- a/src/socket_base.hpp
+++ b/src/socket_base.hpp
@@ -66,6 +66,10 @@ namespace zmq
         void detach_inpipe (class reader_t *pipe_);
         void detach_outpipe (class writer_t *pipe_);
 
+        //  Manipulating index in the app_thread's list of sockets.
+        void set_index (int index);
+        int get_index ();
+
     private:
 
         //  Handlers for incoming commands.
@@ -131,6 +135,9 @@ namespace zmq
         sessions_t sessions;
         mutex_t sessions_sync;
 
+        //  Index of the socket in the app_thread's list of sockets.
+        int index;
+
         socket_base_t (const socket_base_t&);
         void operator = (const socket_base_t&);
     };