poller/routes: added the vrf name management for sonic version 4.2

Added the code to manage correctly those routing table entries
that have the vrf id instead of the vrf name

Signed-off-by: AndryNick98 <[email protected]>

Author: AndryNick98

suzieq/poller/worker/services/routes.py

@@ -1,4 +1,5 @@
 import re
+from typing import Dict, List
 import numpy as np
 
 from suzieq.poller.worker.services.service import Service
@@ -56,9 +57,18 @@ def _clean_eos_data(self, processed_data, _):
 
     def _clean_linux_data(self, processed_data, _):
         """Clean Linux ip route data"""
+        drop_indices: List[int] = []
+        id_vrf_match: Dict[str, str] = {}
 
-        for entry in processed_data:
-            entry["vrf"] = entry["vrf"] or "default"
+        for i, entry in enumerate(processed_data):
+            if table_id := entry.get('table_id'):
+                id_vrf_match[table_id] = entry.get('vrf') or 'default'
+                drop_indices.append(i)
+                continue
+            if vrf_id := entry.get('vrf_id'):
+                entry["vrf"] = id_vrf_match.get(vrf_id) or 'default'
+            else:
+                entry["vrf"] = entry.get("vrf") or "default"
             entry["metric"] = entry["metric"] or 20
             entry['preference'] = entry['metric']
             for ele in ["nexthopIps", "oifs"]:
@@ -85,6 +95,10 @@ def _clean_linux_data(self, processed_data, _):
 
             entry['inHardware'] = True  # Till the offload flag is here
 
+        if drop_indices:
+            processed_data = np.delete(
+                processed_data, drop_indices).tolist()   # type: ignore
+
         return processed_data
 
     def _clean_cumulus_data(self, processed_data, raw_data):