91 lines
3.3 KiB
TypeScript
91 lines
3.3 KiB
TypeScript
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
|
|
} |