doublezero-telemetry 0.0.2__tar.gz → 0.0.3__tar.gz

{doublezero_telemetry-0.0.2 → doublezero_telemetry-0.0.3}/PKG-INFO

@@ -1,9 +1,10 @@
 Metadata-Version: 2.4
 Name: doublezero-telemetry
-Version: 0.0.2
+Version: 0.0.3
 Summary: DoubleZero Telemetry SDK
 Requires-Python: >=3.10
 Requires-Dist: borsh-incremental
+Requires-Dist: doublezero-serviceability
 Requires-Dist: httpx>=0.27
 Requires-Dist: solana>=0.35
 Requires-Dist: solders>=0.21

doublezero_telemetry-0.0.3/examples/fetch.py

@@ -0,0 +1,123 @@
+#!/usr/bin/env python3
+"""Example CLI that fetches and displays telemetry data."""
+
+import argparse
+
+from solders.pubkey import Pubkey  # type: ignore[import-untyped]
+
+from serviceability.client import Client as ServiceabilityClient
+from telemetry.client import Client as TelemetryClient
+from telemetry.config import LEDGER_RPC_URLS
+from telemetry.rpc import new_rpc_client
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="Fetch telemetry data")
+    parser.add_argument(
+        "--env",
+        default="mainnet-beta",
+        choices=["mainnet-beta", "testnet", "devnet", "localnet"],
+        help="Environment to connect to",
+    )
+    parser.add_argument(
+        "--epoch",
+        type=int,
+        default=0,
+        help="Epoch to fetch samples for (0 = current epoch)",
+    )
+    args = parser.parse_args()
+
+    print(f"Fetching telemetry data from {args.env}...\n")
+
+    # First, get serviceability data to discover devices and links
+    svc_client = ServiceabilityClient.from_env(args.env)
+    svc_data = svc_client.get_program_data()
+
+    print("=== Network Overview ===")
+    print(f"Devices: {len(svc_data.devices)}")
+    print(f"Links:   {len(svc_data.links)}")
+    print()
+
+    if not svc_data.links:
+        print("No links found - no telemetry data to fetch.")
+        return
+
+    # Build device code map for display
+    device_codes: dict[str, str] = {}
+    device_pks: dict[str, Pubkey] = {}
+    for dev in svc_data.devices:
+        pk_str = str(dev.owner)  # Using owner as pubkey proxy
+        device_codes[pk_str] = dev.code
+        device_pks[dev.code] = dev.owner
+
+    # Create telemetry client
+    tel_client = TelemetryClient.from_env(args.env)
+
+    # Determine which epoch to use
+    target_epoch = args.epoch
+    if target_epoch == 0:
+        # Get current epoch from RPC
+        rpc = new_rpc_client(LEDGER_RPC_URLS[args.env])
+        epoch_info = rpc.get_epoch_info()
+        target_epoch = epoch_info.value.epoch
+
+    print(f"=== Device Latency Samples (epoch {target_epoch}) ===")
+
+    samples_found = 0
+    for link in svc_data.links:
+        side_a_pk = link.side_a_pub_key
+        side_z_pk = link.side_z_pub_key
+        link_pk = link.owner  # Using owner as link pubkey proxy
+
+        # Find device codes
+        side_a_code = "unknown"
+        side_z_code = "unknown"
+        for dev in svc_data.devices:
+            if str(dev.owner) == str(side_a_pk):
+                side_a_code = dev.code
+            if str(dev.owner) == str(side_z_pk):
+                side_z_code = dev.code
+
+        # Try both directions
+        for origin_pk, target_pk, o_code, t_code in [
+            (side_a_pk, side_z_pk, side_a_code, side_z_code),
+            (side_z_pk, side_a_pk, side_z_code, side_a_code),
+        ]:
+            try:
+                samples = tel_client.get_device_latency_samples(
+                    origin_pk, target_pk, link_pk, target_epoch
+                )
+            except Exception:
+                # Account likely doesn't exist for this epoch
+                continue
+
+            samples_found += 1
+            sample_count = len(samples.samples)
+
+            if sample_count == 0:
+                print(f"  {o_code} -> {t_code} ({link.code}): initialized, no samples yet")
+                continue
+
+            # Calculate stats
+            total = sum(samples.samples)
+            min_val = min(samples.samples)
+            max_val = max(samples.samples)
+            avg_us = total / sample_count
+            avg_ms = avg_us / 1000.0
+            min_ms = min_val / 1000.0
+            max_ms = max_val / 1000.0
+
+            print(
+                f"  {o_code} -> {t_code} ({link.code}): {sample_count} samples, "
+                f"avg {avg_ms:.2f}ms, min {min_ms:.2f}ms, max {max_ms:.2f}ms"
+            )
+
+    if samples_found == 0:
+        print(f"  No samples found for epoch {target_epoch}. Try a different epoch with --epoch flag.")
+
+    print()
+    print("Done.")
+
+
+if __name__ == "__main__":
+    main()

