Reload ro info on ro error

Author: DifferentialOrange

tarantool/connection_pool.py

@@ -9,7 +9,8 @@
                              RECONNECT_DELAY, RECONNECT_MAX_ATTEMPTS,
                              SOCKET_TIMEOUT)
 from tarantool.error import (ClusterConnectWarning, ClusterTolopogyWarning,
-                             ConfigurationError, warn)
+                             ConfigurationError, DatabaseError, tnt_strerror,
+                             warn)
 from tarantool.utils import ENCODING_DEFAULT
 
 try:
@@ -184,10 +185,9 @@ def close(self):
     def is_closed(self):
         raise NotImplementedError
 
-    def connect(self):
+    def update_ro_info(self):
         for addr in self.addrs:
             key = json.dumps(addr)
-            self.pool[key]['conn'].connect()
 
             if not self.pool[key]['conn'].connected:
                 msg = 'Failed to connect to %s:%s' % str(addr['host'], addr['port'])
@@ -203,14 +203,45 @@ def connect(self):
 
         self.strategy.update(self.pool)
 
+    def connect(self):
+        for addr in self.addrs:
+            key = json.dumps(addr)
+            self.pool[key]['conn'].connect()
+
+        self.update_ro_info()
+
     def call(self, func_name, *args, **kwargs):
         mode = kwargs.get('mode', Mode.PREFER_RW)
         conn = self.strategy.getnext(mode)
 
-        return conn.call(func_name, args)
+        # try:
+        resp = conn.call(func_name, args)
+        # except ValueError:
+            #reload
+            # resp = conn.call(func_name, args)
+        
+        return resp
 
-    def eval(self, expr, *args):
-        raise NotImplementedError
+    def eval(self, expr, *args, **kwargs):        
+        mode = kwargs.get('mode', Mode.PREFER_RW)
+        conn = self.strategy.getnext(mode)
+        
+        try:
+            resp = conn.eval(expr, args)
+        except DatabaseError as e:
+            if len(e.args) > 0:
+                strerror = tnt_strerror(e.args[0])
+            else:
+                raise
+
+            if strerror != 'UNDEFINED' and strerror[0] == 'ER_READONLY':
+                self.update_ro_info()
+                conn = self.strategy.getnext(mode)
+                resp = conn.eval(expr, args)
+            else:
+                raise
+
+        return resp
 
     def replace(self, space_name, values):
         raise NotImplementedError