Added Usage page.

This commit is contained in:
decolua 2026-01-19 10:39:30 +07:00
parent da15660681
commit 3804357aba
12 changed files with 237 additions and 138 deletions

View file

@ -1,6 +1,6 @@
"use client";
import { useState, useEffect, useMemo } from "react";
import { useState, useEffect, useMemo, useCallback } from "react";
import { useSearchParams, useRouter } from "next/navigation";
import Card from "./Card";
import Badge from "./Badge";
@ -38,6 +38,8 @@ export default function UsageStats() {
const [loading, setLoading] = useState(true);
const [autoRefresh, setAutoRefresh] = useState(true);
const [viewMode, setViewMode] = useState("tokens"); // 'tokens' or 'costs'
const [refreshInterval, setRefreshInterval] = useState(5000); // Start with 5s
const [prevTotalRequests, setPrevTotalRequests] = useState(0);
const toggleSort = (field) => {
const params = new URLSearchParams(searchParams.toString());
@ -114,34 +116,66 @@ export default function UsageStats() {
return sortData(stats?.byAccount, accountPendingMap);
}, [stats?.byAccount, stats?.pending?.byAccount, sortBy, sortOrder]);
useEffect(() => {
fetchStats();
}, []);
useEffect(() => {
let interval;
if (autoRefresh) {
interval = setInterval(() => {
fetchStats(false); // fetch without loading skeleton
}, 1000);
}
return () => clearInterval(interval);
}, [autoRefresh]);
const fetchStats = async (showLoading = true) => {
const fetchStats = useCallback(async (showLoading = true) => {
if (showLoading) setLoading(true);
try {
const res = await fetch("/api/usage/history");
if (res.ok) {
const data = await res.json();
setStats(data);
// Smart polling: adjust interval based on activity
const currentTotal = data.totalRequests || 0;
if (currentTotal > prevTotalRequests) {
// New requests detected - reset to fast polling
setRefreshInterval(5000);
} else {
// No change - increase interval (exponential backoff)
setRefreshInterval((prev) => Math.min(prev * 2, 60000)); // Max 60s
}
setPrevTotalRequests(currentTotal);
}
} catch (error) {
console.error("Failed to fetch usage stats:", error);
} finally {
if (showLoading) setLoading(false);
}
};
}, [prevTotalRequests]);
useEffect(() => {
fetchStats();
}, [fetchStats]);
useEffect(() => {
let intervalId;
let isPageVisible = true;
// Page Visibility API - pause when tab is hidden
const handleVisibilityChange = () => {
isPageVisible = !document.hidden;
if (isPageVisible && autoRefresh) {
fetchStats(false); // Fetch immediately when tab becomes visible
}
};
document.addEventListener("visibilitychange", handleVisibilityChange);
if (autoRefresh) {
// Clear any existing interval first
if (intervalId) clearInterval(intervalId);
intervalId = setInterval(() => {
if (isPageVisible) {
fetchStats(false); // fetch without loading skeleton
}
}, refreshInterval);
}
return () => {
if (intervalId) clearInterval(intervalId);
document.removeEventListener("visibilitychange", handleVisibilityChange);
};
}, [autoRefresh, refreshInterval, fetchStats]);
if (loading) return <CardSkeleton />;
@ -202,7 +236,7 @@ export default function UsageStats() {
{/* Auto Refresh Toggle */}
<label className="text-sm font-medium text-text-muted flex items-center gap-2 cursor-pointer">
<span>Auto Refresh (1s)</span>
<span>Auto Refresh ({refreshInterval / 1000}s)</span>
<div
onClick={() => setAutoRefresh(!autoRefresh)}
className={`relative inline-flex h-5 w-9 items-center rounded-full transition-colors focus:outline-none ${
@ -291,7 +325,7 @@ export default function UsageStats() {
<div className="flex justify-between items-start gap-4">
<div className="flex flex-col gap-1 flex-1">
<span className="text-text-muted text-sm uppercase font-semibold">
Total Output Tokens
Output Tokens
</span>
<span className="text-2xl font-bold text-success">
{fmt(stats.totalCompletionTokens)}