feat: task scheduler; feat: ffmpeg compress webp to avif
This commit is contained in:
parent
81d73cf17c
commit
f34ce4aed0
85
app/api/tasks/route.ts
Normal file
85
app/api/tasks/route.ts
Normal file
@ -0,0 +1,85 @@
|
||||
import { NextRequest, NextResponse } from 'next/server';
|
||||
import { taskScheduler } from '../../../lib/scheduler';
|
||||
|
||||
export async function GET(request: NextRequest) {
|
||||
try {
|
||||
const url = new URL(request.url);
|
||||
const action = url.searchParams.get('action');
|
||||
const taskName = url.searchParams.get('task');
|
||||
|
||||
switch (action) {
|
||||
case 'status':
|
||||
if (taskName) {
|
||||
const status = taskScheduler.getTaskStatus(taskName);
|
||||
return NextResponse.json({
|
||||
task: taskName,
|
||||
running: status
|
||||
});
|
||||
} else {
|
||||
const tasks = taskScheduler.listTasks();
|
||||
const statuses = tasks.map(task => ({
|
||||
name: task,
|
||||
running: taskScheduler.getTaskStatus(task)
|
||||
}));
|
||||
return NextResponse.json({ tasks: statuses });
|
||||
}
|
||||
|
||||
case 'list':
|
||||
const allTasks = taskScheduler.listTasks();
|
||||
return NextResponse.json({ tasks: allTasks });
|
||||
|
||||
default:
|
||||
return NextResponse.json({
|
||||
message: 'Task scheduler is running',
|
||||
availableActions: ['status', 'list'],
|
||||
tasks: taskScheduler.listTasks()
|
||||
});
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('Task scheduler API error:', error);
|
||||
return NextResponse.json(
|
||||
{ error: 'Internal server error' },
|
||||
{ status: 500 }
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
export async function POST(request: NextRequest) {
|
||||
try {
|
||||
const { action, taskName } = await request.json();
|
||||
|
||||
switch (action) {
|
||||
case 'stop':
|
||||
if (taskName) {
|
||||
taskScheduler.stopTask(taskName);
|
||||
return NextResponse.json({
|
||||
message: `Task ${taskName} stopped`
|
||||
});
|
||||
} else {
|
||||
taskScheduler.stopAll();
|
||||
return NextResponse.json({
|
||||
message: 'All tasks stopped'
|
||||
});
|
||||
}
|
||||
|
||||
case 'start':
|
||||
// 重新启动所有任务
|
||||
taskScheduler.startAll();
|
||||
return NextResponse.json({
|
||||
message: 'All tasks started'
|
||||
});
|
||||
|
||||
default:
|
||||
return NextResponse.json(
|
||||
{ error: 'Invalid action' },
|
||||
{ status: 400 }
|
||||
);
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('Task scheduler API error:', error);
|
||||
return NextResponse.json(
|
||||
{ error: 'Internal server error' },
|
||||
{ status: 500 }
|
||||
);
|
||||
}
|
||||
}
|
||||
@ -3,6 +3,10 @@ import { prisma } from '@/lib/prisma'
|
||||
import { storeFile } from '@/lib/fileStorage'
|
||||
import { push } from '@/lib/push'
|
||||
import { withCors } from '@/lib/middleware'
|
||||
import ffmpeg from 'fluent-ffmpeg'
|
||||
import { writeFileSync, unlinkSync, mkdtempSync, readFileSync, rmSync } from 'fs'
|
||||
import { tmpdir } from 'os'
|
||||
import { join } from 'path'
|
||||
|
||||
interface WindowInfo {
|
||||
title: string
|
||||
@ -38,23 +42,78 @@ async function handleScreenshotUpload(req: NextRequest) {
|
||||
}
|
||||
|
||||
// Process screenshots
|
||||
const processingStartTime = Date.now()
|
||||
console.log(`[${hostname}] 开始处理 ${files.length} 张截图`)
|
||||
|
||||
const screenshots = await Promise.all(files.map(async (file, index) => {
|
||||
const ext = file.name ? file.name.split('.').pop() : 'webp'
|
||||
const filename = `${hostname}-${Date.now()}-${Math.round(Math.random() * 1E9)}.${ext}`
|
||||
const imageStartTime = Date.now()
|
||||
|
||||
// Convert image to AV1 format using FFmpeg
|
||||
const buffer = Buffer.from(await file.arrayBuffer())
|
||||
const storedFile = await storeFile(buffer, filename, file.type, 'screenshot', hostname)
|
||||
const originalSize = buffer.length
|
||||
|
||||
return {
|
||||
fileId: storedFile.id,
|
||||
objectName: storedFile.objectName,
|
||||
filename: storedFile.filename,
|
||||
contentType: storedFile.contentType,
|
||||
fileSize: storedFile.size,
|
||||
monitorName: formData.get(`monitor_name_${index}`) as string || `Monitor ${index + 1}`
|
||||
// Create temporary files
|
||||
const tempDir = mkdtempSync(join(tmpdir(), 'ffmpeg-'))
|
||||
const inputPath = join(tempDir, `input_${index}.png`)
|
||||
const outputPath = join(tempDir, `output_${index}.avif`)
|
||||
|
||||
try {
|
||||
// Write input file
|
||||
writeFileSync(inputPath, buffer)
|
||||
|
||||
const conversionStartTime = Date.now()
|
||||
|
||||
// Convert using FFmpeg with libsvtav1
|
||||
await new Promise<void>((resolve, reject) => {
|
||||
ffmpeg(inputPath)
|
||||
.videoCodec('libsvtav1')
|
||||
.outputOptions([
|
||||
'-crf 30', // 质量控制，值越小质量越好
|
||||
'-preset 6', // 编码速度预设，0-12，值越大速度越快但质量可能下降
|
||||
'-pix_fmt yuv420p' // 像素格式
|
||||
])
|
||||
.output(outputPath)
|
||||
.on('end', () => resolve())
|
||||
.on('error', (err) => reject(err))
|
||||
.run()
|
||||
})
|
||||
|
||||
const conversionTime = Date.now() - conversionStartTime
|
||||
|
||||
// Read the converted file
|
||||
const avifBuffer = readFileSync(outputPath)
|
||||
const filename = `${hostname}-${Date.now()}-${Math.round(Math.random() * 1E9)}.avif`
|
||||
|
||||
const storageStartTime = Date.now()
|
||||
const storedFile = await storeFile(avifBuffer, filename, 'image/avif', 'screenshot', hostname)
|
||||
const storageTime = Date.now() - storageStartTime
|
||||
|
||||
const totalImageTime = Date.now() - imageStartTime
|
||||
const compressionRatio = ((originalSize - avifBuffer.length) / originalSize * 100).toFixed(2)
|
||||
|
||||
console.log(`[${hostname}] 图片 ${index + 1}: FFmpeg转换 ${conversionTime}ms, 存储 ${storageTime}ms, 总计 ${totalImageTime}ms, 压缩率 ${compressionRatio}% (${originalSize} -> ${avifBuffer.length} bytes)`)
|
||||
|
||||
return {
|
||||
fileId: storedFile.id,
|
||||
objectName: storedFile.objectName,
|
||||
filename: storedFile.filename,
|
||||
contentType: storedFile.contentType,
|
||||
fileSize: storedFile.size,
|
||||
monitorName: formData.get(`monitor_name_${index}`) as string || `Monitor ${index + 1}`
|
||||
}
|
||||
} finally {
|
||||
// Clean up temporary files
|
||||
try {
|
||||
rmSync(tempDir, { recursive: true, force: true })
|
||||
} catch (err) {
|
||||
console.warn(`清理临时文件失败: ${err}`)
|
||||
}
|
||||
}
|
||||
}))
|
||||
|
||||
const totalProcessingTime = Date.now() - processingStartTime
|
||||
console.log(`[${hostname}] 完成处理 ${files.length} 张截图，总耗时 ${totalProcessingTime}ms，平均每张 ${(totalProcessingTime / files.length).toFixed(2)}ms`)
|
||||
|
||||
// Ensure host exists first
|
||||
await prisma.host.upsert({
|
||||
where: { hostname },
|
||||
|
||||
18
app/page.tsx
18
app/page.tsx
@ -51,7 +51,23 @@ export default function Home() {
|
||||
<div className="min-h-screen bg-gray-50 dark:bg-gray-900">
|
||||
<div className="max-w-7xl mx-auto py-6 sm:px-6 lg:px-8">
|
||||
<div className="px-4 py-6 sm:px-0">
|
||||
<h1 className="text-3xl font-semibold text-gray-900 dark:text-white mb-8">屏幕截图监控系统</h1>
|
||||
<div className="flex justify-between items-center mb-8">
|
||||
<h1 className="text-3xl font-semibold text-gray-900 dark:text-white">屏幕截图监控系统</h1>
|
||||
<div className="space-x-4">
|
||||
<button
|
||||
onClick={() => router.push('/tasks')}
|
||||
className="px-4 py-2 bg-blue-600 text-white rounded-md hover:bg-blue-700 transition-colors"
|
||||
>
|
||||
定时任务管理
|
||||
</button>
|
||||
<button
|
||||
onClick={() => router.push('/upload')}
|
||||
className="px-4 py-2 bg-green-600 text-white rounded-md hover:bg-green-700 transition-colors"
|
||||
>
|
||||
上传新版本客户端
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{/* 主机列表卡片网格 */}
|
||||
<div className="grid grid-cols-1 sm:grid-cols-2 lg:grid-cols-3 gap-6">
|
||||
|
||||
160
app/tasks/page.tsx
Normal file
160
app/tasks/page.tsx
Normal file
@ -0,0 +1,160 @@
|
||||
'use client';
|
||||
|
||||
import { useState, useEffect } from 'react';
|
||||
|
||||
interface TaskStatus {
|
||||
name: string;
|
||||
running: boolean;
|
||||
}
|
||||
|
||||
export default function TasksPage() {
|
||||
const [tasks, setTasks] = useState<TaskStatus[]>([]);
|
||||
const [loading, setLoading] = useState(true);
|
||||
const [message, setMessage] = useState('');
|
||||
|
||||
useEffect(() => {
|
||||
fetchTasks();
|
||||
}, []);
|
||||
|
||||
const fetchTasks = async () => {
|
||||
try {
|
||||
const response = await fetch('/api/tasks?action=status');
|
||||
const data = await response.json();
|
||||
setTasks(data.tasks || []);
|
||||
} catch (error) {
|
||||
console.error('Failed to fetch tasks:', error);
|
||||
setMessage('获取任务状态失败');
|
||||
} finally {
|
||||
setLoading(false);
|
||||
}
|
||||
};
|
||||
|
||||
const handleAction = async (action: string, taskName?: string) => {
|
||||
try {
|
||||
const response = await fetch('/api/tasks', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({ action, taskName }),
|
||||
});
|
||||
|
||||
const data = await response.json();
|
||||
setMessage(data.message || '操作完成');
|
||||
|
||||
// 刷新任务状态
|
||||
await fetchTasks();
|
||||
} catch (error) {
|
||||
console.error('Failed to perform action:', error);
|
||||
setMessage('操作失败');
|
||||
}
|
||||
};
|
||||
|
||||
if (loading) {
|
||||
return (
|
||||
<div className="container mx-auto px-4 py-8">
|
||||
<div className="text-center">加载中...</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="container mx-auto px-4 py-8">
|
||||
<h1 className="text-2xl font-bold mb-6">定时任务管理</h1>
|
||||
|
||||
{message && (
|
||||
<div className="mb-4 p-4 bg-blue-100 border border-blue-300 rounded">
|
||||
{message}
|
||||
</div>
|
||||
)}
|
||||
|
||||
<div className="bg-white dark:bg-gray-800 rounded-lg shadow p-6">
|
||||
<div className="flex justify-between items-center mb-4">
|
||||
<h2 className="text-lg font-semibold">任务列表</h2>
|
||||
<div className="space-x-2">
|
||||
<button
|
||||
onClick={() => handleAction('start')}
|
||||
className="px-4 py-2 bg-green-500 text-white rounded hover:bg-green-600"
|
||||
>
|
||||
启动所有任务
|
||||
</button>
|
||||
<button
|
||||
onClick={() => handleAction('stop')}
|
||||
className="px-4 py-2 bg-red-500 text-white rounded hover:bg-red-600"
|
||||
>
|
||||
停止所有任务
|
||||
</button>
|
||||
<button
|
||||
onClick={fetchTasks}
|
||||
className="px-4 py-2 bg-blue-500 text-white rounded hover:bg-blue-600"
|
||||
>
|
||||
刷新状态
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{tasks.length === 0 ? (
|
||||
<div className="text-center py-8 text-gray-500">
|
||||
没有找到定时任务
|
||||
</div>
|
||||
) : (
|
||||
<div className="overflow-x-auto">
|
||||
<table className="w-full border-collapse border border-gray-300">
|
||||
<thead>
|
||||
<tr className="bg-gray-50 dark:bg-gray-700">
|
||||
<th className="border border-gray-300 px-4 py-2 text-left">任务名称</th>
|
||||
<th className="border border-gray-300 px-4 py-2 text-left">状态</th>
|
||||
<th className="border border-gray-300 px-4 py-2 text-left">描述</th>
|
||||
<th className="border border-gray-300 px-4 py-2 text-left">操作</th>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
{tasks.map((task) => (
|
||||
<tr key={task.name} className="hover:bg-gray-50 dark:hover:bg-gray-700">
|
||||
<td className="border border-gray-300 px-4 py-2 font-medium">
|
||||
{task.name}
|
||||
</td>
|
||||
<td className="border border-gray-300 px-4 py-2">
|
||||
<span className={`px-2 py-1 rounded text-sm ${
|
||||
task.running
|
||||
? 'bg-green-100 text-green-800'
|
||||
: 'bg-red-100 text-red-800'
|
||||
}`}>
|
||||
{task.running ? '运行中' : '已停止'}
|
||||
</span>
|
||||
</td>
|
||||
<td className="border border-gray-300 px-4 py-2">
|
||||
{task.name === 'hourlyTask' && '每小时第30分钟执行'}
|
||||
{task.name === 'dailyCleanup' && '每日凌晨2点清理任务'}
|
||||
</td>
|
||||
<td className="border border-gray-300 px-4 py-2">
|
||||
<button
|
||||
onClick={() => handleAction('stop', task.name)}
|
||||
className="px-3 py-1 bg-red-500 text-white rounded text-sm hover:bg-red-600"
|
||||
disabled={!task.running}
|
||||
>
|
||||
停止
|
||||
</button>
|
||||
</td>
|
||||
</tr>
|
||||
))}
|
||||
</tbody>
|
||||
</table>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
|
||||
<div className="mt-8 bg-white dark:bg-gray-800 rounded-lg shadow p-6">
|
||||
<h2 className="text-lg font-semibold mb-4">Cron 表达式说明</h2>
|
||||
<div className="space-y-2 text-sm text-gray-600 dark:text-gray-400">
|
||||
<div><code>30 * * * *</code> - 每小时的第30分钟执行</div>
|
||||
<div><code>0 2 * * *</code> - 每天凌晨2点执行</div>
|
||||
<div><code>0 */2 * * *</code> - 每2小时执行一次</div>
|
||||
<div><code>0 9 * * 1-5</code> - 周一到周五上午9点执行</div>
|
||||
<div><code>0 0 1 * *</code> - 每月1号凌晨执行</div>
|
||||
<div><code>*/15 * * * *</code> - 每15分钟执行一次</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
20
bun.lock
20
bun.lock
@ -7,16 +7,19 @@
|
||||
"@prisma/client": "^6.10.1",
|
||||
"@types/bcryptjs": "^3.0.0",
|
||||
"@types/cors": "^2.8.19",
|
||||
"@types/fluent-ffmpeg": "^2.1.27",
|
||||
"@types/minio": "^7.1.1",
|
||||
"@types/multer": "^1.4.13",
|
||||
"bcryptjs": "^3.0.2",
|
||||
"cors": "^2.8.5",
|
||||
"date-fns": "^4.1.0",
|
||||
"dotenv-cli": "^8.0.0",
|
||||
"fluent-ffmpeg": "^2.1.3",
|
||||
"lucide-react": "^0.525.0",
|
||||
"minio": "^8.0.5",
|
||||
"multer": "^2.0.1",
|
||||
"next": "15.3.4",
|
||||
"node-cron": "^4.1.1",
|
||||
"prisma": "^6.10.1",
|
||||
"react": "^19.0.0",
|
||||
"react-dom": "^19.0.0",
|
||||
@ -24,6 +27,7 @@
|
||||
"devDependencies": {
|
||||
"@tailwindcss/postcss": "^4",
|
||||
"@types/node": "^20",
|
||||
"@types/node-cron": "^3.0.11",
|
||||
"@types/react": "^19",
|
||||
"@types/react-dom": "^19",
|
||||
"tailwindcss": "^4",
|
||||
@ -170,6 +174,8 @@
|
||||
|
||||
"@types/express-serve-static-core": ["@types/express-serve-static-core@5.0.6", "", { "dependencies": { "@types/node": "*", "@types/qs": "*", "@types/range-parser": "*", "@types/send": "*" } }, "sha512-3xhRnjJPkULekpSzgtoNYYcTWgEZkp4myc+Saevii5JPnHNvHMRlBSHDbs7Bh1iPPoVTERHEZXyhyLbMEsExsA=="],
|
||||
|
||||
"@types/fluent-ffmpeg": ["@types/fluent-ffmpeg@2.1.27", "https://registry.npmmirror.com/@types/fluent-ffmpeg/-/fluent-ffmpeg-2.1.27.tgz", { "dependencies": { "@types/node": "*" } }, "sha512-QiDWjihpUhriISNoBi2hJBRUUmoj/BMTYcfz+F+ZM9hHWBYABFAE6hjP/TbCZC0GWwlpa3FzvHH9RzFeRusZ7A=="],
|
||||
|
||||
"@types/http-errors": ["@types/http-errors@2.0.5", "", {}, "sha512-r8Tayk8HJnX0FztbZN7oVqGccWgw98T/0neJphO91KkmOzug1KkofZURD4UaD5uH8AqcFLfdPErnBod0u71/qg=="],
|
||||
|
||||
"@types/mime": ["@types/mime@1.3.5", "", {}, "sha512-/pyBZWSLD2n0dcHE3hq8s8ZvcETHtEuF+3E7XVt0Ig2nvsVQXdghHVcEkIWjy9A0wKfTn97a/PSDYohKIlnP/w=="],
|
||||
@ -180,6 +186,8 @@
|
||||
|
||||
"@types/node": ["@types/node@20.19.1", "", { "dependencies": { "undici-types": "~6.21.0" } }, "sha512-jJD50LtlD2dodAEO653i3YF04NWak6jN3ky+Ri3Em3mGR39/glWiboM/IePaRbgwSfqM1TpGXfAg8ohn/4dTgA=="],
|
||||
|
||||
"@types/node-cron": ["@types/node-cron@3.0.11", "", {}, "sha512-0ikrnug3/IyneSHqCBeslAhlK2aBfYek1fGo4bP4QnZPmiqSGRK+Oy7ZMisLWkesffJvQ1cqAcBnJC+8+nxIAg=="],
|
||||
|
||||
"@types/qs": ["@types/qs@6.14.0", "", {}, "sha512-eOunJqu0K1923aExK6y8p6fsihYEn/BYuQ4g0CxAAgFc4b/ZLN4CrsRZ55srTdqoiLzU2B2evC+apEIxprEzkQ=="],
|
||||
|
||||
"@types/range-parser": ["@types/range-parser@1.2.7", "", {}, "sha512-hKormJbkJqzQGhziax5PItDUTMAM9uE2XXQmM37dyd4hVM+5aVl7oVxMVUiVQn2oCQFN/LKCZdvSM0pFRqbSmQ=="],
|
||||
@ -196,7 +204,7 @@
|
||||
|
||||
"append-field": ["append-field@1.0.0", "", {}, "sha512-klpgFSWLW1ZEs8svjfb7g4qWY0YS5imI82dTg+QahUvJ8YqAY0P10Uk8tTyh9ZGuYEZEMaeJYCF5BFuX552hsw=="],
|
||||
|
||||
"async": ["async@3.2.6", "", {}, "sha512-htCUDlxyyCLMgaM3xXg0C0LW2xqfuQ6p05pCEIsXuyQ+a1koYKTuBMzRNwmybfLgvJDMd0r1LTn4+E0Ti6C2AA=="],
|
||||
"async": ["async@0.2.10", "https://registry.npmmirror.com/async/-/async-0.2.10.tgz", {}, "sha512-eAkdoKxU6/LkKDBzLpT+t6Ff5EtfSF4wx1WfJiPEEV7WNLnDaRXk0oVysiEPm262roaachGexwUv94WhSgN5TQ=="],
|
||||
|
||||
"available-typed-arrays": ["available-typed-arrays@1.0.7", "", { "dependencies": { "possible-typed-array-names": "^1.0.0" } }, "sha512-wvUjBtSGN7+7SjNpq/9M2Tg350UZD3q62IFZLbRAR1bSMlCo1ZaeW+BJ+D090e4hIIZLBcTDWe4Mh4jvUDajzQ=="],
|
||||
|
||||
@ -270,6 +278,8 @@
|
||||
|
||||
"filter-obj": ["filter-obj@1.1.0", "", {}, "sha512-8rXg1ZnX7xzy2NGDVkBVaAy+lSlPNwad13BtgSlLuxfIslyt5Vg64U7tFcCt4WS1R0hvtnQybT/IyCkGZ3DpXQ=="],
|
||||
|
||||
"fluent-ffmpeg": ["fluent-ffmpeg@2.1.3", "https://registry.npmmirror.com/fluent-ffmpeg/-/fluent-ffmpeg-2.1.3.tgz", { "dependencies": { "async": "^0.2.9", "which": "^1.1.1" } }, "sha512-Be3narBNt2s6bsaqP6Jzq91heDgOEaDCJAXcE3qcma/EJBSy5FB4cvO31XBInuAuKBx8Kptf8dkhjK0IOru39Q=="],
|
||||
|
||||
"for-each": ["for-each@0.3.5", "", { "dependencies": { "is-callable": "^1.2.7" } }, "sha512-dKx12eRCVIzqCxFGplyFKJMPvLEWgmNtUrpTiJIR5u97zEhRG8ySrtboPHZXx7daLxQVrl643cTzbab2tkQjxg=="],
|
||||
|
||||
"function-bind": ["function-bind@1.1.2", "", {}, "sha512-7XHNxH7qX9xG5mIwxkhumTox/MIRNcOgDrxWsMt2pAr23WHp6MrRlN7FBSFpCpr+oVO0F744iUgR82nJMfG2SA=="],
|
||||
@ -362,6 +372,8 @@
|
||||
|
||||
"next": ["next@15.3.4", "", { "dependencies": { "@next/env": "15.3.4", "@swc/counter": "0.1.3", "@swc/helpers": "0.5.15", "busboy": "1.6.0", "caniuse-lite": "^1.0.30001579", "postcss": "8.4.31", "styled-jsx": "5.1.6" }, "optionalDependencies": { "@next/swc-darwin-arm64": "15.3.4", "@next/swc-darwin-x64": "15.3.4", "@next/swc-linux-arm64-gnu": "15.3.4", "@next/swc-linux-arm64-musl": "15.3.4", "@next/swc-linux-x64-gnu": "15.3.4", "@next/swc-linux-x64-musl": "15.3.4", "@next/swc-win32-arm64-msvc": "15.3.4", "@next/swc-win32-x64-msvc": "15.3.4", "sharp": "^0.34.1" }, "peerDependencies": { "@opentelemetry/api": "^1.1.0", "@playwright/test": "^1.41.2", "babel-plugin-react-compiler": "*", "react": "^18.2.0 || 19.0.0-rc-de68d2f4-20241204 || ^19.0.0", "react-dom": "^18.2.0 || 19.0.0-rc-de68d2f4-20241204 || ^19.0.0", "sass": "^1.3.0" }, "optionalPeers": ["@opentelemetry/api", "@playwright/test", "babel-plugin-react-compiler", "sass"], "bin": { "next": "dist/bin/next" } }, "sha512-mHKd50C+mCjam/gcnwqL1T1vPx/XQNFlXqFIVdgQdVAFY9iIQtY0IfaVflEYzKiqjeA7B0cYYMaCrmAYFjs4rA=="],
|
||||
|
||||
"node-cron": ["node-cron@4.1.1", "", {}, "sha512-oJj9CYV7teeCVs+y2Efi5IQ4FGmAYbsXQOehc1AGLlwteec8pC7DjBCUzSyRQ0LYa+CRCgmD+vtlWQcnPpXowA=="],
|
||||
|
||||
"object-assign": ["object-assign@4.1.1", "", {}, "sha512-rJgTQnkUnH1sFw8yT6VSU3zD3sWmu6sZhIseY8VX+GRu3P6F7Fu+JNDoXfklElbLJSnc3FUQHVe4cU5hj+BcUg=="],
|
||||
|
||||
"path-key": ["path-key@3.1.1", "", {}, "sha512-ojmeN0qd+y0jszEtoY48r0Peq5dwMEkIlCOu6Q5f41lfkswXuKtYrhgoTpLnyIcHm24Uhqx+5Tqm2InSwLhE6Q=="],
|
||||
@ -446,7 +458,7 @@
|
||||
|
||||
"web-encoding": ["web-encoding@1.1.5", "", { "dependencies": { "util": "^0.12.3" }, "optionalDependencies": { "@zxing/text-encoding": "0.9.0" } }, "sha512-HYLeVCdJ0+lBYV2FvNZmv3HJ2Nt0QYXqZojk3d9FJOLkwnuhzM9tmamh8d7HPM8QqjKH8DeHkFTx+CFlWpZZDA=="],
|
||||
|
||||
"which": ["which@2.0.2", "", { "dependencies": { "isexe": "^2.0.0" }, "bin": { "node-which": "./bin/node-which" } }, "sha512-BLI3Tl1TW3Pvl70l3yq3Y64i+awpwXqsGBYWkkqMtnbXgrMD+yj7rhW0kuEDxzJaYXGjEW5ogapKNMEKNMjibA=="],
|
||||
"which": ["which@1.3.1", "https://registry.npmmirror.com/which/-/which-1.3.1.tgz", { "dependencies": { "isexe": "^2.0.0" }, "bin": { "which": "./bin/which" } }, "sha512-HxJdYWq1MTIQbJ3nw0cqssHoTNU267KlrDuGZ1WYlxDStUtKUhOaJmh112/TZmHxxUfuJqPXSOm7tDyas0OSIQ=="],
|
||||
|
||||
"which-typed-array": ["which-typed-array@1.1.19", "", { "dependencies": { "available-typed-arrays": "^1.0.7", "call-bind": "^1.0.8", "call-bound": "^1.0.4", "for-each": "^0.3.5", "get-proto": "^1.0.1", "gopd": "^1.2.0", "has-tostringtag": "^1.0.2" } }, "sha512-rEvr90Bck4WZt9HHFC4DJMsjvu7x+r6bImz0/BrbWb7A2djJ8hnZMrWnHo9F8ssv0OMErasDhftrfROTyqSDrw=="],
|
||||
|
||||
@ -470,6 +482,10 @@
|
||||
|
||||
"@tailwindcss/oxide-wasm32-wasi/tslib": ["tslib@2.8.1", "", { "bundled": true }, "sha512-oJFu94HQb+KVduSUQL7wnpmqnfmLsOA/nAh6b6EH0wCEoK0/mPeXU6c3wKDV83MkOuHPRHtSXKKU99IBazS/2w=="],
|
||||
|
||||
"cross-spawn/which": ["which@2.0.2", "", { "dependencies": { "isexe": "^2.0.0" }, "bin": { "node-which": "./bin/node-which" } }, "sha512-BLI3Tl1TW3Pvl70l3yq3Y64i+awpwXqsGBYWkkqMtnbXgrMD+yj7rhW0kuEDxzJaYXGjEW5ogapKNMEKNMjibA=="],
|
||||
|
||||
"minio/async": ["async@3.2.6", "", {}, "sha512-htCUDlxyyCLMgaM3xXg0C0LW2xqfuQ6p05pCEIsXuyQ+a1koYKTuBMzRNwmybfLgvJDMd0r1LTn4+E0Ti6C2AA=="],
|
||||
|
||||
"next/postcss": ["postcss@8.4.31", "", { "dependencies": { "nanoid": "^3.3.6", "picocolors": "^1.0.0", "source-map-js": "^1.0.2" } }, "sha512-PS08Iboia9mts/2ygV3eLpY5ghnUcfLV/EXTOW1E2qYxJKGGBUtNjN76FYHnMs36RmARn41bC0AZmn+rR0OVpQ=="],
|
||||
|
||||
"tar/mkdirp": ["mkdirp@3.0.1", "", { "bin": { "mkdirp": "dist/cjs/src/bin.js" } }, "sha512-+NsyUUAZDmo6YVHzL/stxSu3t9YS1iljliy3BSDrXJ/dkn1KYdmtZODGGjLcc9XLgVVpH4KshHB8XmZgMhaBXg=="],
|
||||
|
||||
16
lib/init-scheduler.ts
Normal file
16
lib/init-scheduler.ts
Normal file
@ -0,0 +1,16 @@
|
||||
import { taskScheduler } from './scheduler';
|
||||
|
||||
// 确保定时任务只初始化一次
|
||||
let isInitialized = false;
|
||||
|
||||
export function initializeScheduler() {
|
||||
if (!isInitialized && typeof window === 'undefined') {
|
||||
// 只在服务端初始化
|
||||
taskScheduler.startAll();
|
||||
isInitialized = true;
|
||||
console.log('Task scheduler initialized');
|
||||
}
|
||||
}
|
||||
|
||||
// 在模块加载时自动初始化
|
||||
initializeScheduler();
|
||||
121
lib/scheduler.ts
Normal file
121
lib/scheduler.ts
Normal file
@ -0,0 +1,121 @@
|
||||
import * as cron from 'node-cron';
|
||||
import { prisma } from './prisma';
|
||||
|
||||
// 定时任务管理器
|
||||
class TaskScheduler {
|
||||
private tasks: Map<string, cron.ScheduledTask> = new Map();
|
||||
|
||||
// 启动所有定时任务
|
||||
startAll() {
|
||||
this.scheduleHourlyTask();
|
||||
this.scheduleDailyCleanup();
|
||||
console.log('All scheduled tasks started');
|
||||
}
|
||||
|
||||
// 每小时第30分钟执行的任务
|
||||
private scheduleHourlyTask() {
|
||||
const task = cron.schedule('30 * * * *', async () => {
|
||||
console.log('Running hourly task at minute 30...');
|
||||
try {
|
||||
await this.hourlyTaskHandler();
|
||||
} catch (error) {
|
||||
console.error('Hourly task failed:', error);
|
||||
}
|
||||
});
|
||||
|
||||
this.tasks.set('hourlyTask', task);
|
||||
console.log('Hourly task scheduled (every hour at minute 30)');
|
||||
}
|
||||
|
||||
// 每日凌晨2点清理任务
|
||||
private scheduleDailyCleanup() {
|
||||
const task = cron.schedule('0 2 * * *', async () => {
|
||||
console.log('Running daily cleanup task...');
|
||||
try {
|
||||
await this.dailyCleanupHandler();
|
||||
} catch (error) {
|
||||
console.error('Daily cleanup task failed:', error);
|
||||
}
|
||||
});
|
||||
|
||||
this.tasks.set('dailyCleanup', task);
|
||||
console.log('Daily cleanup task scheduled (2:00 AM every day)');
|
||||
}
|
||||
|
||||
// 每小时执行的具体任务逻辑
|
||||
private async hourlyTaskHandler() {
|
||||
// 在这里添加你的业务逻辑
|
||||
// 例如：清理过期数据、同步数据、发送通知等
|
||||
|
||||
// 示例：清理30天前的记录
|
||||
const thirtyDaysAgo = new Date();
|
||||
thirtyDaysAgo.setDate(thirtyDaysAgo.getDate() - 30);
|
||||
|
||||
// 根据你的数据库结构调整这个查询
|
||||
// const result = await prisma.someTable.deleteMany({
|
||||
// where: {
|
||||
// createdAt: {
|
||||
// lt: thirtyDaysAgo
|
||||
// }
|
||||
// }
|
||||
// });
|
||||
|
||||
console.log(`Hourly task completed at ${new Date().toISOString()}`);
|
||||
}
|
||||
|
||||
// 每日清理任务的具体逻辑
|
||||
private async dailyCleanupHandler() {
|
||||
// 在这里添加每日清理逻辑
|
||||
// 例如：清理临时文件、压缩日志、备份数据等
|
||||
|
||||
console.log(`Daily cleanup completed at ${new Date().toISOString()}`);
|
||||
}
|
||||
|
||||
// 停止指定任务
|
||||
stopTask(taskName: string) {
|
||||
const task = this.tasks.get(taskName);
|
||||
if (task) {
|
||||
task.stop();
|
||||
console.log(`Task ${taskName} stopped`);
|
||||
}
|
||||
}
|
||||
|
||||
// 停止所有任务
|
||||
stopAll() {
|
||||
this.tasks.forEach((task, name) => {
|
||||
task.stop();
|
||||
console.log(`Task ${name} stopped`);
|
||||
});
|
||||
this.tasks.clear();
|
||||
}
|
||||
|
||||
// 获取任务状态
|
||||
getTaskStatus(taskName: string): boolean {
|
||||
const task = this.tasks.get(taskName);
|
||||
return task ? true : false; // 如果任务存在则认为正在运行
|
||||
}
|
||||
|
||||
// 列出所有任务
|
||||
listTasks(): string[] {
|
||||
return Array.from(this.tasks.keys());
|
||||
}
|
||||
}
|
||||
|
||||
// 导出单例实例
|
||||
export const taskScheduler = new TaskScheduler();
|
||||
|
||||
// Cron 表达式说明：
|
||||
// ┌───────────── 分钟 (0 - 59)
|
||||
// │ ┌─────────── 小时 (0 - 23)
|
||||
// │ │ ┌───────── 日期 (1 - 31)
|
||||
// │ │ │ ┌─────── 月份 (1 - 12)
|
||||
// │ │ │ │ ┌───── 星期 (0 - 7, 0和7都是周日)
|
||||
// │ │ │ │ │
|
||||
// * * * * *
|
||||
|
||||
// 常用示例：
|
||||
// '30 * * * *' - 每小时的第30分钟
|
||||
// '0 */2 * * *' - 每2小时执行一次
|
||||
// '0 9 * * 1-5' - 周一到周五上午9点
|
||||
// '0 0 1 * *' - 每月1号凌晨执行
|
||||
// '*/15 * * * *' - 每15分钟执行一次
|
||||
10
package.json
10
package.json
@ -16,26 +16,30 @@
|
||||
"@prisma/client": "^6.10.1",
|
||||
"@types/bcryptjs": "^3.0.0",
|
||||
"@types/cors": "^2.8.19",
|
||||
"@types/fluent-ffmpeg": "^2.1.27",
|
||||
"@types/minio": "^7.1.1",
|
||||
"@types/multer": "^1.4.13",
|
||||
"bcryptjs": "^3.0.2",
|
||||
"cors": "^2.8.5",
|
||||
"date-fns": "^4.1.0",
|
||||
"dotenv-cli": "^8.0.0",
|
||||
"fluent-ffmpeg": "^2.1.3",
|
||||
"lucide-react": "^0.525.0",
|
||||
"minio": "^8.0.5",
|
||||
"multer": "^2.0.1",
|
||||
"next": "15.3.4",
|
||||
"node-cron": "^4.1.1",
|
||||
"prisma": "^6.10.1",
|
||||
"react": "^19.0.0",
|
||||
"react-dom": "^19.0.0"
|
||||
},
|
||||
"devDependencies": {
|
||||
"typescript": "^5",
|
||||
"@tailwindcss/postcss": "^4",
|
||||
"@types/node": "^20",
|
||||
"@types/node-cron": "^3.0.11",
|
||||
"@types/react": "^19",
|
||||
"@types/react-dom": "^19",
|
||||
"@tailwindcss/postcss": "^4",
|
||||
"tailwindcss": "^4"
|
||||
"tailwindcss": "^4",
|
||||
"typescript": "^5"
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user