implement dummy uuid comparison

Author: c2bo

src/bench.py

@@ -0,0 +1,71 @@
+import math
+import zlib
+from random import choices
+from uuid import uuid4
+
+from py_markdown_table.markdown_table import markdown_table
+
+from status_list import StatusList
+
+
+def display_size(size_bytes):
+    if size_bytes == 0:
+        return "0B"
+    size_name = ("B", "KB", "MB", "GB")
+    floored = int(math.floor(math.log(size_bytes, 1024)))
+    p = math.pow(1024, floored)
+    rounded = round(size_bytes / p, 1)
+    return "%s %s" % (rounded, size_name[floored])
+
+
+sizes = [100000, 1000000, 10000000, 100000000]
+rev_rates = [0.0001, 0.001, 0.01, 0.02, 0.05, 0.1, 0.25, 0.5, 0.75, 1]
+
+data_sl = []
+data_uuid = []
+
+for size in sizes:
+    newdata_sl = {"size": size}
+    newdata_uuid = {"size": size}
+    for rev_rate in rev_rates:
+        print(f"Revocation Rate: {rev_rate}")
+        vals = [0, 1]
+        p = [1 - rev_rate, rev_rate]
+        sample = choices(population=vals, weights=p, k=size)
+
+        statuslist = StatusList(size, 1)
+        idlist = bytearray()
+        for idx, val in enumerate(sample):
+            statuslist.set(idx, val)
+            if val == 1:
+                idlist.extend(uuid4().bytes)
+
+        rawsl = statuslist.encodeAsBytes()
+        rawidlist = zlib.compress(idlist, level=9)
+
+        percentage = str(rev_rate * 100) + "%"
+        newdata_sl[percentage] = display_size(len(rawsl))
+        newdata_uuid[percentage] = display_size(len(rawidlist))
+        print(f"Size in Bytes: {display_size(len(rawsl))}")
+        print(f"Size in Bytes uuid: {display_size(len(rawidlist))}")
+    data_sl.append(newdata_sl)
+    data_uuid.append(newdata_uuid)
+
+markdown_sl = (
+    markdown_table(data_sl)
+    .set_params(
+        padding_width=3,
+        padding_weight="centerleft",
+    )
+    .get_markdown()
+)
+print(markdown_sl)
+markdown_uuid = (
+    markdown_table(data_uuid)
+    .set_params(
+        padding_width=3,
+        padding_weight="centerleft",
+    )
+    .get_markdown()
+)
+print(markdown_uuid)

src/requirements.txt

@@ -4,3 +4,4 @@ git+https://github.com/wbond/oscrypto.git@1547f535001ba568b239b8797465536759c742
 jwcrypto==1.5.6
 cbor2==5.6.2
 cwt==2.7.4
+py_markdown_table==1.3.0

src/status_list.py

@@ -1,9 +1,11 @@
+import math
 import zlib
 from base64 import urlsafe_b64decode, urlsafe_b64encode
 from typing import Dict
 
 from cbor2 import dumps, loads
 
+LEVEL=9
 
 class StatusList:
     list: bytearray
@@ -13,7 +15,7 @@ class StatusList:
 
     def __init__(self, size: int, bits: int):
         self.divisor = 8 // bits
-        self.list = bytearray([0] * (size // self.divisor))
+        self.list = bytearray([0] * math.ceil(size / self.divisor))
         self.bits = bits
         self.size = size
 
@@ -24,11 +26,11 @@ def fromEncoded(cls, encoded: str, bits: int = 1):
         return new
 
     def encodeAsString(self) -> str:
-        zipped = zlib.compress(self.list, level=9)
+        zipped = zlib.compress(self.list, level=LEVEL)
         return urlsafe_b64encode(zipped).decode().strip("=")
 
     def encodeAsBytes(self) -> bytes:
-        return zlib.compress(self.list, level=9)
+        return zlib.compress(self.list, level=LEVEL)
 
     def encodeAsJSON(self) -> Dict:
         encoded_list = self.encodeAsString()