oib/aitbc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Based on the repository's commit message style and the changes in the diff, here's an appropriate commit message:

Browse Source 
```
feat: add websocket tests, PoA metrics, marketplace endpoints, and enhanced observability

- Add comprehensive websocket tests for blocks and transactions streams including multi-subscriber and high-volume scenarios
- Extend PoA consensus with per-proposer block metrics and rotation tracking
- Add latest block interval gauge and RPC error spike alerting
- Enhance mock coordinator
This commit is contained in:
oib
2025-12-22 07:55:09 +01:00
parent fb60505cdf
commit d98b2c7772
70 changed files with 3472 additions and 246 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
apps/blockchain-node/scripts/mock_coordinator.py
View File

@ -3,9 +3,11 @@
from __future__ import annotations
import asyncio
import contextlib
import random
import time
from typing import Dict
from collections import deque
from typing import Deque, Dict, List
from fastapi import FastAPI
from fastapi.responses import PlainTextResponse
@ -14,21 +16,78 @@ from aitbc_chain.metrics import metrics_registry
app = FastAPI(title="Mock Coordinator API", version="0.1.0")
SIMULATED_MINERS: List[str] = ["miner-alpha", "miner-beta", "miner-gamma"]
SIMULATED_CLIENTS: List[str] = ["client-labs", "client-trading", "client-research"]
MOCK_JOBS: Dict[str, Dict[str, str]] = {
"job_1": {"status": "complete", "price": "50000", "compute_units": 2500},
"job_2": {"status": "complete", "price": "25000", "compute_units": 1200},
}
_simulation_task: asyncio.Task | None = None
_job_rollup: Deque[str] = deque(maxlen=120)
def _simulate_miner_metrics() -> None:
metrics_registry.set_gauge("miner_active_jobs", float(random.randint(0, 5)))
active_jobs = random.randint(1, 6)
metrics_registry.set_gauge("miner_active_jobs", float(active_jobs))
metrics_registry.set_gauge("miner_error_rate", float(random.randint(0, 1)))
metrics_registry.observe("miner_job_duration_seconds", random.uniform(1.0, 5.0))
metrics_registry.observe("miner_job_duration_seconds", random.uniform(1.5, 8.0))
metrics_registry.observe("miner_queue_depth", float(random.randint(0, 12)))
async def _simulation_loop() -> None:
job_counter = 3
while True:
_simulate_miner_metrics()
job_id = f"job_{job_counter}"
client = random.choice(SIMULATED_CLIENTS)
miner = random.choice(SIMULATED_MINERS)
price = random.randint(15_000, 75_000)
compute_units = random.randint(750, 5000)
MOCK_JOBS[job_id] = {
"status": random.choice(["complete", "pending", "failed"]),
"price": str(price),
"compute_units": compute_units,
"client": client,
"assigned_miner": miner,
}
_job_rollup.append(job_id)
if len(MOCK_JOBS) > _job_rollup.maxlen:
oldest = _job_rollup.popleft()
MOCK_JOBS.pop(oldest, None)
metrics_registry.increment("coordinator_jobs_submitted_total")
metrics_registry.observe("coordinator_job_price", float(price))
metrics_registry.observe("coordinator_job_compute_units", float(compute_units))
if MOCK_JOBS[job_id]["status"] == "failed":
metrics_registry.increment("coordinator_jobs_failed_total")
else:
metrics_registry.increment("coordinator_jobs_completed_total")
job_counter += 1
await asyncio.sleep(random.uniform(1.5, 3.5))
@app.on_event("startup")
async def _startup() -> None:
global _simulation_task
_simulate_miner_metrics()
_simulation_task = asyncio.create_task(_simulation_loop())
@app.on_event("shutdown")
async def _shutdown() -> None:
global _simulation_task
if _simulation_task:
_simulation_task.cancel()
with contextlib.suppress(asyncio.CancelledError):
await _simulation_task
_simulation_task = None
@app.get("/health")