{doublezero_telemetry-0.0.2 → doublezero_telemetry-0.0.3}/pyproject.toml

@@ -1,10 +1,11 @@
 [project]
 name = "doublezero-telemetry"
-version = "0.0.2"
+version = "0.0.3"
 description = "DoubleZero Telemetry SDK"
 requires-python = ">=3.10"
 dependencies = [
     "borsh-incremental",
+    "doublezero-serviceability",
     "solana>=0.35",
     "solders>=0.21",
     "httpx>=0.27",
@@ -27,3 +28,4 @@ dev = [
 
 [tool.uv.sources]
 borsh-incremental = { path = "../../borsh-incremental/python", editable = true }
+doublezero-serviceability = { path = "../../serviceability/python", editable = true }

{doublezero_telemetry-0.0.2 → doublezero_telemetry-0.0.3}/telemetry/state.py

@@ -9,7 +9,7 @@ from __future__ import annotations
 
 from dataclasses import dataclass, field
 
-from borsh_incremental import IncrementalReader
+from borsh_incremental import DefensiveReader
 from solders.pubkey import Pubkey  # type: ignore[import-untyped]
 
 
@@ -23,7 +23,7 @@ MAX_INTERNET_LATENCY_SAMPLES_PER_ACCOUNT = 3_000
 DEVICE_LATENCY_HEADER_SIZE = 1 + 8 + 32 * 6 + 8 + 8 + 4 + 128
 
 
-def _read_pubkey(r: IncrementalReader) -> Pubkey:
+def _read_pubkey(r: DefensiveReader) -> Pubkey:
     return Pubkey.from_bytes(r.read_pubkey_raw())
 
 
@@ -49,7 +49,7 @@ class DeviceLatencySamples:
                 f"data too short for device latency header: {len(data)} < {DEVICE_LATENCY_HEADER_SIZE}"
             )
 
-        r = IncrementalReader(data)
+        r = DefensiveReader(data)
 
         account_type = r.read_u8()
         epoch = r.read_u64()
@@ -106,7 +106,7 @@ class InternetLatencySamples:
         if len(data) < 10:
             raise ValueError("data too short")
 
-        r = IncrementalReader(data)
+        r = DefensiveReader(data)
 
         account_type = r.read_u8()
         epoch = r.read_u64()

{doublezero_telemetry-0.0.2 → doublezero_telemetry-0.0.3}/uv.lock

@@ -66,12 +66,35 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/8c/0c/2db6f7e1ae9795e436c6a0dc0bc38b12b8c8a228cb63203e24190b755b3b/construct_typing-0.7.0-py3-none-any.whl", hash = "sha256:c92383c6e8e5d07ba25811c8d5163820458d821e73bb1006541f43f89788646c", size = 24350, upload-time = "2025-10-27T19:30:27.505Z" },
 ]
 
+[[package]]
+name = "doublezero-serviceability"
+version = "0.0.2"
+source = { editable = "../../serviceability/python" }
+dependencies = [
+    { name = "borsh-incremental" },
+    { name = "httpx" },
+    { name = "solana" },
+    { name = "solders" },
+]
+
+[package.metadata]
+requires-dist = [
+    { name = "borsh-incremental", editable = "../../borsh-incremental/python" },
+    { name = "httpx", specifier = ">=0.27" },
+    { name = "solana", specifier = ">=0.35" },
+    { name = "solders", specifier = ">=0.21" },
+]
+
+[package.metadata.requires-dev]
+dev = [{ name = "pytest", specifier = ">=9.0.2" }]
+
 [[package]]
 name = "doublezero-telemetry"
-version = "0.0.1"
+version = "0.0.2"
 source = { editable = "." }
 dependencies = [
     { name = "borsh-incremental" },
+    { name = "doublezero-serviceability" },
     { name = "httpx" },
     { name = "solana" },
     { name = "solders" },
@@ -85,6 +108,7 @@ dev = [
 [package.metadata]
 requires-dist = [
     { name = "borsh-incremental", editable = "../../borsh-incremental/python" },
+    { name = "doublezero-serviceability", editable = "../../serviceability/python" },
     { name = "httpx", specifier = ">=0.27" },
     { name = "solana", specifier = ">=0.35" },
     { name = "solders", specifier = ">=0.21" },