iproto: support feature push

Adds support for receiving out-of-band messages
from a server that uses box.session.push call.

Data obtaining is possible for methods:
`call`, `eval`, `select`, `insert`, `replace`,
`update`, `upsert`, `delete`.
To do this, an optional argument `on_push_ctx`
is used in the form of a python list, where the
received data from out-of-band messages will be added.

Closes #201

Author: GRISHNOV

tarantool/connection.py

@@ -55,7 +55,8 @@
     REQUEST_TYPE_ERROR,
     IPROTO_GREETING_SIZE,
     ITERATOR_EQ,
-    ITERATOR_ALL
+    ITERATOR_ALL,
+    IPROTO_CHUNK
 )
 from tarantool.error import (
     Error,
@@ -748,7 +749,7 @@ def _read_response(self):
         # Read the packet
         return self._recv(length)
 
-    def _send_request_wo_reconnect(self, request):
+    def _send_request_wo_reconnect(self, request, on_push_ctx=[]):
         """
         Send request without trying to reconnect.
         Reload schema, if required.
@@ -772,6 +773,11 @@ def _send_request_wo_reconnect(self, request):
             try:
                 self._socket.sendall(bytes(request))
                 response = request.response_class(self, self._read_response())
+                if response._code == IPROTO_CHUNK:
+                    # Сase of receiving an out-of-band message
+                    on_push_ctx.append(response._data)
+                    # Next comes the main message
+                    response = request.response_class(self, self._read_response())
                 break
             except SchemaReloadException as e:
                 self.update_schema(e.schema_version)
@@ -851,7 +857,7 @@ def check():  # Check that connection is alive
             self.wrap_socket_ssl()
         self.handshake()
 
-    def _send_request(self, request):
+    def _send_request(self, request, on_push_ctx=[]):
         """
         Send a request to the server through the socket.
 
@@ -872,7 +878,7 @@ def _send_request(self, request):
 
         self._opt_reconnect()
 
-        return self._send_request_wo_reconnect(request)
+        return self._send_request_wo_reconnect(request, on_push_ctx)
 
     def load_schema(self):
         """
@@ -914,7 +920,7 @@ def flush_schema(self):
         self.schema.flush()
         self.load_schema()
 
-    def call(self, func_name, *args):
+    def call(self, func_name, *args, **kwargs):
         """
         Execute a CALL request: call a stored Lua function.
 
@@ -930,19 +936,26 @@ def call(self, func_name, *args):
             :exc:`~tarantool.error.SchemaError`,
             :exc:`~tarantool.error.NetworkError`,
             :exc:`~tarantool.error.SslError`
+
+        !!!
+        TODO: write docs
+        !!!
         """
 
         assert isinstance(func_name, str)
 
         # This allows to use a tuple or list as an argument
         if len(args) == 1 and isinstance(args[0], (list, tuple)):
             args = args[0]
+        # Case for absence of optional arg for accepting out-of-band msg data
+        if not 'on_push_ctx' in kwargs:
+            kwargs['on_push_ctx'] = []
 
         request = RequestCall(self, func_name, args, self.call_16)
-        response = self._send_request(request)
+        response = self._send_request(request, kwargs['on_push_ctx'])
         return response
 
-    def eval(self, expr, *args):
+    def eval(self, expr, *args, **kwargs):
         """
         Execute an EVAL request: evaluate a Lua expression.
 
@@ -966,12 +979,15 @@ def eval(self, expr, *args):
         # This allows to use a tuple or list as an argument
         if len(args) == 1 and isinstance(args[0], (list, tuple)):
             args = args[0]
+        # Case for absence of optional arg for accepting out-of-band msg data
+        if not 'on_push_ctx' in kwargs:
+            kwargs['on_push_ctx'] = []
 
         request = RequestEval(self, expr, args)
-        response = self._send_request(request)
+        response = self._send_request(request, kwargs['on_push_ctx'])
         return response
 
-    def replace(self, space_name, values):
+    def replace(self, space_name, values, **kwargs):
         """
         Execute a REPLACE request: `replace`_ a tuple in the space.
         Doesn't throw an error if there is no tuple with the specified
@@ -994,10 +1010,14 @@ def replace(self, space_name, values):
         .. _replace: https://www.tarantool.io/en/doc/latest/reference/reference_lua/box_space/replace/
         """
 
+        # Case for absence of optional arg for accepting out-of-band msg data
+        if not 'on_push_ctx' in kwargs:
+            kwargs['on_push_ctx'] = []
+
         if isinstance(space_name, str):
             space_name = self.schema.get_space(space_name).sid
         request = RequestReplace(self, space_name, values)
-        return self._send_request(request)
+        return self._send_request(request, kwargs['on_push_ctx'])
 
     def authenticate(self, user, password):
         """
@@ -1149,7 +1169,7 @@ def subscribe(self, cluster_uuid, server_uuid, vclock=None):
                 return
         self.close()  # close connection after SUBSCRIBE
 
-    def insert(self, space_name, values):
+    def insert(self, space_name, values, **kwargs):
         """
         Execute an INSERT request: `insert`_ a tuple to the space.
         Throws an error if there is already a tuple with the same
@@ -1172,12 +1192,16 @@ def insert(self, space_name, values):
         .. _insert: https://www.tarantool.io/en/doc/latest/reference/reference_lua/box_space/insert/
         """
 
+        # Case for absence of optional arg for accepting out-of-band msg data
+        if not 'on_push_ctx' in kwargs:
+            kwargs['on_push_ctx'] = []
+
         if isinstance(space_name, str):
             space_name = self.schema.get_space(space_name).sid
         request = RequestInsert(self, space_name, values)
-        return self._send_request(request)
+        return self._send_request(request, kwargs['on_push_ctx'])
 
-    def delete(self, space_name, key, *, index=0):
+    def delete(self, space_name, key, *, index=0, **kwargs):
         """
         Execute a DELETE request: `delete`_ a tuple in the space.
 
@@ -1202,15 +1226,19 @@ def delete(self, space_name, key, *, index=0):
         .. _delete: https://www.tarantool.io/en/doc/latest/reference/reference_lua/box_space/delete/
         """
 
+        # Case for absence of optional arg for accepting out-of-band msg data
+        if not 'on_push_ctx' in kwargs:
+            kwargs['on_push_ctx'] = []
+
         key = check_key(key)
         if isinstance(space_name, str):
             space_name = self.schema.get_space(space_name).sid
         if isinstance(index, str):
             index = self.schema.get_index(space_name, index).iid
         request = RequestDelete(self, space_name, index, key)
-        return self._send_request(request)
+        return self._send_request(request, kwargs['on_push_ctx'])
 
-    def upsert(self, space_name, tuple_value, op_list, *, index=0):
+    def upsert(self, space_name, tuple_value, op_list, *, index=0, **kwargs):
         """
         Execute an UPSERT request: `upsert`_ a tuple to the space.
 
@@ -1252,16 +1280,20 @@ def upsert(self, space_name, tuple_value, op_list, *, index=0):
         .. _upsert: https://www.tarantool.io/en/doc/latest/reference/reference_lua/box_space/upsert/
         """
 
+        # Case for absence of optional arg for accepting out-of-band msg data
+        if not 'on_push_ctx' in kwargs:
+            kwargs['on_push_ctx'] = []
+
         if isinstance(space_name, str):
             space_name = self.schema.get_space(space_name).sid
         if isinstance(index, str):
             index = self.schema.get_index(space_name, index).iid
         op_list = self._ops_process(space_name, op_list)
         request = RequestUpsert(self, space_name, index, tuple_value,
                                 op_list)
-        return self._send_request(request)
+        return self._send_request(request, kwargs['on_push_ctx'])
 
-    def update(self, space_name, key, op_list, *, index=0):
+    def update(self, space_name, key, op_list, *, index=0, **kwargs):
         """
         Execute an UPDATE request: `update`_ a tuple in the space.
 
@@ -1331,14 +1363,18 @@ def update(self, space_name, key, op_list, *, index=0):
         .. _update: https://www.tarantool.io/en/doc/latest/reference/reference_lua/box_space/update/
         """
 
+        # Case for absence of optional arg for accepting out-of-band msg data
+        if not 'on_push_ctx' in kwargs:
+            kwargs['on_push_ctx'] = []
+
         key = check_key(key)
         if isinstance(space_name, str):
             space_name = self.schema.get_space(space_name).sid
         if isinstance(index, str):
             index = self.schema.get_index(space_name, index).iid
         op_list = self._ops_process(space_name, op_list)
         request = RequestUpdate(self, space_name, index, key, op_list)
-        return self._send_request(request)
+        return self._send_request(request, kwargs['on_push_ctx'])
 
     def ping(self, notime=False):
         """
@@ -1368,7 +1404,7 @@ def ping(self, notime=False):
             return "Success"
         return t1 - t0
 
-    def select(self, space_name, key=None, *, offset=0, limit=0xffffffff, index=0, iterator=None):
+    def select(self, space_name, key=None, *, offset=0, limit=0xffffffff, index=0, iterator=None, **kwargs):
         """
         Execute a SELECT request: `select`_ a tuple from the space.
 
@@ -1518,13 +1554,17 @@ def select(self, space_name, key=None, *, offset=0, limit=0xffffffff, index=0, i
         # tuples)
         key = check_key(key, select=True)
 
+        # Case for absence of optional arg for accepting out-of-band msg data
+        if not 'on_push_ctx' in kwargs:
+            kwargs['on_push_ctx'] = []
+
         if isinstance(space_name, str):
             space_name = self.schema.get_space(space_name).sid
         if isinstance(index, str):
             index = self.schema.get_index(space_name, index).iid
         request = RequestSelect(self, space_name, index, key, offset,
                                 limit, iterator)
-        response = self._send_request(request)
+        response = self._send_request(request, kwargs['on_push_ctx'])
         return response
 
     def space(self, space_name):

tarantool/const.py

@@ -38,6 +38,7 @@
 
 IPROTO_GREETING_SIZE = 128
 IPROTO_BODY_MAX_LEN = 2147483648
+IPROTO_CHUNK=0x80
 
 REQUEST_TYPE_OK = 0
 REQUEST_TYPE_SELECT = 1

tarantool/mesh_connection.py

@@ -581,7 +581,7 @@ def _opt_refresh_instances(self):
             update_connection(self, addr)
             self._opt_reconnect()
 
-    def _send_request(self, request):
+    def _send_request(self, request, on_push_ctx=[]):
         """
         Send a request to a Tarantool server. If required, refresh
         addresses list before sending a request.