116 lines
5.1 KiB
JavaScript
116 lines
5.1 KiB
JavaScript
import cron from 'node-cron'
|
|
import { prisma } from './db.js'
|
|
import mysql from 'mysql2/promise'
|
|
import { fetchNewsList, persistNews } from '../scripts/fetchers/news.fetcher.js'
|
|
import { checkAfterRefresh } from './alert-engine.js'
|
|
import { insertCronAlert } from '../scripts/lib/cron-alert.js'
|
|
import { ApizeroError, oil, history } from './apizero/index.js'
|
|
import * as mihome from './mihome/index.js'
|
|
|
|
function parseUrl(url) {
|
|
const u = new URL(url)
|
|
return { host: u.hostname, port: Number(u.port) || 3306, user: decodeURIComponent(u.username), password: decodeURIComponent(u.password), database: u.pathname.replace(/^\//, '') }
|
|
}
|
|
|
|
async function openConn() {
|
|
return mysql.createConnection(parseUrl(process.env.DATABASE_URL))
|
|
}
|
|
|
|
// 每天 6:00 (Asia/Shanghai) 拉取新闻 + 跑跨源对比, 出错生成报警
|
|
function scheduleNews() {
|
|
cron.schedule('0 6 * * *', async () => {
|
|
console.log('[cron] 6:00 触发')
|
|
try {
|
|
const list = await fetchNewsList()
|
|
const r = await persistNews(list)
|
|
console.log('[cron] 新闻入库: 新增 ' + r.inserted + ' / 跳过 ' + r.skipped)
|
|
const conn = await openConn()
|
|
try {
|
|
const a = await checkAfterRefresh(conn)
|
|
console.log('[cron] 报警检测: 新增 ' + a.inserted + ' 条 (扫描 ' + a.scanned + ')')
|
|
} finally { await conn.end() }
|
|
} catch (e) {
|
|
console.error('[cron] 任务失败:', e.message)
|
|
await insertCronAlert('新闻拉取+报警检测', e.message)
|
|
}
|
|
}, { timezone: 'Asia/Shanghai' })
|
|
console.log('[cron] 已注册: 每天 06:00 拉取新闻 + 跨源报警检测')
|
|
}
|
|
|
|
// 每天 6:10 (Asia/Shanghai) 抓浙江油价历史, 出错生成报警
|
|
// 逻辑: 拉今年 schedule, 检查待定且 <= 昨天的调价日, 命中就抓 price?province=浙江 存库
|
|
// - 0 命中: 不调 apizero (省配额)
|
|
// - >0 命中: 调一次, 4 个油品批量入库 (unique 兜底去重)
|
|
function scheduleOilHistory() {
|
|
cron.schedule('10 6 * * *', async () => {
|
|
console.log('[cron] 6:10 油价历史触发')
|
|
try {
|
|
const year = new Date().getFullYear()
|
|
const sched = await oil.getSchedule(year)
|
|
const list = sched.schedule || []
|
|
// 待定 且 日期 <= 昨天 (用 Asia/Shanghai 时区推导, 与 cron 触发时区对齐)
|
|
const todayStr = new Date().toLocaleDateString('en-CA', { timeZone: 'Asia/Shanghai' })
|
|
const today = new Date(todayStr + 'T00:00:00+08:00')
|
|
const yesterdayMs = today.getTime() - 86400000
|
|
const pendingPast = list.filter((it) => {
|
|
if (it.status !== '待定') return false
|
|
const t = new Date(it.date + 'T00:00:00').getTime()
|
|
return Number.isFinite(t) && t <= yesterdayMs
|
|
})
|
|
console.log('[cron] 待定且已过的调价日: ' + pendingPast.length + ' 个')
|
|
if (pendingPast.length === 0) {
|
|
console.log('[cron] 油价历史: 0 命中, 跳过 apizero 调用')
|
|
return
|
|
}
|
|
const r = await history.fetchAndStore()
|
|
console.log('[cron] 油价历史入库: 新增 ' + r.inserted + ' / 跳过 ' + r.skipped + ' (date=' + r.date + ')')
|
|
} catch (e) {
|
|
if (e instanceof ApizeroError) {
|
|
console.error('[cron] 油价历史 apizero 错误: code=' + e.code + ' msg=' + e.message)
|
|
} else {
|
|
console.error('[cron] 油价历史任务失败:', e.message)
|
|
}
|
|
await insertCronAlert('浙江油价历史抓取', e.message)
|
|
}
|
|
}, { timezone: 'Asia/Shanghai' })
|
|
console.log('[cron] 已注册: 每天 06:10 浙江油价历史抓取')
|
|
}
|
|
|
|
scheduleNews()
|
|
scheduleOilHistory()
|
|
|
|
// 环境监控: 每小时 5 分小时采样 + 每日 03:05 聚合/电量/vendor 历史补齐 (Asia/Shanghai)
|
|
// 整点 +5min 缓冲, 避开米家高峰
|
|
function scheduleMihome() {
|
|
cron.schedule('5 * * * *', async () => {
|
|
console.log('[cron] 米家小时采样触发 @ 上海 xx:05')
|
|
try {
|
|
const r = await mihome.syncNow()
|
|
console.log('[cron] 米家采样完成: 家庭=' + r.homes + ' 设备=' + r.devices + ' 采样=' + r.sampled + ' 聚合=' + r.aggregated + ' (' + r.ms + 'ms)')
|
|
} catch (e) {
|
|
console.error('[cron] 米家小时采样失败:', e.message)
|
|
}
|
|
}, { timezone: 'Asia/Shanghai' })
|
|
console.log('[cron] 已注册: 每小时 xx:05 米家温湿度采样')
|
|
|
|
cron.schedule('5 0,3 * * *', async () => {
|
|
console.log('[cron] 米家每日维护触发 (00:05 / 03:05)')
|
|
try {
|
|
const r = await mihome.dailyMaintenance()
|
|
console.log('[cron] 米家聚合完成: 采样=' + r.sampled + ' 聚合=' + r.aggregated
|
|
+ ' vendor小时=' + r.vendorHours + ' vendor天=' + r.vendorDays + ' 缺口聚合=' + r.gapAggregated)
|
|
} catch (e) {
|
|
console.error('[cron] 米家每日维护失败:', e.message)
|
|
}
|
|
}, { timezone: 'Asia/Shanghai' })
|
|
console.log('[cron] 已注册: 每天 00:05/03:05 米家日聚合 + vendor 补齐')
|
|
|
|
// 启动追赶: 服务重启/停机恢复后立即补一次 (不打断 5 分钟内的整点采样)
|
|
setTimeout(() => {
|
|
mihome.dailyMaintenance().catch((e) => console.error('[cron] 米家启动追赶失败:', e.message))
|
|
}, 8000)
|
|
console.log('[cron] 已注册: 启动后 8s 执行米家追赶同步')
|
|
}
|
|
|
|
scheduleMihome()
|