96 lines
4.7 KiB
PHP
96 lines
4.7 KiB
PHP
<?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');
|
|
timingExpect(isset($first['request_elapsed_ms']) && is_int($first['request_elapsed_ms']), 'request timing includes the phase before SSE');
|
|
timingExpect($one->event('first_delta')['request_elapsed_ms'] >= $first['request_elapsed_ms'], 'full request time remains monotonic across preparation and first text');
|
|
timingExpect($one->event('prepare', ['compaction_cache_hit' => true])['compaction_cache_hit'] === true, 'cache hit is safe boolean telemetry');
|
|
|
|
// Exercise a buffered web-style channel: timing must reach disk before HTTP end/exit.
|
|
$app = new \think\App(dirname(__DIR__) . '/');
|
|
$directory = sys_get_temp_dir() . '/zyt-ai-timing-' . bin2hex(random_bytes(8));
|
|
$app->config->set(['default' => 'timing', 'channels' => ['timing' => [
|
|
'type' => 'File', 'path' => $directory, 'single' => 'timing', 'realtime_write' => false,
|
|
]]], 'log');
|
|
$logger = $app->make('log');
|
|
$channel = $logger->channel('timing');
|
|
(new ReflectionProperty(\think\log\Channel::class, 'lazy'))->setValue($channel, true);
|
|
$logger->info('buffered control');
|
|
$logFile = $directory . '/timing.log';
|
|
timingExpect(!is_file($logFile), 'control channel really buffers before the SSE timing write');
|
|
$one->log('first_delta', ['duration_ms' => 123, 'api_key' => 'must-not-be-logged']);
|
|
timingExpect(is_file($logFile), 'SSE timing persists immediately without HTTP-end save');
|
|
$logged = file_get_contents($logFile);
|
|
timingExpect(str_contains($logged, $one->id()) && !str_contains($logged, 'must-not-be-logged'), 'persisted timing remains patient and credential free');
|
|
unlink($logFile);
|
|
rmdir($directory);
|
|
|
|
$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";
|