|
|
const axios = require('axios') |
|
|
|
|
|
const BASE_URL = 'http://localhost:3312' |
|
|
|
|
|
|
|
|
const API_KEY = 'cr_your_api_key_here' |
|
|
|
|
|
async function testWindowRemaining() { |
|
|
try { |
|
|
console.log('🔍 测试时间窗口剩余时间功能...\n') |
|
|
|
|
|
|
|
|
console.log('1. 获取 API Key ID...') |
|
|
const idResponse = await axios.post(`${BASE_URL}/api-stats/api/get-key-id`, { |
|
|
apiKey: API_KEY |
|
|
}) |
|
|
|
|
|
if (!idResponse.data.success) { |
|
|
throw new Error('Failed to get API Key ID') |
|
|
} |
|
|
|
|
|
const apiId = idResponse.data.data.id |
|
|
console.log(` ✅ API Key ID: ${apiId}\n`) |
|
|
|
|
|
|
|
|
console.log('2. 查询统计数据(包含时间窗口信息)...') |
|
|
const statsResponse = await axios.post(`${BASE_URL}/api-stats/api/user-stats`, { |
|
|
apiId |
|
|
}) |
|
|
|
|
|
if (!statsResponse.data.success) { |
|
|
throw new Error('Failed to get stats data') |
|
|
} |
|
|
|
|
|
const stats = statsResponse.data.data |
|
|
console.log(` ✅ 成功获取统计数据\n`) |
|
|
|
|
|
|
|
|
console.log('3. 时间窗口信息:') |
|
|
console.log(` - 窗口时长: ${stats.limits.rateLimitWindow} 分钟`) |
|
|
console.log(` - 请求限制: ${stats.limits.rateLimitRequests || '无限制'}`) |
|
|
console.log(` - Token限制: ${stats.limits.tokenLimit || '无限制'}`) |
|
|
console.log(` - 当前请求数: ${stats.limits.currentWindowRequests}`) |
|
|
console.log(` - 当前Token数: ${stats.limits.currentWindowTokens}`) |
|
|
|
|
|
if (stats.limits.windowStartTime) { |
|
|
const startTime = new Date(stats.limits.windowStartTime) |
|
|
const endTime = new Date(stats.limits.windowEndTime) |
|
|
|
|
|
console.log(`\n ⏰ 时间窗口状态:`) |
|
|
console.log(` - 窗口开始时间: ${startTime.toLocaleString()}`) |
|
|
console.log(` - 窗口结束时间: ${endTime.toLocaleString()}`) |
|
|
console.log(` - 剩余时间: ${stats.limits.windowRemainingSeconds} 秒`) |
|
|
|
|
|
if (stats.limits.windowRemainingSeconds > 0) { |
|
|
const minutes = Math.floor(stats.limits.windowRemainingSeconds / 60) |
|
|
const seconds = stats.limits.windowRemainingSeconds % 60 |
|
|
console.log(` - 格式化剩余时间: ${minutes}分${seconds}秒`) |
|
|
console.log(` - 窗口状态: 🟢 活跃中`) |
|
|
} else { |
|
|
console.log(` - 窗口状态: 🔴 已过期(下次请求时重置)`) |
|
|
} |
|
|
} else { |
|
|
console.log(`\n ⏰ 时间窗口状态: ⚪ 未启动(还没有任何请求)`) |
|
|
} |
|
|
|
|
|
console.log('\n✅ 测试完成!时间窗口剩余时间功能正常工作。') |
|
|
} catch (error) { |
|
|
console.error('❌ 测试失败:', error.message) |
|
|
if (error.response) { |
|
|
console.error(' 响应数据:', error.response.data) |
|
|
} |
|
|
process.exit(1) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
testWindowRemaining() |
|
|
|