compare-best-tip.py

#!/usr/bin/env python3

# script to find common best-tip prefix over a list of nodes using GraphQL query

import os
import sys
import json
import click
import subprocess
import requests
import time
import logging
import random

from kubernetes import client, config
from kubernetes.stream import stream


import CodaClient

@click.command()
@click.option('--namespace', default="regeneration", help='Namespace to Query.')
@click.option('--remote-graphql-port', default=3085, help='Remote GraphQL Port to Query.')
def check(namespace, remote_graphql_port):
  '''
  A quick hack script to tunnel into all nodes in a Testnet and query their GraphQL Endpoints. 
  This would be better as a k8s job with direct access, but the GraphQL endpoints are unsafe. 
  '''
  logging.basicConfig(stream=sys.stdout, level=logging.INFO)
  logger = logging.getLogger()

  config.load_kube_config()

  v1 = client.CoreV1Api()
  pods = v1.list_namespaced_pod(namespace)
  items = pods.items
  random.shuffle(items)
  common_prefix = None
  for pod in items:
    if pod.metadata.namespace == namespace and 'block-producer' in pod.metadata.name:
      logger.info("Processing {}".format(pod.metadata.name))
      # Set up Port Forward
      logger.debug("Setting up Port Forward")
      
      command = "kubectl port-forward --namespace {} {} {}".format(pod.metadata.namespace, pod.metadata.name, remote_graphql_port)
      logger.debug("Running Bash Command: {}".format(command))
      proc = subprocess.Popen(["bash", "-c", command],
                            stdout=subprocess.PIPE,
                            stderr=subprocess.STDOUT)

      time.sleep(5)

      coda = CodaClient.Client (graphql_host="localhost", graphql_port=3085)
      result = coda._send_query (query="query bestChainQuery { bestChain { stateHash } }")
      logger.debug("Got Response: {}".format(result))

      chain = list(map(lambda a: a["stateHash"], result['data']['bestChain']))
      logger.info("New Best Chain: {}".format(json.dumps(chain, indent=3)))
      logger.info("Current Common Prefix: {}".format(json.dumps(common_prefix, indent=2)))
      
      if common_prefix == None or common_prefix == []:
        common_prefix = chain
      else:
        common_prefix = os.path.commonprefix([chain, common_prefix])
      
      logger.info("Resulting Common Prefix: {}".format(json.dumps(common_prefix, indent=2)))
      

      proc.terminate()
      try:
        outs, _ = proc.communicate(timeout=0.2)
        #print('== subprocess exited with rc =', proc.returncode)
        #print(outs.decode('utf-8'))
      except subprocess.TimeoutExpired:
        logger.error('subprocess did not terminate in time')
      
  print(common_prefix)
      
      
if __name__ == '__main__':
    check()