feat(api-keys): 支持独立密钥额度重置（手动+定时自动）

- 新增手动重置接口 PATCH /api/admin/api-keys/{id}/reset-usage - 前端 ApiKeys 页面增加重置按钮，支持确认后归零已使用额度 - 新增独立密钥额度定时自动重置任务，支持配置周期和执行时间 - 支持 all/selected 两种重置模式，selected 模式可指定密钥 - 移除已废弃的 CleanupScheduler 兼容别名
2026-09-02 01:10:23 +08:00 · 2026-02-26 02:16:48 +08:00
parent 1a1bb0a99c
commit 0ecf8b703e
11 changed files with 646 additions and 17 deletions
--- a/frontend/src/api/admin.ts
+++ b/frontend/src/api/admin.ts
@@ -518,6 +518,14 @@ export const adminApi = {
    return response.data
  },

+  // 重置独立余额Key的已使用额度
+  async resetApiKeyUsage(keyId: string): Promise<AdminApiKey & { message: string }> {
+    const response = await apiClient.patch<AdminApiKey & { message: string }>(
+      `/api/admin/api-keys/${keyId}/reset-usage`
+    )
+    return response.data
+  },
+
  // 获取API密钥详情（可选包含完整密钥）
  async getApiKeyDetail(keyId: string, includeKey: boolean = false): Promise<AdminApiKey & { key?: string }> {
    const response = await apiClient.get<AdminApiKey & { key?: string }>(
--- a/frontend/src/views/admin/ApiKeys.vue
+++ b/frontend/src/views/admin/ApiKeys.vue
@@ -279,6 +279,15 @@
                    >
                      <DollarSign class="h-4 w-4" />
                    </Button>
+                    <Button
+                      variant="ghost"
+                      size="icon"
+                      class="h-8 w-8"
+                      title="重置额度"
+                      @click="resetKeyUsage(apiKey)"
+                    >
+                      <RotateCcw class="h-4 w-4" />
+                    </Button>
                    <Button
                      variant="ghost"
                      size="icon"
@@ -480,6 +489,15 @@
                  <DollarSign class="h-3.5 w-3.5 mr-1.5" />
                  调整
                </Button>
+                <Button
+                  variant="outline"
+                  size="sm"
+                  class="text-amber-600"
+                  @click="resetKeyUsage(apiKey)"
+                >
+                  <RotateCcw class="h-3.5 w-3.5 mr-1.5" />
+                  重置
+                </Button>
                <Button
                  variant="outline"
                  size="sm"
@@ -727,7 +745,8 @@ import {
  SquarePen,
  Search,
  Lock,
-  LockOpen
+  LockOpen,
+  RotateCcw
 } from 'lucide-vue-next'

 import { StandaloneKeyFormDialog, type StandaloneKeyFormData } from '@/features/api-keys'
@@ -971,6 +990,23 @@ async function handleAddBalance() {
  }
 }

+async function resetKeyUsage(apiKey: AdminApiKey) {
+  const confirmed = await confirmDanger(
+    `确定要重置此 Key 的已使用额度吗？\n\n${apiKey.name || apiKey.key_display || 'sk-****'}\n\n已使用额度将归零，当前余额不变。`,
+    '重置使用额度'
+  )
+  if (!confirmed) return
+
+  try {
+    const response = await adminApi.resetApiKeyUsage(apiKey.id)
+    await loadApiKeys()
+    success(response.message)
+  } catch (err: unknown) {
+    log.error('重置使用额度失败:', err)
+    error(parseApiError(err, '重置失败'))
+  }
+}
+
 function selectKey() {
  keyInput.value?.select()
 }
--- a/frontend/src/views/admin/SystemSettings.vue
+++ b/frontend/src/views/admin/SystemSettings.vue
@@ -114,7 +114,14 @@
            id="section-scheduled"
            :scheduled-tasks="scheduledTasks"
            :quota-reset-interval-days="systemConfig.user_quota_reset_interval_days"
+            :standalone-key-reset-interval-days="systemConfig.standalone_key_quota_reset_interval_days"
+            :standalone-key-reset-mode="systemConfig.standalone_key_quota_reset_mode"
+            :standalone-key-reset-key-ids="systemConfig.standalone_key_quota_reset_key_ids"
+            :standalone-keys="standaloneKeys"
            @update:quota-reset-interval-days="systemConfig.user_quota_reset_interval_days = $event"
+            @update:standalone-key-reset-interval-days="systemConfig.standalone_key_quota_reset_interval_days = $event"
+            @update:standalone-key-reset-mode="handleStandaloneKeyResetModeChange"
+            @toggle-standalone-key-reset-key-id="handleToggleStandaloneKeyResetKeyId"
          />

          <!-- 系统版本信息 -->
@@ -194,6 +201,7 @@
 import { ref, onMounted, onBeforeUnmount, nextTick } from 'vue'
 import { PageHeader, PageContainer } from '@/components/layout'
 import { useProxyNodesStore } from '@/stores/proxy-nodes'
+import { adminApi } from '@/api/admin'

 // Composables
 import { useSystemConfig } from './system-settings/composables/useSystemConfig'
@@ -336,13 +344,50 @@ const {
 const {
  scheduledTasks,
  initPreviousValues,
+  saveStandaloneKeyResetMode,
+  saveStandaloneKeyResetKeyIds,
 } = useScheduledTasks(systemConfig)

+// 独立密钥列表（用于定时任务配置中的密钥选择）
+const standaloneKeys = ref<Array<{ id: string; name?: string; key_display?: string; current_balance_usd?: number | null }>>([])
+
+async function loadStandaloneKeys() {
+  try {
+    const result = await adminApi.getAllApiKeys({ limit: 2000 })
+    standaloneKeys.value = result.api_keys.map((k) => ({
+      id: k.id,
+      name: k.name,
+      key_display: k.key_display,
+      current_balance_usd: k.current_balance_usd,
+    }))
+  } catch {
+    // 加载失败不影响其他功能
+  }
+}
+
+function handleStandaloneKeyResetModeChange(mode: string) {
+  systemConfig.value.standalone_key_quota_reset_mode = mode
+  saveStandaloneKeyResetMode(mode)
+}
+
+function handleToggleStandaloneKeyResetKeyId(keyId: string) {
+  const ids = [...systemConfig.value.standalone_key_quota_reset_key_ids]
+  const idx = ids.indexOf(keyId)
+  if (idx >= 0) {
+    ids.splice(idx, 1)
+  } else {
+    ids.push(keyId)
+  }
+  systemConfig.value.standalone_key_quota_reset_key_ids = ids
+  saveStandaloneKeyResetKeyIds(ids)
+}
+
 onMounted(async () => {
  await Promise.all([
    loadSystemConfig(),
    loadSystemVersion(),
    proxyNodesStore.ensureLoaded(),
+    loadStandaloneKeys(),
  ])
  // 配置加载完成后初始化定时任务的原始值
  initPreviousValues()
--- a/frontend/src/views/admin/system-settings/ScheduledTasksSection.vue
+++ b/frontend/src/views/admin/system-settings/ScheduledTasksSection.vue
@@ -145,6 +145,96 @@
              滚动计算：距离上次成功执行满 N 天后再次执行
            </p>
          </div>
+
+          <!-- 独立密钥额度重置额外配置 -->
+          <div
+            v-if="task.id === 'standalone-key-quota-reset' && task.enabled"
+            class="px-4 pb-4 pt-0 space-y-3"
+          >
+            <!-- 重置周期 -->
+            <div class="flex items-center gap-3 p-3 rounded-lg bg-muted/30 border border-border/50">
+              <div class="flex items-center gap-2 text-sm">
+                <span class="text-muted-foreground">重置周期</span>
+                <div class="flex items-center gap-1.5">
+                  <span class="text-muted-foreground">每</span>
+                  <Input
+                    :model-value="standaloneKeyResetIntervalDays"
+                    type="number"
+                    min="1"
+                    step="1"
+                    class="w-14 h-7 text-xs text-center px-2"
+                    @update:model-value="$emit('update:standaloneKeyResetIntervalDays', Number($event))"
+                  />
+                  <span class="text-muted-foreground">天</span>
+                </div>
+              </div>
+            </div>
+
+            <!-- 重置范围 -->
+            <div class="p-3 rounded-lg bg-muted/30 border border-border/50 space-y-3">
+              <div class="flex items-center gap-2 text-sm">
+                <span class="text-muted-foreground">重置范围</span>
+                <Select
+                  :model-value="standaloneKeyResetMode"
+                  @update:model-value="$emit('update:standaloneKeyResetMode', $event)"
+                >
+                  <SelectTrigger class="w-32 h-7 text-xs">
+                    <SelectValue />
+                  </SelectTrigger>
+                  <SelectContent>
+                    <SelectItem value="all">
+                      全部独立密钥
+                    </SelectItem>
+                    <SelectItem value="selected">
+                      指定密钥
+                    </SelectItem>
+                  </SelectContent>
+                </Select>
+              </div>
+
+              <!-- 密钥多选列表 -->
+              <div
+                v-if="standaloneKeyResetMode === 'selected'"
+                class="space-y-2"
+              >
+                <div class="text-xs text-muted-foreground">
+                  选择需要重置的密钥:
+                </div>
+                <div
+                  v-if="standaloneKeys.length === 0"
+                  class="text-xs text-muted-foreground/60 py-2"
+                >
+                  暂无独立密钥
+                </div>
+                <div
+                  v-else
+                  class="max-h-48 overflow-y-auto space-y-1"
+                >
+                  <label
+                    v-for="key in standaloneKeys"
+                    :key="key.id"
+                    class="flex items-center gap-2 p-2 rounded hover:bg-muted/50 cursor-pointer text-xs"
+                  >
+                    <Checkbox
+                      :model-value="standaloneKeyResetKeyIds.includes(key.id)"
+                      @update:model-value="$emit('toggleStandaloneKeyResetKeyId', key.id)"
+                    />
+                    <span class="truncate">{{ key.name || key.key_display || 'sk-****' }}</span>
+                    <span
+                      v-if="key.current_balance_usd != null"
+                      class="text-muted-foreground ml-auto shrink-0"
+                    >
+                      ${{ key.current_balance_usd.toFixed(2) }}
+                    </span>
+                  </label>
+                </div>
+              </div>
+            </div>
+
+            <p class="text-[11px] text-muted-foreground ml-1">
+              滚动计算：距离上次成功执行满 N 天后再次执行
+            </p>
+          </div>
        </div>
      </template>
    </div>
@@ -156,6 +246,7 @@ import { Clock, Check, Loader2, X } from 'lucide-vue-next'
 import Button from '@/components/ui/button.vue'
 import Input from '@/components/ui/input.vue'
 import Switch from '@/components/ui/switch.vue'
+import Checkbox from '@/components/ui/checkbox.vue'
 import Select from '@/components/ui/select.vue'
 import SelectTrigger from '@/components/ui/select-trigger.vue'
 import SelectValue from '@/components/ui/select-value.vue'
@@ -181,12 +272,26 @@ interface ScheduledTask {
  onCancel: () => void
 }

+interface StandaloneKeyOption {
+  id: string
+  name?: string
+  key_display?: string
+  current_balance_usd?: number | null
+}
+
 defineProps<{
  scheduledTasks: ScheduledTask[]
  quotaResetIntervalDays: number
+  standaloneKeyResetIntervalDays: number
+  standaloneKeyResetMode: string
+  standaloneKeyResetKeyIds: string[]
+  standaloneKeys: StandaloneKeyOption[]
 }>()

 defineEmits<{
  'update:quotaResetIntervalDays': [value: number]
+  'update:standaloneKeyResetIntervalDays': [value: number]
+  'update:standaloneKeyResetMode': [value: string]
+  'toggleStandaloneKeyResetKeyId': [keyId: string]
 }>()
 </script>
--- a/frontend/src/views/admin/system-settings/composables/useScheduledTasks.ts
+++ b/frontend/src/views/admin/system-settings/composables/useScheduledTasks.ts
@@ -1,5 +1,5 @@
 import { ref, computed, type Ref } from 'vue'
-import { CalendarCheck, RotateCcw, RefreshCw } from 'lucide-vue-next'
+import { CalendarCheck, RotateCcw, RefreshCw, KeyRound } from 'lucide-vue-next'
 import { useToast } from '@/composables/useToast'
 import { adminApi } from '@/api/admin'
 import { log } from '@/utils/logger'
@@ -10,18 +10,24 @@ export function useScheduledTasks(systemConfig: Ref<SystemConfig>) {

  const checkinConfigLoading = ref(false)
  const quotaResetConfigLoading = ref(false)
+  const standaloneKeyResetConfigLoading = ref(false)

  // 签到时间的原始值（用于回滚）
  const previousCheckinTime = ref('')
  // 用户配额重置时间的原始值
  const previousUserQuotaResetTime = ref('')
  const previousUserQuotaResetIntervalDays = ref(1)
+  // 独立密钥额度重置的原始值
+  const previousStandaloneKeyResetTime = ref('')
+  const previousStandaloneKeyResetIntervalDays = ref(1)

  // 初始化原始值（在配置加载完成后调用）
  function initPreviousValues() {
    previousCheckinTime.value = systemConfig.value.provider_checkin_time
    previousUserQuotaResetTime.value = systemConfig.value.user_quota_reset_time
    previousUserQuotaResetIntervalDays.value = systemConfig.value.user_quota_reset_interval_days
+    previousStandaloneKeyResetTime.value = systemConfig.value.standalone_key_quota_reset_time
+    previousStandaloneKeyResetIntervalDays.value = systemConfig.value.standalone_key_quota_reset_interval_days
  }

  // 签到时间
@@ -77,6 +83,38 @@ export function useScheduledTasks(systemConfig: Ref<SystemConfig>) {
    return hasUserQuotaResetTimeChanged.value || hasUserQuotaResetIntervalChanged.value
  })

+  // 独立密钥额度重置时间
+  const standaloneKeyResetHour = computed(() => {
+    const time = systemConfig.value.standalone_key_quota_reset_time
+    if (!time || !time.includes(':')) return '05'
+    return time.split(':')[0]
+  })
+
+  const standaloneKeyResetMinute = computed(() => {
+    const time = systemConfig.value.standalone_key_quota_reset_time
+    if (!time || !time.includes(':')) return '00'
+    return time.split(':')[1]
+  })
+
+  function updateStandaloneKeyResetTime(hour: string, minute: string) {
+    systemConfig.value.standalone_key_quota_reset_time = `${hour}:${minute}`
+  }
+
+  const hasStandaloneKeyResetTimeChanged = computed(() => {
+    return systemConfig.value.standalone_key_quota_reset_time !== previousStandaloneKeyResetTime.value
+  })
+
+  const hasStandaloneKeyResetIntervalChanged = computed(() => {
+    return (
+      systemConfig.value.standalone_key_quota_reset_interval_days !==
+      previousStandaloneKeyResetIntervalDays.value
+    )
+  })
+
+  const hasStandaloneKeyResetConfigChanged = computed(() => {
+    return hasStandaloneKeyResetTimeChanged.value || hasStandaloneKeyResetIntervalChanged.value
+  })
+
  // Toggle handlers
  async function handleProviderCheckinToggle(enabled: boolean) {
    const previousValue = systemConfig.value.enable_provider_checkin
@@ -129,6 +167,23 @@ export function useScheduledTasks(systemConfig: Ref<SystemConfig>) {
    }
  }

+  async function handleStandaloneKeyResetToggle(enabled: boolean) {
+    const previousValue = systemConfig.value.enable_standalone_key_quota_reset
+    systemConfig.value.enable_standalone_key_quota_reset = enabled
+    try {
+      await adminApi.updateSystemConfig(
+        'enable_standalone_key_quota_reset',
+        enabled,
+        '是否启用独立密钥额度自动重置任务'
+      )
+      success(enabled ? '已启用独立密钥额度自动重置' : '已禁用独立密钥额度自动重置')
+    } catch (err) {
+      error('保存配置失败')
+      log.error('保存独立密钥额度自动重置配置失败:', err)
+      systemConfig.value.enable_standalone_key_quota_reset = previousValue
+    }
+  }
+
  // Cancel handlers
  function handleCheckinTimeCancel() {
    systemConfig.value.provider_checkin_time = previousCheckinTime.value
@@ -139,6 +194,11 @@ export function useScheduledTasks(systemConfig: Ref<SystemConfig>) {
    systemConfig.value.user_quota_reset_interval_days = previousUserQuotaResetIntervalDays.value
  }

+  function handleStandaloneKeyResetConfigCancel() {
+    systemConfig.value.standalone_key_quota_reset_time = previousStandaloneKeyResetTime.value
+    systemConfig.value.standalone_key_quota_reset_interval_days = previousStandaloneKeyResetIntervalDays.value
+  }
+
  // Save handlers
  async function handleCheckinTimeSave() {
    const newTime = systemConfig.value.provider_checkin_time
@@ -232,6 +292,103 @@ export function useScheduledTasks(systemConfig: Ref<SystemConfig>) {
    }
  }

+  async function handleStandaloneKeyResetConfigSave() {
+    const configItems: Array<{
+      key: string
+      value: unknown
+      description: string
+      onSuccess: () => void
+    }> = []
+
+    if (hasStandaloneKeyResetTimeChanged.value) {
+      const newTime = systemConfig.value.standalone_key_quota_reset_time
+      if (!newTime || !/^\d{2}:\d{2}$/.test(newTime)) {
+        error('请输入有效的时间格式 (HH:MM)')
+        return
+      }
+      configItems.push({
+        key: 'standalone_key_quota_reset_time',
+        value: newTime,
+        description: '独立密钥额度自动重置执行时间（HH:MM 格式）',
+        onSuccess: () => {
+          previousStandaloneKeyResetTime.value = newTime
+        },
+      })
+    }
+
+    if (hasStandaloneKeyResetIntervalChanged.value) {
+      let intervalDays = Number(systemConfig.value.standalone_key_quota_reset_interval_days)
+      if (!Number.isFinite(intervalDays) || intervalDays < 1) intervalDays = 1
+      intervalDays = Math.trunc(intervalDays)
+
+      systemConfig.value.standalone_key_quota_reset_interval_days = intervalDays
+
+      configItems.push({
+        key: 'standalone_key_quota_reset_interval_days',
+        value: intervalDays,
+        description: '独立密钥额度重置周期（天数），滚动计算',
+        onSuccess: () => {
+          previousStandaloneKeyResetIntervalDays.value = intervalDays
+        },
+      })
+    }
+
+    if (configItems.length === 0) return
+
+    standaloneKeyResetConfigLoading.value = true
+    const failedKeys: string[] = []
+
+    try {
+      for (const item of configItems) {
+        try {
+          await adminApi.updateSystemConfig(item.key, item.value, item.description)
+          item.onSuccess()
+        } catch (err) {
+          failedKeys.push(item.key)
+          log.error(`保存独立密钥额度重置配置失败: ${item.key}`, err)
+        }
+      }
+
+      if (failedKeys.length > 0) {
+        error(`部分配置保存失败: ${failedKeys.join(', ')}`)
+        return
+      }
+
+      success('独立密钥额度重置配置已保存')
+    } finally {
+      standaloneKeyResetConfigLoading.value = false
+    }
+  }
+
+  // 保存独立密钥额度重置模式和选中密钥
+  async function saveStandaloneKeyResetMode(mode: string) {
+    try {
+      await adminApi.updateSystemConfig(
+        'standalone_key_quota_reset_mode',
+        mode,
+        '独立密钥额度重置模式'
+      )
+      success('重置模式已保存')
+    } catch (err) {
+      error('保存重置模式失败')
+      log.error('保存独立密钥额度重置模式失败:', err)
+    }
+  }
+
+  async function saveStandaloneKeyResetKeyIds(keyIds: string[]) {
+    try {
+      await adminApi.updateSystemConfig(
+        'standalone_key_quota_reset_key_ids',
+        keyIds,
+        '独立密钥额度重置指定的密钥 ID 列表'
+      )
+      success('已保存选中密钥')
+    } catch (err) {
+      error('保存选中密钥失败')
+      log.error('保存独立密钥额度重置密钥列表失败:', err)
+    }
+  }
+
  // 定时任务配置列表
  const scheduledTasks = computed(() => [
    {
@@ -282,12 +439,31 @@ export function useScheduledTasks(systemConfig: Ref<SystemConfig>) {
      onSave: () => {},
      onCancel: () => {},
    },
+    {
+      id: 'standalone-key-quota-reset',
+      icon: KeyRound,
+      title: '独立密钥额度自动重置',
+      description: '定时将独立密钥已使用额度重置为零',
+      enabled: systemConfig.value.enable_standalone_key_quota_reset,
+      hasTimeConfig: true,
+      hour: standaloneKeyResetHour.value,
+      minute: standaloneKeyResetMinute.value,
+      updateTime: updateStandaloneKeyResetTime,
+      hasChanges: hasStandaloneKeyResetConfigChanged.value,
+      loading: standaloneKeyResetConfigLoading.value,
+      onToggle: handleStandaloneKeyResetToggle,
+      onSave: handleStandaloneKeyResetConfigSave,
+      onCancel: handleStandaloneKeyResetConfigCancel,
+    },
  ])

  return {
    checkinConfigLoading,
    quotaResetConfigLoading,
+    standaloneKeyResetConfigLoading,
    scheduledTasks,
    initPreviousValues,
+    saveStandaloneKeyResetMode,
+    saveStandaloneKeyResetKeyIds,
  }
 }
--- a/frontend/src/views/admin/system-settings/composables/useSystemConfig.ts
+++ b/frontend/src/views/admin/system-settings/composables/useSystemConfig.ts
@@ -38,6 +38,12 @@ export interface SystemConfig {
  user_quota_reset_time: string
  user_quota_reset_interval_days: number
  enable_oauth_token_refresh: boolean
+  // 独立密钥额度重置
+  enable_standalone_key_quota_reset: boolean
+  standalone_key_quota_reset_time: string
+  standalone_key_quota_reset_interval_days: number
+  standalone_key_quota_reset_mode: string
+  standalone_key_quota_reset_key_ids: string[]
 }

 const CONFIG_KEYS = [
@@ -74,6 +80,12 @@ const CONFIG_KEYS = [
  'user_quota_reset_time',
  'user_quota_reset_interval_days',
  'enable_oauth_token_refresh',
+  // 独立密钥额度重置
+  'enable_standalone_key_quota_reset',
+  'standalone_key_quota_reset_time',
+  'standalone_key_quota_reset_interval_days',
+  'standalone_key_quota_reset_mode',
+  'standalone_key_quota_reset_key_ids',
 ]

 function createDefaultConfig(): SystemConfig {
@@ -111,6 +123,12 @@ function createDefaultConfig(): SystemConfig {
    user_quota_reset_time: '05:00',
    user_quota_reset_interval_days: 1,
    enable_oauth_token_refresh: true,
+    // 独立密钥额度重置
+    enable_standalone_key_quota_reset: false,
+    standalone_key_quota_reset_time: '05:00',
+    standalone_key_quota_reset_interval_days: 1,
+    standalone_key_quota_reset_mode: 'all',
+    standalone_key_quota_reset_key_ids: [],
  }
 }