85 lines
3.5 KiB
SQL
85 lines
3.5 KiB
SQL
-- name: CreateKolUsageLog :one
|
|
INSERT INTO kol_usage_logs (tenant_id, subscription_id, subscription_prompt_id,
|
|
package_id, prompt_id, prompt_revision_no,
|
|
generation_task_id, status, variables_snapshot,
|
|
schema_snapshot, rendered_prompt_hash)
|
|
VALUES (sqlc.arg(tenant_id)::bigint, sqlc.arg(subscription_id),
|
|
sqlc.arg(subscription_prompt_id), sqlc.arg(package_id),
|
|
sqlc.arg(prompt_id), sqlc.arg(prompt_revision_no),
|
|
sqlc.arg(generation_task_id), 'pending',
|
|
sqlc.arg(variables_snapshot)::jsonb, sqlc.arg(schema_snapshot)::jsonb,
|
|
sqlc.arg(rendered_prompt_hash))
|
|
RETURNING id, tenant_id, generation_task_id, status, created_at;
|
|
|
|
-- name: MarkKolUsageCompleted :exec
|
|
UPDATE kol_usage_logs
|
|
SET status = 'completed', article_id = sqlc.arg(article_id),
|
|
finished_at = NOW()
|
|
WHERE id = sqlc.arg(id)
|
|
AND tenant_id = sqlc.arg(tenant_id)::bigint;
|
|
|
|
-- name: MarkKolUsageFailed :exec
|
|
UPDATE kol_usage_logs
|
|
SET status = 'failed', error_message = sqlc.arg(error_message),
|
|
finished_at = NOW()
|
|
WHERE id = sqlc.arg(id)
|
|
AND tenant_id = sqlc.arg(tenant_id)::bigint;
|
|
|
|
-- name: MarkKolUsageCompletedByTask :exec
|
|
UPDATE kol_usage_logs
|
|
SET status = 'completed', article_id = sqlc.arg(article_id), finished_at = NOW()
|
|
WHERE generation_task_id = sqlc.arg(generation_task_id)
|
|
AND tenant_id = sqlc.arg(tenant_id)::bigint;
|
|
|
|
-- name: MarkKolUsageFailedByTask :exec
|
|
UPDATE kol_usage_logs
|
|
SET status = 'failed', error_message = sqlc.arg(error_message), finished_at = NOW()
|
|
WHERE generation_task_id = sqlc.arg(generation_task_id)
|
|
AND tenant_id = sqlc.arg(tenant_id)::bigint;
|
|
|
|
-- name: GetKolUsageByGenerationTask :one
|
|
SELECT id, tenant_id, package_id, prompt_id, status, created_at
|
|
FROM kol_usage_logs
|
|
WHERE generation_task_id = sqlc.arg(generation_task_id)
|
|
AND tenant_id = sqlc.arg(tenant_id)::bigint;
|
|
|
|
-- name: KolDashboardOverview :one
|
|
SELECT
|
|
(SELECT COUNT(DISTINCT s.tenant_id) FROM kol_subscriptions s
|
|
WHERE s.package_id = ANY(sqlc.arg(package_ids)::bigint[]) AND s.status='active'
|
|
AND (s.end_at IS NULL OR s.end_at > NOW())) AS total_subs,
|
|
(SELECT COUNT(*) FROM kol_subscriptions s
|
|
WHERE s.package_id = ANY(sqlc.arg(package_ids)::bigint[])
|
|
AND s.status='active'
|
|
AND s.created_at >= date_trunc('month', NOW())) AS month_subs,
|
|
(SELECT COUNT(*) FROM kol_usage_logs u
|
|
WHERE u.package_id = ANY(sqlc.arg(package_ids)::bigint[])
|
|
AND u.tenant_id IS NOT NULL) AS total_usage,
|
|
(SELECT COUNT(*) FROM kol_usage_logs u
|
|
WHERE u.package_id = ANY(sqlc.arg(package_ids)::bigint[])
|
|
AND u.tenant_id IS NOT NULL
|
|
AND u.created_at >= date_trunc('month', NOW())) AS month_usage,
|
|
(SELECT COUNT(*) FROM kol_usage_logs u
|
|
WHERE u.package_id = ANY(sqlc.arg(package_ids)::bigint[])
|
|
AND u.tenant_id IS NOT NULL
|
|
AND u.status='failed') AS failure_count;
|
|
|
|
-- name: KolDashboardTrend :many
|
|
SELECT date_trunc('day', u.created_at)::date AS d,
|
|
COUNT(*) FILTER (WHERE u.status='completed') AS usages
|
|
FROM kol_usage_logs u
|
|
WHERE u.package_id = ANY(sqlc.arg(package_ids)::bigint[])
|
|
AND u.tenant_id IS NOT NULL
|
|
AND u.created_at >= NOW() - sqlc.arg(period_days)::int * INTERVAL '1 day'
|
|
GROUP BY d
|
|
ORDER BY d ASC;
|
|
|
|
-- name: KolUsageCountByPackage :many
|
|
SELECT package_id, COUNT(*) AS total,
|
|
SUM(CASE WHEN status = 'completed' THEN 1 ELSE 0 END) AS completed,
|
|
SUM(CASE WHEN status = 'failed' THEN 1 ELSE 0 END) AS failed
|
|
FROM kol_usage_logs
|
|
WHERE package_id = ANY(sqlc.arg(package_ids)::bigint[])
|
|
AND tenant_id IS NOT NULL
|
|
GROUP BY package_id;
|