更新

2026-10-09 14:13:57 +08:00
parent 51d8fd409e
commit 5706e9a6a5
15 changed files with 732 additions and 162 deletions
@@ -0,0 +1,73 @@
+<?php
+
+declare(strict_types=1);
+
+require dirname(__DIR__) . '/vendor/autoload.php';
+
+use app\adminapi\service\AssistantSseTiming;
+
+function timingExpect(bool $condition, string $message): void
+{
+    if (!$condition) {
+        fwrite(STDERR, "FAIL: {$message}\n");
+        exit(1);
+    }
+}
+
+$one = new AssistantSseTiming();
+$two = new AssistantSseTiming();
+$first = $one->event('prepare', [
+    'duration_ms' => 120,
+    'aggregate_ms' => 75,
+    'compaction_ms' => 0,
+    'source_bytes' => 1000,
+    'query_bytes' => 1100,
+    'attachment_count' => 2,
+    'compaction_attempted' => false,
+    'status' => 'ok',
+    'patient_name' => '患者姓名',
+    'query' => '患者正文',
+    'url' => 'https://private.example/path',
+    'api_key' => 'secret',
+]);
+$second = $one->event('upstream_done', [
+    'duration_ms' => 4300,
+    'retry_ms' => 1200,
+    'retry_attempted' => true,
+    'delta_count' => 4,
+    'delta_bytes' => 24,
+    'status' => 'error',
+    'error_code' => 'UPSTREAM_TIMEOUT',
+]);
+
+timingExpect(preg_match('/^[a-f0-9]{32}$/', $first['trace_id']) === 1, 'trace id is random hex only');
+timingExpect($first['trace_id'] !== $two->event('prepare')['trace_id'], 'requests use distinct trace ids');
+timingExpect($first['trace_id'] === $second['trace_id'], 'all phases share one trace id');
+timingExpect(
+    $first['aggregate_ms'] === 75 && $first['attachment_count'] === 2 && $first['compaction_attempted'] === false,
+    'prepare phase retains useful numeric metadata'
+);
+timingExpect(
+    $second['retry_attempted'] === true && $second['retry_ms'] === 1200
+        && $second['error_code'] === 'UPSTREAM_TIMEOUT',
+    'upstream phase retains retry, duration and safe status'
+);
+$encoded = json_encode($first, JSON_UNESCAPED_UNICODE);
+foreach (['患者姓名', '患者正文', 'private.example', 'secret'] as $sensitive) {
+    timingExpect(!str_contains($encoded, $sensitive), 'timing log excludes sensitive payload');
+}
+timingExpect(!isset($one->event('prepare', ['duration_ms' => '患者正文'])['duration_ms']), 'numeric fields reject text');
+timingExpect(!isset($one->event('upstream_done', ['error_code' => 'KEY=secret'])['error_code']), 'error code is validated');
+timingExpect(!isset($one->event('upstream_done', ['error_code' => 'API_KEY_SECRET'])['error_code']), 'unknown machine codes are not logged');
+
+$startedAt = hrtime(true);
+timingExpect(AssistantSseTiming::elapsedMs($startedAt) >= 0, 'monotonic elapsed time is non-negative');
+
+$controller = file_get_contents(dirname(__DIR__) . '/app/adminapi/controller/tcm/DiagnosisController.php');
+$logic = file_get_contents(dirname(__DIR__) . '/app/adminapi/logic/tcm/DiagnosisAiLogic.php');
+timingExpect(is_string($controller) && is_string($logic), 'instrumented source is readable');
+timingExpect(str_contains($controller, "log('first_delta'") && str_contains($controller, "log('sse_done'"), 'SSE first delta and completion are timed');
+timingExpect(str_contains($logic, "log('prepare'") && str_contains($logic, "log('upstream_done'"), 'prepare and upstream are timed');
+timingExpect(str_contains($logic, 'compaction_attempted') && str_contains($logic, 'retry_attempted'), 'compression and compatibility retry are recorded');
+
+echo "Diagnosis AI assistant timing: OK\n";