test(acceptance): 验证厂商三类额度与优先路由
为 provider-burst 固定三平台并发、RPM、TPM 和优先级组合，使用稳定 token 用量的本地视频负载验证同级按容量分流及低优先级溢出。\n\n报告升级为 v2，强制校验三类额度峰值精确命中上限、不越界、队列回收、三 Worker 分发以及重复提交和回调安全。\n\n验证：\n- env -u AI_GATEWAY_TEST_DATABASE_URL go test ./... -count=1\n- go vet ./...\n- bash -n scripts/acceptance/provider-burst.sh scripts/acceptance/run-local-acceptance.sh\n- shellcheck -x scripts/acceptance/provider-burst.sh scripts/acceptance/run-local-acceptance.sh
This commit is contained in:
@@ -44,11 +44,11 @@ WITH source AS (
|
||||
AND COALESCE((platform.config->>'acceptanceSnapshot')::boolean, false)
|
||||
ORDER BY platform.priority, platform.created_at
|
||||
LIMIT 1
|
||||
), variants(platform_id, platform_key, platform_name, concurrency_limit) AS (
|
||||
), variants(platform_id, platform_key, platform_name, concurrency_limit, rpm_limit, tpm_limit, priority) AS (
|
||||
VALUES
|
||||
('${provider_burst_platform_ids[0]}'::uuid, 'acceptance-provider-burst-2', 'Acceptance Provider Burst C2', 2),
|
||||
('${provider_burst_platform_ids[1]}'::uuid, 'acceptance-provider-burst-4', 'Acceptance Provider Burst C4', 4),
|
||||
('${provider_burst_platform_ids[2]}'::uuid, 'acceptance-provider-burst-6', 'Acceptance Provider Burst C6', 6)
|
||||
('${provider_burst_platform_ids[0]}'::uuid, 'acceptance-provider-burst-2', 'Acceptance Provider Burst C2', 2, 4, 28, 100),
|
||||
('${provider_burst_platform_ids[1]}'::uuid, 'acceptance-provider-burst-4', 'Acceptance Provider Burst C4', 4, 8, 56, 100),
|
||||
('${provider_burst_platform_ids[2]}'::uuid, 'acceptance-provider-burst-6', 'Acceptance Provider Burst C6', 6, 12, 84, 200)
|
||||
)
|
||||
INSERT INTO integration_platforms (
|
||||
id, provider, platform_key, name, base_url, auth_type, credentials, config,
|
||||
@@ -61,12 +61,14 @@ SELECT variants.platform_id, source.provider, variants.platform_key, variants.pl
|
||||
source.config || jsonb_build_object(
|
||||
'acceptanceProviderBurst', true,
|
||||
'acceptanceBurstRunId', '$run_id',
|
||||
'acceptanceConcurrencyLimit', variants.concurrency_limit
|
||||
'acceptanceConcurrencyLimit', variants.concurrency_limit,
|
||||
'acceptanceRPMLimit', variants.rpm_limit,
|
||||
'acceptanceTPMLimit', variants.tpm_limit
|
||||
),
|
||||
source.visibility_scope, source.tenant_id, source.tenant_key,
|
||||
source.default_pricing_mode, source.default_discount_factor,
|
||||
source.pricing_rule_set_id, source.retry_policy, '{\"rules\":[]}'::jsonb,
|
||||
100, NULL, 'enabled', NULL, NULL, variants.platform_name
|
||||
variants.priority, NULL, 'enabled', NULL, NULL, variants.platform_name
|
||||
FROM source CROSS JOIN variants
|
||||
ON CONFLICT (id) DO UPDATE SET
|
||||
provider = EXCLUDED.provider,
|
||||
@@ -102,11 +104,11 @@ WITH source AS (
|
||||
AND COALESCE((platform.config->>'acceptanceSnapshot')::boolean, false)
|
||||
ORDER BY platform.priority, platform.created_at
|
||||
LIMIT 1
|
||||
), variants(model_id, platform_id, concurrency_limit) AS (
|
||||
), variants(model_id, platform_id, concurrency_limit, rpm_limit, tpm_limit) AS (
|
||||
VALUES
|
||||
('${provider_burst_model_ids[0]}'::uuid, '${provider_burst_platform_ids[0]}'::uuid, 2),
|
||||
('${provider_burst_model_ids[1]}'::uuid, '${provider_burst_platform_ids[1]}'::uuid, 4),
|
||||
('${provider_burst_model_ids[2]}'::uuid, '${provider_burst_platform_ids[2]}'::uuid, 6)
|
||||
('${provider_burst_model_ids[0]}'::uuid, '${provider_burst_platform_ids[0]}'::uuid, 2, 4, 28),
|
||||
('${provider_burst_model_ids[1]}'::uuid, '${provider_burst_platform_ids[1]}'::uuid, 4, 8, 56),
|
||||
('${provider_burst_model_ids[2]}'::uuid, '${provider_burst_platform_ids[2]}'::uuid, 6, 12, 84)
|
||||
)
|
||||
INSERT INTO platform_models (
|
||||
id, platform_id, base_model_id, model_name, model_alias, model_type,
|
||||
@@ -127,7 +129,8 @@ SELECT variants.model_id, variants.platform_id, source.base_model_id,
|
||||
'metric', 'concurrent', 'limit', variants.concurrency_limit,
|
||||
'leaseTtlSeconds', 120
|
||||
),
|
||||
jsonb_build_object('metric', 'rpm', 'limit', 600, 'windowSeconds', 60)
|
||||
jsonb_build_object('metric', 'rpm', 'limit', variants.rpm_limit, 'windowSeconds', 60),
|
||||
jsonb_build_object('metric', 'tpm_total', 'limit', variants.tpm_limit, 'windowSeconds', 60)
|
||||
)),
|
||||
'override', NULL, '{}'::jsonb, source.provider_model_name, NULL, true
|
||||
FROM source CROSS JOIN variants
|
||||
@@ -219,13 +222,18 @@ DO UPDATE SET status = 'active', metadata = EXCLUDED.metadata,
|
||||
SELECT jsonb_build_object(
|
||||
'platforms', count(*),
|
||||
'limits', jsonb_agg((platform.config->>'acceptanceConcurrencyLimit')::int ORDER BY (platform.config->>'acceptanceConcurrencyLimit')::int),
|
||||
'rpmLimits', jsonb_agg((platform.config->>'acceptanceRPMLimit')::int ORDER BY (platform.config->>'acceptanceConcurrencyLimit')::int),
|
||||
'tpmLimits', jsonb_agg((platform.config->>'acceptanceTPMLimit')::int ORDER BY (platform.config->>'acceptanceConcurrencyLimit')::int),
|
||||
'priorities', jsonb_agg(platform.priority ORDER BY (platform.config->>'acceptanceConcurrencyLimit')::int),
|
||||
'enabled', count(*) FILTER (WHERE platform.status='enabled' AND model.enabled)
|
||||
)
|
||||
FROM integration_platforms platform
|
||||
JOIN platform_models model ON model.platform_id = platform.id
|
||||
WHERE COALESCE((platform.config->>'acceptanceProviderBurst')::boolean, false)
|
||||
AND platform.config->>'acceptanceBurstRunId' = '$run_id';")
|
||||
jq -e '.platforms == 3 and .enabled == 3 and .limits == [2,4,6]' <<<"$configured" >/dev/null
|
||||
jq -e '.platforms == 3 and .enabled == 3 and .limits == [2,4,6] and
|
||||
.rpmLimits == [4,8,12] and .tpmLimits == [28,56,84] and .priorities == [100,100,200]' \
|
||||
<<<"$configured" >/dev/null
|
||||
}
|
||||
|
||||
sample_provider_burst() {
|
||||
@@ -235,7 +243,10 @@ sample_provider_burst() {
|
||||
database_query "
|
||||
WITH configured AS (
|
||||
SELECT platform.id platform_id, platform.platform_key, model.id platform_model_id,
|
||||
(platform.config->>'acceptanceConcurrencyLimit')::int concurrency_limit
|
||||
platform.priority,
|
||||
(platform.config->>'acceptanceConcurrencyLimit')::int concurrency_limit,
|
||||
(platform.config->>'acceptanceRPMLimit')::int rpm_limit,
|
||||
(platform.config->>'acceptanceTPMLimit')::int tpm_limit
|
||||
FROM integration_platforms platform
|
||||
JOIN platform_models model ON model.platform_id = platform.id
|
||||
WHERE COALESCE((platform.config->>'acceptanceProviderBurst')::boolean, false)
|
||||
@@ -302,7 +313,10 @@ build_provider_burst_report() {
|
||||
platforms=$(database_query "
|
||||
WITH configured AS (
|
||||
SELECT platform.id platform_id, platform.platform_key, model.id platform_model_id,
|
||||
(platform.config->>'acceptanceConcurrencyLimit')::int concurrency_limit
|
||||
platform.priority,
|
||||
(platform.config->>'acceptanceConcurrencyLimit')::int concurrency_limit,
|
||||
(platform.config->>'acceptanceRPMLimit')::int rpm_limit,
|
||||
(platform.config->>'acceptanceTPMLimit')::int tpm_limit
|
||||
FROM integration_platforms platform
|
||||
JOIN platform_models model ON model.platform_id=platform.id
|
||||
WHERE COALESCE((platform.config->>'acceptanceProviderBurst')::boolean, false)
|
||||
@@ -332,18 +346,34 @@ WITH configured AS (
|
||||
FROM gateway_task_attempts attempt
|
||||
WHERE attempt.task_id IN (SELECT id FROM gateway_tasks WHERE acceptance_run_id='$run_id'::uuid)
|
||||
GROUP BY attempt.platform_model_id
|
||||
), rate_peaks AS (
|
||||
SELECT split_part(counter.scope_key, ':', 3)::uuid platform_model_id,
|
||||
max(counter.limit_value) FILTER (WHERE counter.metric='rpm') rpm_limit,
|
||||
max(counter.used_value + counter.reserved_value) FILTER (WHERE counter.metric='rpm') rpm_peak,
|
||||
max(counter.limit_value) FILTER (WHERE counter.metric='tpm_total') tpm_limit,
|
||||
max(counter.used_value + counter.reserved_value) FILTER (WHERE counter.metric='tpm_total') tpm_peak
|
||||
FROM gateway_rate_limit_counters counter
|
||||
WHERE counter.scope_type='platform_model'
|
||||
AND counter.scope_key LIKE 'acceptance:$run_id:%'
|
||||
GROUP BY split_part(counter.scope_key, ':', 3)::uuid
|
||||
)
|
||||
SELECT COALESCE(jsonb_agg(jsonb_build_object(
|
||||
'platformId', configured.platform_id,
|
||||
'platformModelId', configured.platform_model_id,
|
||||
'platformKey', configured.platform_key,
|
||||
'priority', configured.priority,
|
||||
'limit', configured.concurrency_limit,
|
||||
'peak', COALESCE(peaks.peak,0),
|
||||
'rpmLimit', configured.rpm_limit,
|
||||
'rpmPeak', COALESCE(rate_peaks.rpm_peak,0),
|
||||
'tpmLimit', configured.tpm_limit,
|
||||
'tpmPeak', COALESCE(rate_peaks.tpm_peak,0),
|
||||
'attempts', COALESCE(attempts.attempts,0)
|
||||
) ORDER BY configured.concurrency_limit), '[]'::jsonb)
|
||||
FROM configured
|
||||
LEFT JOIN peaks ON peaks.platform_model_id=configured.platform_model_id
|
||||
LEFT JOIN attempts ON attempts.platform_model_id=configured.platform_model_id;")
|
||||
LEFT JOIN attempts ON attempts.platform_model_id=configured.platform_model_id
|
||||
LEFT JOIN rate_peaks ON rate_peaks.platform_model_id=configured.platform_model_id;")
|
||||
queue=$(jq -s '{
|
||||
samples:length,
|
||||
maxQueuedTasks:(map(.queuedTasks)|max),
|
||||
@@ -437,7 +467,16 @@ SELECT jsonb_build_object(
|
||||
FROM gateway_task_callback_outbox callback
|
||||
WHERE callback.task_id IN (SELECT id FROM gateway_tasks WHERE acceptance_run_id='$run_id'::uuid);")
|
||||
resource_summary=$(provider_burst_resource_summary "$resources")
|
||||
if ! jq -e 'length==3 and all(.[]; .peak == .limit and .attempts > 0)' <<<"$platforms" >/dev/null; then passed=false; fi
|
||||
if ! jq -e 'length==3 and
|
||||
[.[].priority] == [100,100,200] and
|
||||
[.[].attempts] == [4,8,12] and
|
||||
all(.[]; .peak == .limit and .rpmPeak == .rpmLimit and .tpmPeak == .tpmLimit)' \
|
||||
<<<"$platforms" >/dev/null; then passed=false; fi
|
||||
if ! jq -e '.[0].priority==.[1].priority and
|
||||
.[0].attempts*2==.[1].attempts and
|
||||
.[2].priority>.[1].priority and
|
||||
.[0].peak==.[0].limit and .[1].peak==.[1].limit and .[2].attempts>0' \
|
||||
<<<"$platforms" >/dev/null; then passed=false; fi
|
||||
if ! jq -e '.queueObserved and .admissionWaitObserved and .drainedAtEnd and .maxQueuedTasks > 0 and .maxWaitingAdmissions > 0' <<<"$queue" >/dev/null; then passed=false; fi
|
||||
if ! jq -e --argjson requests "$requests" 'length==3 and all(.[]; .tasks>0 and .workerInstanceId!="unclaimed") and ([.[].tasks]|add)==$requests' <<<"$distribution" >/dev/null; then passed=false; fi
|
||||
if ! jq -e 'length==3 and all(.[]; .peakActiveTasks>0 and (.pressureStates|index("critical")|not))' <<<"$worker_peaks" >/dev/null; then passed=false; fi
|
||||
@@ -458,11 +497,20 @@ WHERE callback.task_id IN (SELECT id FROM gateway_tasks WHERE acceptance_run_id=
|
||||
--argjson resources "$resource_summary" --argjson leaks "$leaks" \
|
||||
--argjson duplicates "$duplicates" --argjson callbacks "$callbacks" \
|
||||
'{
|
||||
schemaVersion:"acceptance-provider-burst-report/v1",
|
||||
schemaVersion:"acceptance-provider-burst-report/v2",
|
||||
runId:$runId,profile:"provider-burst",model:$model,
|
||||
startedAt:(now|todateiso8601),finishedAt:(now|todateiso8601),
|
||||
passed:$passed,secretSafe:true,requests:$requests,phases:$phases,
|
||||
providerRouting:{platforms:$platforms,totalConfiguredConcurrency:([$platforms[].limit]|add)},
|
||||
providerRouting:{
|
||||
platforms:$platforms,
|
||||
totalConfiguredConcurrency:([$platforms[].limit]|add),
|
||||
samePriorityLoadBalanced:($platforms[0].priority==$platforms[1].priority and
|
||||
$platforms[0].attempts*2==$platforms[1].attempts),
|
||||
lowerPrioritySpilloverObserved:($platforms[2].priority>$platforms[1].priority and
|
||||
$platforms[0].peak==$platforms[0].limit and
|
||||
$platforms[1].peak==$platforms[1].limit and
|
||||
$platforms[2].attempts>0)
|
||||
},
|
||||
queue:$queue,
|
||||
cluster:{workers:$workers,workerPeaks:$workerPeaks,taskDistribution:$distribution,
|
||||
rateLimits:$rateLimits,concurrencyPeaks:$concurrencyPeaks,
|
||||
@@ -473,14 +521,14 @@ WHERE callback.task_id IN (SELECT id FROM gateway_tasks WHERE acceptance_run_id=
|
||||
}
|
||||
|
||||
provider_burst() {
|
||||
local requests=${AI_GATEWAY_LOCAL_PROVIDER_BURST_REQUESTS:-48}
|
||||
local requests=24
|
||||
local load_report="$report_root/provider-burst-load.json"
|
||||
local samples="$report_root/provider-burst-samples.ndjson"
|
||||
local worker_samples="$report_root/provider-burst-worker-samples.ndjson"
|
||||
local resources="$report_root/provider-burst-resources.csv"
|
||||
local report="$report_root/provider-burst.json"
|
||||
worker_sample_stop="$report_root/provider-burst-samples.stop"
|
||||
[[ $requests =~ ^[0-9]+$ ]] && ((requests >= 24 && requests <= 256))
|
||||
[[ $requests == 24 ]]
|
||||
[[ ! -e $load_report && ! -e $samples && ! -e $worker_samples && ! -e $resources && ! -e $report && ! -e $worker_sample_stop ]] ||
|
||||
fail_gate acceptance_report_exists "provider burst evidence already exists"
|
||||
current_phase=provider_burst_platforms
|
||||
@@ -495,7 +543,7 @@ provider_burst() {
|
||||
sample_resources "$resources" "$worker_sample_stop" &
|
||||
active_resource_sampler_pid=$!
|
||||
current_phase=provider_burst_load
|
||||
run_load video-throughput "$load_report" -requests "$requests"
|
||||
run_load video-provider-quota "$load_report" -requests "$requests"
|
||||
sleep 2
|
||||
touch "$worker_sample_stop"
|
||||
wait "$active_provider_sampler_pid"
|
||||
|
||||
Reference in New Issue
Block a user