初次提交

2026-09-06 08:19:02 +08:00
commit 4ce4649aaf
1061 changed files with 337258 additions and 0 deletions
@@ -0,0 +1,34 @@
+import { verifyAdminAuth } from '~~/server/utils/auth'
+import { databaseManager } from '~~/server/utils/database-manager'
+
+export default defineEventHandler(async (event) => {
+  try {
+    // 验证管理员权限
+    await verifyAdminAuth(event)
+
+    // 清理过期会话
+    const cleanedCount = await databaseManager.cleanupExpiredSessions()
+
+    // 清除健康检查缓存以确保下次检查获取最新状态
+    databaseManager.clearHealthCheckCache()
+
+    return {
+      success: true,
+      message: `Successfully cleaned up ${cleanedCount} expired sessions`,
+      cleanedCount,
+      timestamp: new Date().toISOString()
+    }
+  } catch (error) {
+    if (error instanceof Error && error.message.includes('认证')) {
+      throw createError({
+        statusCode: 401,
+        statusMessage: 'Unauthorized'
+      })
+    }
+
+    throw createError({
+      statusCode: 500,
+      statusMessage: error instanceof Error ? error.message : 'Cleanup failed'
+    })
+  }
+})
@@ -0,0 +1,34 @@
+import { verifyAdminAuth } from '~~/server/utils/auth'
+import { databaseManager } from '~~/server/utils/database-manager'
+
+export default defineEventHandler(async (event) => {
+  try {
+    // 验证管理员权限
+    const authResult = await verifyAdminAuth(event)
+
+    if (!authResult.success) {
+      throw createError({
+        statusCode: 401,
+        statusMessage: authResult.message
+      })
+    }
+
+    // 使用改进的数据库管理器获取性能指标
+    const performanceMetrics = await databaseManager.getPerformanceMetrics()
+
+    return {
+      ...performanceMetrics,
+      timestamp: new Date().toISOString()
+    }
+  } catch (error: any) {
+    if (error.statusCode === 401) {
+      throw error
+    }
+
+    console.error('Database performance API error:', error)
+    throw createError({
+      statusCode: 500,
+      statusMessage: 'Internal Server Error'
+    })
+  }
+})
@@ -0,0 +1,34 @@
+import { verifyAdminAuth } from '~~/server/utils/auth'
+import { databaseManager } from '~~/server/utils/database-manager'
+
+export default defineEventHandler(async (event) => {
+  try {
+    // 验证管理员权限
+    const authResult = await verifyAdminAuth(event)
+
+    if (!authResult.success) {
+      throw createError({
+        statusCode: 401,
+        statusMessage: authResult.message
+      })
+    }
+
+    // 使用改进的数据库管理器获取连接池状态
+    const poolStatus = await databaseManager.getConnectionPoolStatus()
+
+    return {
+      ...poolStatus,
+      timestamp: new Date().toISOString()
+    }
+  } catch (error: any) {
+    if (error.statusCode === 401) {
+      throw error
+    }
+
+    console.error('Database pool status API error:', error)
+    throw createError({
+      statusCode: 500,
+      statusMessage: 'Internal Server Error'
+    })
+  }
+})
@@ -0,0 +1,254 @@
+import { createError, defineEventHandler } from 'h3'
+import { db } from '~/drizzle/db'
+import { CacheService } from '../../../services/cacheService'
+import {
+  apiKeyPermissions,
+  apiKeys,
+  apiLogs,
+  collaborationLogs,
+  emailTemplates,
+  notificationSettings,
+  notifications,
+  playTimes,
+  requestTimes,
+  schedules,
+  semesters,
+  songBlacklists,
+  songCollaborators,
+  songReplayRequests,
+  songs,
+  systemSettings,
+  userIdentities,
+  users,
+  userStatusLogs,
+  votes
+} from '~/drizzle/schema'
+import { ne, sql } from 'drizzle-orm'
+
+// 重置所有表的自增序列
+async function resetAutoIncrementSequences() {
+  const tables = [
+    { name: 'users', dbName: 'User' },
+    { name: 'songs', dbName: 'Song' },
+    { name: 'votes', dbName: 'Vote' },
+    { name: 'schedules', dbName: 'Schedule' },
+    { name: 'notifications', dbName: 'Notification' },
+    { name: 'notificationSettings', dbName: 'NotificationSettings' },
+    { name: 'playTimes', dbName: 'PlayTime' },
+    { name: 'semesters', dbName: 'Semester' },
+    { name: 'systemSettings', dbName: 'SystemSettings' },
+    { name: 'songBlacklists', dbName: 'SongBlacklist' },
+    { name: 'userStatusLogs', dbName: 'user_status_logs' },
+    { name: 'userIdentities', dbName: 'UserIdentity' },
+    { name: 'requestTimes', dbName: 'RequestTime' },
+    { name: 'songReplayRequests', dbName: 'song_replay_requests' },
+    { name: 'emailTemplates', dbName: 'EmailTemplate' }
+  ]
+
+  const results = []
+
+  for (const table of tables) {
+    try {
+      // 获取序列名称
+      const sequenceNameResult = await db.execute(
+        sql.raw(`SELECT pg_get_serial_sequence('"${table.dbName}"', 'id') as sequence_name`)
+      )
+      const sequenceName = (sequenceNameResult as any)[0]?.sequence_name
+
+      if (sequenceName) {
+        // 重置序列值到 1
+        await db.execute(sql.raw(`ALTER SEQUENCE ${sequenceName} RESTART WITH 1`))
+        results.push({ table: table.name, success: true })
+      } else {
+        results.push({ table: table.name, success: false, message: 'Sequence not found' })
+      }
+    } catch (error) {
+      console.warn(`重置 ${table.name} 表序列失败: ${error.message}`)
+      results.push({ table: table.name, success: false, error: error.message })
+    }
+  }
+  return results
+}
+
+export default defineEventHandler(async (event) => {
+  try {
+    // 验证管理员权限
+    const user = event.context.user
+    if (!user || !['ADMIN', 'SUPER_ADMIN'].includes(user.role)) {
+      throw createError({
+        statusCode: 403,
+        statusMessage: '权限不足'
+      })
+    }
+
+    const resetResults = {
+      success: true,
+      message: '数据库重置完成',
+      details: {
+        tablesCleared: 0,
+        recordsDeleted: 0,
+        sequencesReset: 0,
+        errors: [],
+        preservedData: []
+      }
+    }
+
+    try {
+      // 使用事务确保数据一致性
+      await db.transaction(async (tx) => {
+        // 按照外键依赖顺序删除数据
+
+        // 1. API 日志与权限 (依赖于 API 密钥)
+        const deletedApiLogs = await tx.delete(apiLogs).returning({ id: apiLogs.id })
+        resetResults.details.recordsDeleted += deletedApiLogs.length
+        resetResults.details.tablesCleared++
+
+        const deletedApiKeyPermissions = await tx
+          .delete(apiKeyPermissions)
+          .returning({ id: apiKeyPermissions.id })
+        resetResults.details.recordsDeleted += deletedApiKeyPermissions.length
+        resetResults.details.tablesCleared++
+
+        // 2. API 密钥 (依赖于用户)
+        const deletedApiKeys = await tx.delete(apiKeys).returning({ id: apiKeys.id })
+        resetResults.details.recordsDeleted += deletedApiKeys.length
+        resetResults.details.tablesCleared++
+
+        // 3. 通知与设置 (依赖于用户、歌曲)
+        const deletedNotifications = await tx
+          .delete(notifications)
+          .returning({ id: notifications.id })
+        resetResults.details.recordsDeleted += deletedNotifications.length
+        resetResults.details.tablesCleared++
+
+        const deletedNotificationSettings = await tx
+          .delete(notificationSettings)
+          .returning({ id: notificationSettings.id })
+        resetResults.details.recordsDeleted += deletedNotificationSettings.length
+        resetResults.details.tablesCleared++
+
+        // 4. 协作日志 (依赖于歌曲协作者)
+        const deletedCollaborationLogs = await tx
+          .delete(collaborationLogs)
+          .returning({ id: collaborationLogs.id })
+        resetResults.details.recordsDeleted += deletedCollaborationLogs.length
+        resetResults.details.tablesCleared++
+
+        // 5. 歌曲协作者 (依赖于歌曲、用户)
+        const deletedSongCollaborators = await tx
+          .delete(songCollaborators)
+          .returning({ id: songCollaborators.id })
+        resetResults.details.recordsDeleted += deletedSongCollaborators.length
+        resetResults.details.tablesCleared++
+
+        // 6. 歌曲重播申请 (依赖于歌曲、用户)
+        const deletedReplayRequests = await tx
+          .delete(songReplayRequests)
+          .returning({ id: songReplayRequests.id })
+        resetResults.details.recordsDeleted += deletedReplayRequests.length
+        resetResults.details.tablesCleared++
+
+        // 7. 排期 (依赖于歌曲、播出时段)
+        const deletedSchedules = await tx.delete(schedules).returning({ id: schedules.id })
+        resetResults.details.recordsDeleted += deletedSchedules.length
+        resetResults.details.tablesCleared++
+
+        // 8. 投票 (依赖于歌曲、用户)
+        const deletedVotes = await tx.delete(votes).returning({ id: votes.id })
+        resetResults.details.recordsDeleted += deletedVotes.length
+        resetResults.details.tablesCleared++
+
+        // 9. 歌曲 (依赖于用户、播出时段)
+        const deletedSongs = await tx.delete(songs).returning({ id: songs.id })
+        resetResults.details.recordsDeleted += deletedSongs.length
+        resetResults.details.tablesCleared++
+
+        // 10. 黑名单 (依赖于用户 [创建者])
+        const deletedBlacklists = await tx
+          .delete(songBlacklists)
+          .returning({ id: songBlacklists.id })
+        resetResults.details.recordsDeleted += deletedBlacklists.length
+        resetResults.details.tablesCleared++
+
+        // 11. 用户状态日志 (依赖于用户)
+        const deletedStatusLogs = await tx
+          .delete(userStatusLogs)
+          .returning({ id: userStatusLogs.id })
+        resetResults.details.recordsDeleted += deletedStatusLogs.length
+        resetResults.details.tablesCleared++
+
+        // 12. 第三方身份关联 (依赖于用户)
+        const deletedUserIdentities = await tx
+          .delete(userIdentities)
+          .returning({ id: userIdentities.id })
+        resetResults.details.recordsDeleted += deletedUserIdentities.length
+        resetResults.details.tablesCleared++
+
+        // 13. 邮件模板 (依赖于用户 [更新者])
+        const deletedEmailTemplates = await tx
+          .delete(emailTemplates)
+          .returning({ id: emailTemplates.id })
+        resetResults.details.recordsDeleted += deletedEmailTemplates.length
+        resetResults.details.tablesCleared++
+
+        // 14. 用户 (除自己外)
+        const deletedUsers = await tx
+          .delete(users)
+          .where(ne(users.id, user.id))
+          .returning({ id: users.id })
+        resetResults.details.recordsDeleted += deletedUsers.length
+        resetResults.details.tablesCleared++
+        resetResults.details.preservedData.push(`保留当前管理员账户: ${user.name}`)
+
+        // 15. 播出时段
+        const deletedPlayTimes = await tx.delete(playTimes).returning({ id: playTimes.id })
+        resetResults.details.recordsDeleted += deletedPlayTimes.length
+        resetResults.details.tablesCleared++
+
+        // 16. 学期
+        const deletedSemesters = await tx.delete(semesters).returning({ id: semesters.id })
+        resetResults.details.recordsDeleted += deletedSemesters.length
+        resetResults.details.tablesCleared++
+
+        // 17. 请求时段
+        const deletedRequestTimes = await tx.delete(requestTimes).returning({ id: requestTimes.id })
+        resetResults.details.recordsDeleted += deletedRequestTimes.length
+        resetResults.details.tablesCleared++
+
+        // 18. 系统设置
+        const deletedSystemSettings = await tx
+          .delete(systemSettings)
+          .returning({ id: systemSettings.id })
+        resetResults.details.recordsDeleted += deletedSystemSettings.length
+        resetResults.details.tablesCleared++
+      })
+
+      // 重置自增序列
+      const sequenceResults = await resetAutoIncrementSequences()
+      resetResults.details.sequencesReset = sequenceResults.filter((r) => r.success).length
+
+      // 清除缓存
+      try {
+        const cacheService = CacheService.getInstance()
+        await cacheService.clearAllCache()
+      } catch (error) {
+        console.warn('清除缓存失败:', error)
+        resetResults.details.warnings = resetResults.details.warnings || []
+        resetResults.details.warnings.push('清除缓存失败')
+      }
+    } catch (error) {
+      console.error('重置数据库失败:', error)
+      throw createError({
+        statusCode: 500,
+        statusMessage: `重置数据库失败: ${error.message}`
+      })
+    }
+
+    return resetResults
+  } catch (error) {
+    throw createError({
+      statusCode: error.statusCode || 500,
+      statusMessage: error.statusMessage || '服务器内部错误'
+    })
+  }
+})
@@ -0,0 +1,57 @@
+import { verifyAdminAuth } from '~~/server/utils/auth'
+import { databaseManager } from '~~/server/utils/database-manager'
+
+export default defineEventHandler(async (event) => {
+  try {
+    // 验证管理员权限
+    const authResult = await verifyAdminAuth(event)
+
+    if (!authResult.success) {
+      throw createError({
+        statusCode: 401,
+        statusMessage: authResult.message
+      })
+    }
+
+    // 使用改进的数据库管理器获取状态
+    const healthCheck = await databaseManager.healthCheck()
+
+    // 根据延迟确定状态
+    let status = 'error'
+    let message = 'Database connection failed'
+
+    if (healthCheck.status) {
+      if (healthCheck.latency < 100) {
+        status = 'healthy'
+        message = 'Database connection is healthy'
+      } else if (healthCheck.latency < 500) {
+        status = 'slow'
+        message = 'Database connection is slow'
+      } else {
+        status = 'error'
+        message = 'Database connection timeout'
+      }
+    } else {
+      message = healthCheck.error || 'Database connection failed'
+    }
+
+    return {
+      success: healthCheck.status,
+      status: status,
+      message: message,
+      responseTime: healthCheck.latency,
+      timestamp: healthCheck.timestamp.toISOString(),
+      error: healthCheck.error
+    }
+  } catch (error: any) {
+    if (error.statusCode === 401) {
+      throw error
+    }
+
+    console.error('Database status API error:', error)
+    throw createError({
+      statusCode: 500,
+      statusMessage: 'Internal Server Error'
+    })
+  }
+})