GG-32709 [IGNITE-14167] Simplify reconnecting, fix affinity topology … (#24)

Author: isapego

pygridgain/api/affinity.py

@@ -55,11 +55,13 @@
 partition_mapping = StructArray([
     ('is_applicable', Bool),
 
-    ('cache_mapping', Conditional(lambda ctx: ctx['is_applicable'] and ctx['is_applicable'].value == 1,
+    ('cache_mapping', Conditional(['is_applicable'],
+                                  lambda ctx: ctx['is_applicable'] and ctx['is_applicable'].value == 1,
                                   lambda ctx: ctx['is_applicable'],
                                   cache_mapping, empty_cache_mapping)),
 
-    ('node_mapping', Conditional(lambda ctx: ctx['is_applicable'] and ctx['is_applicable'].value == 1,
+    ('node_mapping', Conditional(['is_applicable'],
+                                 lambda ctx: ctx['is_applicable'] and ctx['is_applicable'].value == 1,
                                  lambda ctx: ctx['is_applicable'],
                                  node_mapping, empty_node_mapping)),
 ])

pygridgain/cache.py

@@ -283,6 +283,10 @@ def get_best_node(
                         parts += len(p)
 
                 self.affinity['number_of_partitions'] = parts
+
+                for conn in self.client._nodes:
+                    if not conn.alive:
+                        conn.reconnect()
             else:
                 # get number of partitions
                 parts = self.affinity.get('number_of_partitions')

pygridgain/client.py

@@ -182,15 +182,9 @@ def connect(self, *args):
                     if not self.partition_aware:
                         # do not try to open more nodes
                         self._current_node = i
-                    else:
-                        # take a chance to schedule the reconnection
-                        # for all the failed connections, that was probed
-                        # before this
-                        for failed_node in self._nodes[:i]:
-                            failed_node.reconnect()
 
             except connection_errors:
-                conn._fail()
+                conn.failed = True
                 if self.partition_aware:
                     # schedule the reconnection
                     conn.reconnect()