bin.Databases.KvrocksDB

  1from Databases.database import Database
  2import redis
  3
  4
  5class KvrocksDB(Database):
  6    """
  7    This database store data as live
  8    --> When a withdraw is received, concerned prefixes are removed
  9    """
 10
 11    name = "kvrocks"
 12
 13    def __init__(self, config):
 14        super().__init__()
 15        self.client = redis.Redis(
 16            host=config["host"], port=config["port"], db=config["db"]
 17        )
 18
 19    def start(self):
 20        print(f"Database size : {self.client.dbsize()}")
 21
 22    def stop(self):
 23        pass
 24
 25    ###############
 26    #   INSERTS   #
 27    ###############
 28
 29    def save(self, record):
 30        """Store data in a sorted set named "bgp" with a scorebased on Time
 31
 32        Format :
 33            bgp-prefix:path:source time time:type:peer_asn:collector:country_code
 34
 35        Args:
 36            record (`BGPElem`): BGP Element to save
 37        """
 38        e = record
 39
 40        if e["type"] == "A":
 41            self.client.sadd(
 42                f"prefixes-{e['source']}", e["prefix"]
 43            )  # prefixes-{ASN} : (cidr, cidr)
 44            self.client.sadd(
 45                f"prefixes-{e['country_code']}", e["prefix"]
 46            )  # prefixes-{LU} : (cidr, cidr)
 47            self.client.sadd(
 48                f"as-{e['prefix']}", e["source"]
 49            )  # as-{cidr} : (as, as, as)
 50            self.client.sadd(
 51                f"countries-{e['prefix'] or ''}", e["country_code"]
 52            )  # countries-{cidr} : (LU, FR)
 53            self.client.sadd(
 54                f"paths-{e['prefix']}", e.get("as-path")
 55            )  # paths-{cidr} : (path, path, path)
 56
 57            self.client.zadd(
 58                "bgp-{}:{}:{}".format(e["prefix"], e.get("as-path"), e["source"]),
 59                {
 60                    f"{e['time']}:{e['type']}:{e['peer_asn']}:"
 61                    f"{e['collector']}:{e['country_code']}": int(float(e["time"]))
 62                },
 63            )
 64        #            self.client.execute()
 65
 66        elif e["type"] == "W":
 67            return
 68            for as_source in self.client.smembers(f"as-{e['prefix']}"):
 69                self.client.srem(
 70                    f"prefixes-{as_source}", e["prefix"]
 71                )  # pr AS {cidr, cidr}
 72
 73            for pr in self.client.smembers(f"country-{e['prefix']}"):
 74                self.client.srem(
 75                    f"prefixes-{e['country_code']}", pr
 76                )  # pr LU {cidr, cidr}
 77
 78            self.client.delete(f"countries-{e['prefix']}")  # countries-cidr {LU, FR}
 79            self.client.delete(f"as-{e['prefix']}")  # as-cidr {as, as, as }
 80            self.client.delete(f"paths-{e['prefix']}")  # paths-cidr {path, path, path}
 81
 82            self.client.zadd(
 83                "bgp-{}".format(e["prefix"]),
 84                {
 85                    f"{e['time']}:{e['type']}:{e['peer_asn']}:{e['collector']}": int(
 86                        float(e["time"])
 87                    )
 88                },
 89            )
 90            self.client.zadd(
 91                "bgp-{}:{}".format(e["prefix"], e.get("as-path")),
 92                {
 93                    f"{e['time']}:{e['type']}:{e['peer_asn']}:"
 94                    f"{e['collector']}": int(float(e["time"]))
 95                },
 96            )
 97
 98    ##############
 99    #   GETTER   #
100    ##############
101
102    def get(
103        self,
104        time_start,
105        time_end,
106        record_type=None,
107        peer_asn=None,
108        collectors=None,
109        countries=None,
110        as_numbers=None,
111        prefixes=None,
112        as_paths=None,
113    ):
114        """Can't be tested now"""
115        pass