Set mode for insert request

This approach supports inheritance approach. If something will be
changed in Connection code (except for removing _build_*_request func),
you should not do anything more to support it in MeshConnection code.

Author: DifferentialOrange

tarantool/connection.py

@@ -527,6 +527,11 @@ def subscribe(self, cluster_uuid, server_uuid, vclock=None):
                 return
         self.close()  # close connection after SUBSCRIBE
 
+    def _build_insert_request(self, space_name, values):
+        if isinstance(space_name, string_types):
+            space_name = self.schema.get_space(space_name).sid
+        return RequestInsert(self, space_name, values)
+
     def insert(self, space_name, values):
         '''
         Execute INSERT request.
@@ -540,9 +545,7 @@ def insert(self, space_name, values):
 
         :rtype: `Response` instance
         '''
-        if isinstance(space_name, string_types):
-            space_name = self.schema.get_space(space_name).sid
-        request = RequestInsert(self, space_name, values)
+        request = self._build_insert_request(space_name, values)
         return self._send_request(request)
 
     def delete(self, space_name, key, **kwargs):

tarantool/mesh_connection.py

@@ -5,7 +5,7 @@
 '''
 
 import time
-
+from enum import Enum
 
 from tarantool.connection import Connection
 from tarantool.error import (
@@ -34,6 +34,11 @@
     string_types = str
 
 
+class Mode(Enum):
+    RW = 1
+    PREFER_RW = 2
+    PREFER_RO = 3
+
 def parse_uri(uri):
     # TODO: Support Unix sockets.
     def parse_error(uri, msg):
@@ -363,3 +368,8 @@ def _send_request(self, request):
         '''
         self._opt_refresh_instances()
         return super(MeshConnection, self)._send_request(request)
+
+    def _build_insert_request(self, space_name, values):
+        request = super(MeshConnection, self)._build_insert_request(space_name, values)
+        request.mode = Mode.RW
+        return request

tarantool/request.py

@@ -76,6 +76,8 @@ def __init__(self, conn):
         self._sync = None
         self._body = ''
         self.response_class = Response
+        # Mode is supported only in MeshConnection.
+        self.mode = None
 
         packer_kwargs = dict()