Update files

2026-07-29 12:01:31 +00:00 · 2026-02-08 15:59:38 +07:00
parent b4a93955e3
commit c5a27a4ddc
8 changed files with 736 additions and 329 deletions
--- a/scripts/measure-relay-latency.ts
+++ b/scripts/measure-relay-latency.ts
@@ -0,0 +1,142 @@
+import { DaemonClient } from "../packages/server/src/client/daemon-client.js";
+import { buildRelayWebSocketUrl } from "../packages/server/src/shared/daemon-endpoints.js";
+import { buildDaemonWebSocketUrl } from "../packages/server/src/shared/daemon-endpoints.js";
+
+const OFFER = {
+  serverId: "srv_ETXtcjYRGrCI",
+  daemonPublicKeyB64: "12yCG8sqNumkwHMOQyRM/vMXfPc6nb430pj27sfARBc=",
+  relay: { endpoint: "relay.paseo.sh:443" },
+};
+
+const DIRECT_ENDPOINT = "localhost:6767";
+const PING_COUNT = 20;
+const WARMUP_COUNT = 3;
+
+async function connectClient(
+  label: string,
+  config: ConstructorParameters<typeof DaemonClient>[0],
+): Promise<DaemonClient> {
+  const client = new DaemonClient(config);
+
+  await new Promise<void>((resolve, reject) => {
+    const timeout = setTimeout(() => reject(new Error(`${label}: connect timeout`)), 15_000);
+    const unsub = client.on("status", (msg) => {
+      if (msg.type === "status") {
+        clearTimeout(timeout);
+        unsub();
+        resolve();
+      }
+    });
+    client.connect().catch(reject);
+  });
+
+  return client;
+}
+
+async function measurePings(
+  label: string,
+  client: DaemonClient,
+  count: number,
+  warmup: number,
+): Promise<void> {
+  // Warmup
+  for (let i = 0; i < warmup; i++) {
+    await client.ping({ timeoutMs: 10_000 });
+  }
+
+  const results: number[] = [];
+  const serverTimings: { serverReceivedAt: number; serverSentAt: number; clientSentAt: number }[] = [];
+
+  for (let i = 0; i < count; i++) {
+    const result = await client.ping({ timeoutMs: 10_000 });
+    results.push(result.rttMs);
+    serverTimings.push({
+      serverReceivedAt: result.serverReceivedAt,
+      serverSentAt: result.serverSentAt,
+      clientSentAt: result.clientSentAt,
+    });
+    // Small delay between pings to avoid batching effects
+    await new Promise((r) => setTimeout(r, 100));
+  }
+
+  const sorted = [...results].sort((a, b) => a - b);
+  const avg = results.reduce((a, b) => a + b, 0) / results.length;
+  const min = sorted[0]!;
+  const max = sorted[sorted.length - 1]!;
+  const p50 = sorted[Math.floor(sorted.length * 0.5)]!;
+  const p95 = sorted[Math.floor(sorted.length * 0.95)]!;
+
+  console.log(`\n${label}:`);
+  console.log(`  samples: ${count} (after ${warmup} warmup)`);
+  console.log(`  min: ${min}ms`);
+  console.log(`  max: ${max}ms`);
+  console.log(`  avg: ${avg.toFixed(1)}ms`);
+  console.log(`  p50: ${p50}ms`);
+  console.log(`  p95: ${p95}ms`);
+  console.log(`  all: [${results.join(", ")}]`);
+
+  // Server-side processing time (serverSentAt - serverReceivedAt)
+  const serverProcessing = serverTimings.map((t) => t.serverSentAt - t.serverReceivedAt);
+  const avgServerProcessing = serverProcessing.reduce((a, b) => a + b, 0) / serverProcessing.length;
+  console.log(`  server processing avg: ${avgServerProcessing.toFixed(1)}ms`);
+  console.log(`  server processing: [${serverProcessing.join(", ")}]`);
+}
+
+async function main() {
+  console.log("=== Relay Latency Measurement ===\n");
+
+  // Measure direct connection
+  console.log("Connecting direct...");
+  const directClient = await connectClient("Direct", {
+    url: buildDaemonWebSocketUrl(DIRECT_ENDPOINT),
+  });
+
+  await measurePings("Direct (localhost:6767)", directClient, PING_COUNT, WARMUP_COUNT);
+
+  // Measure relay connection
+  console.log("\nConnecting via relay...");
+  const relayUrl = buildRelayWebSocketUrl({
+    endpoint: OFFER.relay.endpoint,
+    serverId: OFFER.serverId,
+    role: "client",
+  });
+
+  const relayClient = await connectClient("Relay", {
+    url: relayUrl,
+    e2ee: {
+      enabled: true,
+      daemonPublicKeyB64: OFFER.daemonPublicKeyB64,
+    },
+  });
+
+  await measurePings("Relay (relay.paseo.sh:443)", relayClient, PING_COUNT, WARMUP_COUNT);
+
+  // Measure raw WebSocket to relay (no E2EE, no daemon, just WS open+close timing)
+  console.log("\nMeasuring raw WebSocket connect time to relay...");
+  const wsConnectTimes: number[] = [];
+  for (let i = 0; i < 5; i++) {
+    const start = Date.now();
+    const { WebSocket } = await import("ws");
+    const ws = new WebSocket(`wss://relay.paseo.sh/ws?serverId=latency_probe_${Date.now()}&role=client&clientId=probe_${i}`);
+    await new Promise<void>((resolve, reject) => {
+      ws.on("open", () => {
+        wsConnectTimes.push(Date.now() - start);
+        ws.close();
+        resolve();
+      });
+      ws.on("error", reject);
+      setTimeout(() => reject(new Error("ws connect timeout")), 5000);
+    });
+  }
+  console.log(`  Raw WS connect times: [${wsConnectTimes.join(", ")}]ms`);
+
+  await directClient.close();
+  await relayClient.close();
+
+  console.log("\nDone.");
+}
+
+main().catch((err) => {
+  console.error(err);
+  process.exit(1);
+});