152 lines
4.0 KiB
TypeScript
152 lines
4.0 KiB
TypeScript
import { db, testConnection } from '~/drizzle/db'
|
|
import { sql } from 'drizzle-orm'
|
|
|
|
// 数据库健康检查
|
|
export async function checkDatabaseHealth() {
|
|
try {
|
|
const startTime = Date.now()
|
|
|
|
// 基本连接测试
|
|
const isConnected = await testConnection()
|
|
if (!isConnected) {
|
|
return {
|
|
status: 'unhealthy',
|
|
message: 'Database connection failed',
|
|
responseTime: Date.now() - startTime
|
|
}
|
|
}
|
|
|
|
// 查询测试
|
|
await db.execute(sql`SELECT 1 as test`)
|
|
|
|
const responseTime = Date.now() - startTime
|
|
|
|
return {
|
|
status: responseTime < 1000 ? 'healthy' : 'slow',
|
|
message: responseTime < 1000 ? 'Database is healthy' : 'Database is responding slowly',
|
|
responseTime
|
|
}
|
|
} catch (error) {
|
|
return {
|
|
status: 'error',
|
|
message: error instanceof Error ? error.message : 'Unknown database error',
|
|
responseTime: 0
|
|
}
|
|
}
|
|
}
|
|
|
|
// 数据库连接重试机制
|
|
export async function retryDatabaseOperation<T>(
|
|
operation: () => Promise<T>,
|
|
maxRetries: number = 3,
|
|
delay: number = 1000
|
|
): Promise<T> {
|
|
let lastError: Error
|
|
|
|
for (let attempt = 1; attempt <= maxRetries; attempt++) {
|
|
try {
|
|
return await operation()
|
|
} catch (error) {
|
|
lastError = error instanceof Error ? error : new Error('Unknown error')
|
|
|
|
if (attempt === maxRetries) {
|
|
throw lastError
|
|
}
|
|
|
|
// 等待后重试
|
|
await new Promise((resolve) => setTimeout(resolve, delay * attempt))
|
|
}
|
|
}
|
|
|
|
throw lastError!
|
|
}
|
|
|
|
// 数据库连接池状态检查
|
|
export async function getConnectionPoolStatus() {
|
|
try {
|
|
const result = await db.execute(sql`
|
|
SELECT
|
|
setting::int as max_connections
|
|
FROM pg_settings
|
|
WHERE name = 'max_connections'
|
|
`)
|
|
|
|
const activeConnections = await db.execute(sql`
|
|
SELECT count(*) as active_connections
|
|
FROM pg_stat_activity
|
|
WHERE state = 'active'
|
|
`)
|
|
|
|
const totalConnections = await db.execute(sql`
|
|
SELECT count(*) as total_connections
|
|
FROM pg_stat_activity
|
|
`)
|
|
|
|
return {
|
|
maxConnections: result[0]?.max_connections || 0,
|
|
activeConnections: activeConnections[0]?.active_connections || 0,
|
|
totalConnections: totalConnections[0]?.total_connections || 0,
|
|
utilization: result[0]?.max_connections
|
|
? (
|
|
((totalConnections[0]?.total_connections || 0) / result[0].max_connections) *
|
|
100
|
|
).toFixed(2)
|
|
: '0'
|
|
}
|
|
} catch (error) {
|
|
throw new Error(
|
|
`Failed to get connection pool status: ${error instanceof Error ? error.message : 'Unknown error'}`
|
|
)
|
|
}
|
|
}
|
|
|
|
// 数据库性能指标
|
|
export async function getDatabaseMetrics() {
|
|
try {
|
|
const startTime = Date.now()
|
|
|
|
// 测试查询响应时间
|
|
await db.execute(sql`SELECT 1`)
|
|
const responseTime = Date.now() - startTime
|
|
|
|
// 获取数据库统计信息
|
|
const stats = await db.execute(sql`
|
|
SELECT
|
|
numbackends as active_connections,
|
|
xact_commit as transactions_committed,
|
|
xact_rollback as transactions_rolled_back,
|
|
blks_read as blocks_read,
|
|
blks_hit as blocks_hit
|
|
FROM pg_stat_database
|
|
WHERE datname = current_database()
|
|
`)
|
|
|
|
const stat = stats[0] || {}
|
|
|
|
return {
|
|
responseTime,
|
|
activeConnections: stat.active_connections || 0,
|
|
transactionsCommitted: stat.transactions_committed || 0,
|
|
transactionsRolledBack: stat.transactions_rolled_back || 0,
|
|
cacheHitRatio:
|
|
stat.blocks_read && stat.blocks_hit
|
|
? ((stat.blocks_hit / (stat.blocks_hit + stat.blocks_read)) * 100).toFixed(2)
|
|
: '0'
|
|
}
|
|
} catch (error) {
|
|
throw new Error(
|
|
`Failed to get database metrics: ${error instanceof Error ? error.message : 'Unknown error'}`
|
|
)
|
|
}
|
|
}
|
|
|
|
// 数据库备份状态检查（模拟）
|
|
export async function checkBackupStatus() {
|
|
// 这里可以集成实际的备份服务
|
|
return {
|
|
lastBackup: new Date().toISOString(),
|
|
status: 'completed',
|
|
size: '0 MB' // 模拟数据
|
|
}
|
|
}
|