Add Filecoin (#14)

Author: namikmesic

src/collectors/filecoin.py

@@ -0,0 +1,48 @@
+from helpers import strip_url
+from settings import cfg, logger
+from time import perf_counter
+import requests
+
+
+class filecoin_collector():
+
+    def __init__(self, https_url, provider):
+        self.https_url = https_url
+        self.labels = [
+            'https_url', 'provider', 'blockchain', 'network_name',
+            'network_type'
+        ]
+        self.labels_values = [
+            https_url, provider, cfg.blockchain, cfg.network_name,
+            cfg.network_type
+        ]
+
+    def probe(self, metrics):
+        try:
+            payload = {
+                'jsonrpc': '2.0',
+                'method': "Filecoin.ChainHead",
+                'id': 1
+            }
+            start = perf_counter()
+            response = requests.post(self.https_url, json=payload).json()
+            latency = (perf_counter() - start) * 1000
+
+            if response:
+                metrics['ws_rpc_health'].add_metric(self.labels_values, True)
+                metrics['ws_rpc_latency'].add_metric(self.labels_values,
+                                                     latency)
+                metrics['ws_rpc_block_height'].add_metric(
+                    self.labels_values, response['result']['Height'])
+            else:
+                logger.error("Bad response from client {}: {}".format(
+                    strip_url(self.https_url), exc))
+                metrics['ws_rpc_health'].add_metric(self.labels_values, False)
+        except requests.RequestException as exc:
+            logger.error("Health check failed for {}: {}".format(
+                strip_url(self.https_url), exc))
+            metrics['ws_rpc_health'].add_metric(self.labels_values, False)
+        except Exception as e:
+            logger.error("Health check failed for {}: {}".format(
+                strip_url(self.https_url), e))
+            metrics['ws_rpc_health'].add_metric(self.labels_values, False)

src/exporter.py

@@ -9,6 +9,7 @@
 from collectors.conflux import conflux_collector
 from collectors.bitcoin import bitcoin_collector
 from collectors.dogecoin import doge_collector
+from collectors.filecoin import filecoin_collector
 from settings import logger, cfg
 
 
@@ -30,6 +31,8 @@ def __init__(self):
             self._instantiate_bitcoin()
         if cfg.isDoge():
             self._instantiate_doge()
+        if cfg.isFilecoin():
+            self._instantiate_filecoin()
 
     def _instantiate_evm(self):
         for item in cfg.endpoints:
@@ -82,6 +85,14 @@ def _instantiate_doge(self):
                 doge_collector(item['https_url'], item['provider']))
             self.labels = self.collectors[0].labels
 
+    def _instantiate_filecoin(self):
+        for item in cfg.endpoints:
+            logger.info("Initializing filecoin node {}".format(
+                strip_url(item['https_url'])))
+            self.collectors.append(
+                filecoin_collector(item['https_url'], item['provider']))
+            self.labels = self.collectors[0].labels
+
     def collect(self):
         metrics = {
             "ws_rpc_health":

src/settings.py

@@ -18,7 +18,7 @@ def __init__(self, config_file_path: str, validation_file_path: str):
         self.configuration = self._load_configuration_file(config_file_path)
         self.blockchain = self.configuration['blockchain']
         # Load chain_id only if evm compatible collector
-        if self.configuration['collector'] not in ['cardano', 'solana', 'bitcoin', 'doge']:
+        if self.configuration['collector'] not in ['cardano', 'solana', 'bitcoin', 'doge' , 'filecoin']:
             try:
                 self.chain_id = self.configuration['chain_id']
             except KeyError:
@@ -82,6 +82,9 @@ def isBitcoin(self) -> bool:
     def isDoge(self) -> bool:
         return self.configuration['collector'] == "doge"
 
+    def isFilecoin(self) -> bool:
+        return self.configuration['collector'] == "filecoin"
+
     def _load_configuration_file(self, path):
         logger.info('Loading {}'.format(path))
         configuration_file_schema = Schema({
@@ -94,7 +97,7 @@ def _load_configuration_file(self, path):
             'network_type':
             And(str, lambda s: s in ('Testnet', 'Mainnet')),
             'collector':
-            And(str, lambda s: s in ('evm', 'cardano', 'conflux', 'solana', 'bitcoin', 'doge')),
+            And(str, lambda s: s in ('evm', 'cardano', 'conflux', 'solana', 'bitcoin', 'doge', 'filecoin')),
             Optional('connection_parameters'): {
                 Optional('open_timeout'): And(int),
                 Optional('close_timeout'): And(int),