Enh: CLI GOSSIP add a ping method to directly test a node

Author: naparuba

data/core-configuration/packs/core-cli-gossip/cli/cli.py

@@ -31,6 +31,7 @@
                      NODE_STATES.LEAVE  : 'cyan',
                      NODE_STATES.UNKNOWN: 'grey',
                      }
+
 NODE_STATE_PREFIXS = {NODE_STATES.ALIVE  : CHARACTERS.check,
                       NODE_STATES.DEAD   : CHARACTERS.cross,
                       NODE_STATES.SUSPECT: CHARACTERS.double_exclamation,
@@ -203,6 +204,45 @@ def do_join(seed=''):
         cprint('FAILED', color='red')
 
 
+def do_ping(node):
+    # The information is available only if the agent is started
+    wait_for_agent_started(visual_wait=True)
+    
+    try:
+        node_obj = get_opsbro_json('/agent/query/guess/%s' % node)
+    except get_request_errors() as exp:
+        logger.error('Cannot query the node: %s' % exp)
+        sys.exit(2)
+    if node_obj is None:
+        cprint('FAILED: cannot find the node %s' % node, color='red')
+        sys.exit(2)
+    node_uuid = node_obj['uuid']
+    
+    try:
+        ping_result = get_opsbro_json('/agent/ping/%s' % node_uuid)
+    except get_request_errors() as exp:
+        logger.error('Cannot launch the node ping: %s' % exp)
+        sys.exit(2)
+    if 'error' in ping_result:
+        cprint('FAILED: %s' % ping_result['error'], color='red')
+        sys.exit(2)
+    node_state = ping_result['state']
+    display_name = node_obj['display_name']
+    if not display_name:
+        display_name = node_obj['name']
+    
+    state_color = NODE_STATE_COLORS.get(node_state)
+    state_char = NODE_STATE_PREFIXS.get(node_state)
+    cprint(' %s ' % state_char, color=state_color, end='')
+    cprint('%-15s ' % display_name, color='magenta', end='')
+    cprint('is: ', end='')
+    cprint(node_state, color=state_color)
+    
+    # If not alive, it's an error
+    if node_state != NODE_STATES.ALIVE:
+        sys.exit(2)
+
+
 def do_zone_change(name=''):
     if not name:
         cprint("Need a zone name")
@@ -628,6 +668,14 @@ def do_wait_members(name='', display_name='', group='', count=1, timeout=30):
         ],
     },
 
+    do_ping             : {
+        'keywords'   : ['gossip', 'ping'],
+        'description': 'Ping another node of the cluster',
+        'args'       : [
+            {'name': 'node', 'default': '', 'description': 'uuid, name or display name of the node to ping'},
+        ],
+    },
+    
     do_leave            : {
         'keywords'   : ['gossip', 'leave'],
         'description': 'Put in leave a cluster node',

opsbro/gossip.py

@@ -1917,6 +1917,26 @@ def get_reload_zone_key(zone_name):
             return jsoner.dumps(encrypter.load_or_reload_key_for_zone_if_need(zone_name))
 
 
+        @http_export('/agent/query/guess/:name_or_uuid', method='GET', protected=True)
+        def get_query_by_name(name_or_uuid):
+            response.content_type = 'application/json'
+            for node in self.nodes.values():
+                if node['uuid'] == name_or_uuid or node['name'] == name_or_uuid or node['display_name'] == name_or_uuid:
+                    return node
+            return None
+        
+        
+        @http_export('/agent/ping/:node_uuid', method='GET', protected=True)
+        def get_ping_node(node_uuid):
+            response.content_type = 'application/json'
+            node = self.get(node_uuid)
+            if node is None:
+                return {'error': 'No such node %s' % node_uuid}
+            self.__do_ping(node)
+            logger.info('NODE STATE: %s' % node['state'])
+            return {'state': node['state']}
+        
+        
         @http_export('/agent/event/:event_type', method='GET', protected=True)
         def get_event(event_type):
             response.content_type = 'application/json'