add tcp_nodelay option to Client

thefab · Jun 5, 2015 · f2ff25e · f2ff25e
1 parent 7bd0b1c
commit f2ff25e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tornadis/client.py b/tornadis/client.py
@@ -32,6 +32,7 @@ class Client(object):
         read_page_size (int): page size for reading.
         write_page_size (int): page size for writing.
         connect_timeout (int): timeout (in seconds) for connecting.
+        tcp_nodelay (boolean): set TCP_NODELAY on socket.
         subscribed (boolean): True if the client is in subscription mode.
         autoconnect (boolean): True if the client is in autoconnect mode
             (and in autoreconnection mode) (default True).
@@ -41,7 +42,7 @@ def __init__(self, host=tornadis.DEFAULT_HOST, port=tornadis.DEFAULT_PORT,
                  read_page_size=tornadis.DEFAULT_READ_PAGE_SIZE,
                  write_page_size=tornadis.DEFAULT_WRITE_PAGE_SIZE,
                  connect_timeout=tornadis.DEFAULT_CONNECT_TIMEOUT,
-                 autoconnect=True, ioloop=None):
+                 tcp_nodelay=False, autoconnect=True, ioloop=None):
         """Constructor.
 
         Args:
@@ -50,6 +51,7 @@ def __init__(self, host=tornadis.DEFAULT_HOST, port=tornadis.DEFAULT_PORT,
             read_page_size (int): page size for reading.
             write_page_size (int): page size for writing.
             connect_timeout (int): timeout (in seconds) for connecting.
+            tcp_nodelay (boolean): set TCP_NODELAY on socket.
             autoconnect (boolean): True if the client is in autoconnect mode
                 (and in autoreconnection mode) (default True).
             ioloop (IOLoop): the tornado ioloop to use.
@@ -59,6 +61,7 @@ def __init__(self, host=tornadis.DEFAULT_HOST, port=tornadis.DEFAULT_PORT,
         self.read_page_size = read_page_size
         self.write_page_size = write_page_size
         self.connect_timeout = connect_timeout
+        self.tcp_nodelay = tcp_nodelay
         self.autoconnect = autoconnect
         self.__ioloop = ioloop or tornado.ioloop.IOLoop.instance()
         self.__connection = None
@@ -99,7 +102,8 @@ def connect(self):
                                        port=self.port, ioloop=self.__ioloop,
                                        read_page_size=self.read_page_size,
                                        write_page_size=self.write_page_size,
-                                       connect_timeout=self.connect_timeout)
+                                       connect_timeout=self.connect_timeout,
+                                       tcp_nodelay=self.tcp_nodelay)
         return self.__connection.connect()
 
     def disconnect(self):