feat: align with upstream 1.5.9.9
This commit is contained in:
@@ -0,0 +1,57 @@
|
||||
import { db, eq, requestTimes } from '~/drizzle/db'
|
||||
import { and, gt, lte } from 'drizzle-orm'
|
||||
import { getBeijingTimeISOString } from '~/utils/timeUtils'
|
||||
import { getSystemSettingsCached } from '~~/server/utils/system-settings-helper'
|
||||
|
||||
export default defineEventHandler(async (event) => {
|
||||
try {
|
||||
const settings = await getSystemSettingsCached()
|
||||
let enabled = settings?.enableRequestTimeLimitation || false
|
||||
const forceBlockAllRequests = settings?.forceBlockAllRequests || false
|
||||
let hit = false
|
||||
let accepted = 0
|
||||
let expected = 0
|
||||
|
||||
if (enabled) {
|
||||
const currentTime = getBeijingTimeISOString()
|
||||
|
||||
const hitRequestTimeResult = await db
|
||||
.select()
|
||||
.from(requestTimes)
|
||||
.where(
|
||||
and(
|
||||
and(lte(requestTimes.startTime, currentTime), gt(requestTimes.endTime, currentTime)),
|
||||
eq(requestTimes.enabled, true)
|
||||
)
|
||||
)
|
||||
.limit(1)
|
||||
const hitRequestTime = hitRequestTimeResult[0]
|
||||
|
||||
if (hitRequestTime) {
|
||||
hit = true
|
||||
accepted = hitRequestTime?.accepted || 0
|
||||
expected = hitRequestTime?.expected || 0
|
||||
}
|
||||
} else {
|
||||
hit = true
|
||||
}
|
||||
|
||||
if (forceBlockAllRequests) {
|
||||
hit = false
|
||||
enabled = true
|
||||
}
|
||||
|
||||
return {
|
||||
hit,
|
||||
enabled,
|
||||
accepted,
|
||||
expected
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('获取播出时段失败:', error)
|
||||
throw createError({
|
||||
statusCode: 500,
|
||||
message: '获取播出时段失败'
|
||||
})
|
||||
}
|
||||
})
|
||||
Reference in New Issue
Block a user