109 lines
3.6 KiB
JavaScript
109 lines
3.6 KiB
JavaScript
import { PrismaClient } from "@prisma/client";
|
||
import { PrismaMariaDb } from "@prisma/adapter-mariadb";
|
||
import "dotenv/config";
|
||
|
||
const base = process.env.DATABASE_URL || "mysql://localhost:3306/zhuiguang_ai?charset=utf8mb4";
|
||
const sep = base.includes("?") ? "&" : "?";
|
||
const connectionString = `${base}${sep}connection_limit=20&pool_timeout=30`;
|
||
const adapter = new PrismaMariaDb(connectionString);
|
||
const prisma = new PrismaClient({ adapter });
|
||
|
||
const TASK_DEFS = [
|
||
{
|
||
taskKey: "task1-discover-tools",
|
||
taskName: "发现AI工具",
|
||
enabled: true,
|
||
cronExpr: "0 6 * * *",
|
||
description: "每天早上6点从36kr/IT之家/开源中国RSS抓取新闻，由DeepSeek提取AI工具信息并入库待审核",
|
||
config: { sources: ["36kr", "IT之家", "开源中国"], model: "deepseek-v4-pro" },
|
||
},
|
||
{
|
||
taskKey: "task3-check-tools",
|
||
taskName: "工具巡检",
|
||
enabled: true,
|
||
cronExpr: "0 3 * * *",
|
||
description: "每天凌晨3点检查所有已发布工具的网站可访问性和Logo有效性，自动发布通过检查的待审核工具",
|
||
config: { checkWebsite: true, checkLogo: true, autoPublish: true },
|
||
},
|
||
{
|
||
taskKey: "task4-discover-skills",
|
||
taskName: "发现AI技能",
|
||
enabled: true,
|
||
cronExpr: "0 5 * * *",
|
||
description: "每天凌晨5点从GitHub搜索高星AI项目(stars≥1000)，DeepSeek预评测后入库PendingSkill待审核",
|
||
config: { minStars: 1000, dailyLimit: 30, model: "deepseek-v4-pro" },
|
||
},
|
||
{
|
||
taskKey: "task5-update-stars",
|
||
taskName: "更新技能星值",
|
||
enabled: true,
|
||
cronExpr: "0 1 * * *",
|
||
description: "每天凌晨1点更新所有GitHub技能的最新Stars数，记录日/周/月变化趋势",
|
||
config: { updateAll: true, historyDays: 90 },
|
||
},
|
||
{
|
||
taskKey: "task6-review-hot",
|
||
taskName: "评测热门技能",
|
||
enabled: true,
|
||
cronExpr: "0 4 * * *",
|
||
description: "每天凌晨4点对2个最热门且未评测的GitHub技能进行五维深度评测，自动生成SkillReview",
|
||
config: { dailyLimit: 2, model: "deepseek-v4-pro" },
|
||
},
|
||
{
|
||
taskKey: "daily-news",
|
||
taskName: "AI日报生成",
|
||
enabled: true,
|
||
cronExpr: "30 7 * * *",
|
||
description: "每天早上7:30从多源RSS搜索真实AI新闻，由DeepSeek整理格式生成当日日报",
|
||
config: { sources: ["36kr", "IT之家", "开源中国", "GoogleNews", "HackerNews"], model: "deepseek-v4-pro" },
|
||
},
|
||
];
|
||
|
||
async function main() {
|
||
console.log("🌱 初始化任务配置种子数据...\n");
|
||
|
||
let created = 0;
|
||
let updated = 0;
|
||
|
||
for (const def of TASK_DEFS) {
|
||
const existing = await prisma.taskConfig.findUnique({
|
||
where: { taskKey: def.taskKey },
|
||
});
|
||
|
||
if (existing) {
|
||
await prisma.taskConfig.update({
|
||
where: { taskKey: def.taskKey },
|
||
data: {
|
||
taskName: def.taskName,
|
||
cronExpr: def.cronExpr,
|
||
description: def.description,
|
||
config: def.config,
|
||
},
|
||
});
|
||
updated++;
|
||
console.log(` 🔄 更新: ${def.taskName} (${def.taskKey})`);
|
||
} else {
|
||
await prisma.taskConfig.create({
|
||
data: {
|
||
taskKey: def.taskKey,
|
||
taskName: def.taskName,
|
||
enabled: def.enabled,
|
||
cronExpr: def.cronExpr,
|
||
description: def.description,
|
||
config: def.config,
|
||
},
|
||
});
|
||
created++;
|
||
console.log(` ✅ 创建: ${def.taskName} (${def.taskKey})`);
|
||
}
|
||
}
|
||
|
||
console.log(`\n🏁 完成: 创建${created}个, 更新${updated}个`);
|
||
await prisma.$disconnect();
|
||
}
|
||
|
||
main().catch(async (e) => {
|
||
console.error("❌ 种子数据初始化失败:", e.message);
|
||
if (prisma) await prisma.$disconnect();
|
||
process.exit(1);
|
||
}); |