import { MarqueeType, MarqueeModel } from "../db/Marquee"; import { MARQUEE_TYPE, STATUS } from "../consts"; import { scheduleJob, scheduledJobs, Job } from 'node-schedule'; import { createMarqueeMsg as sysCreateMarqueeMsg, pushMarqueeMsg as sysPushMarqueeMsg } from './sysChatService'; import { GroupMessageType } from "../db/GroupMessage"; import { resResult } from "../pubUtils/util"; export async function initMarquee() { const marquees = await MarqueeModel.findEffectiveMarque(); for(let marquee of marquees) { await generateMarqueeSchedule(marquee); } } export async function setMarquee(code: string) { let marquee = await MarqueeModel.findByCode(code); if(marquee.type == MARQUEE_TYPE.INSTANT) { let msgDatas = await createMarqueeMsg(marquee); await pushMarqueeMsg(msgDatas); return true; } else { return await generateMarqueeSchedule(marquee); } } async function generateMarqueeSchedule(marquee: MarqueeType) { if(marquee.type != MARQUEE_TYPE.SCHEDULE) return false; if(marquee.startTime.getTime() >= marquee.endTime.getTime()) return false; if(marquee.endTime.getTime() < Date.now()) return false; if(marquee.startTime.getTime() > Date.now()) { let startJob = scheduleJob(`start${marquee.code}`, marquee.startTime.getTime(), async () => { await startMarquee(marquee, startJob) }); } else { await startMarquee(marquee) } return true; } async function startMarquee(marquee: MarqueeType, startJob?: Job) { console.log('************', `跑马灯定时器 ${marquee.code}开始`, '************'); await MarqueeModel.updateData(marquee.code, { isRunning: true }); let msgDatas = await createMarqueeMsg(marquee); await pushMarqueeMsg(msgDatas); let secondsJob = scheduleJob(`setSeconds${marquee.code}`, `*/${marquee.interval} * * * * *`, async () => { console.log(`*****setSeconds${marquee.code}****`) await pushMarqueeMsg(msgDatas); }); let endJob = scheduleJob(`end${marquee.code}`, marquee.endTime.getTime(), async () => { if(startJob) { startJob.cancel(); startJob = undefined; } if(secondsJob) { secondsJob.cancel(); secondsJob = undefined; } if(endJob) { endJob.cancel(); endJob = undefined; } await MarqueeModel.updateData(marquee.code, { isRunning: false }); }); } async function createMarqueeMsg(marquee: MarqueeType) { let { serverIds, content } = marquee; let msgDatas: GroupMessageType[] = []; for(let serverId of serverIds) { let msgData = await sysCreateMarqueeMsg('', '系统', serverId, content); msgDatas.push(msgData); } return msgDatas; } async function pushMarqueeMsg(msgDatas: GroupMessageType[]) { for(let msgData of msgDatas) { await sysPushMarqueeMsg(msgData); } } export async function cancelMarquee(code: string) { let startMarquee = scheduledJobs[`start${code}`]; let setSecondsMarquee = scheduledJobs[`setSeconds${code}`]; let endMarquee = scheduledJobs[`end${code}`]; if(startMarquee) startMarquee.cancel(); if(setSecondsMarquee) setSecondsMarquee.cancel(); if(endMarquee) endMarquee.cancel(); await MarqueeModel.updateData(code, { isRunning: false }); return true }