系统：停服维护逻辑

2021-07-15 14:40:26 +08:00
parent 352aecfb56
commit 0028c9b992
26 changed files with 348 additions and 50 deletions
@@ -0,0 +1,141 @@
+import { MarqueeType, MarqueeModel } from "../db/Marquee";
+import { MARQUEE_TYPE, STATUS, SERVER_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";
+import { MaintenanceModel, MaintenanceType } from '../db/Maintenance';
+import { pinus } from "pinus";
+import { getWorldChannelSid } from "./chatChannelService";
+import { ServerlistModel } from "../db/Serverlist";
+import { NoticeModel, NoticeType } from "../db/Notice";
+
+// —————————————— 跑马灯 —————————————— //
+// 初始
+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) {
+    let setSecondsMarquee = scheduledJobs[`setSeconds${marquee.code}`];
+    if(setSecondsMarquee) return false; // 已经在运行中了
+
+    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
+}
+
+// —————————————— 停服维护 —————————————— //
+// 维护信息
+export async function initMaintenance() {
+    const maintenance = await MaintenanceModel.findOpenMaintenance();
+    if(maintenance) {
+        let { startTime, marquee } = maintenance;
+        if(marquee) await generateMarqueeSchedule(<MarqueeType>marquee);
+        if(Date.now() > startTime.getTime()) {
+            await startMaintenance(maintenance);
+        } else {
+            let startJob = scheduleJob(`startMainten${maintenance.code}`, startTime.getTime(), async () => {
+                await startMaintenance(maintenance, startJob);
+            });
+        }
+    }
+}
+
+async function startMaintenance(maintenance: MaintenanceType, startJob?: Job) {
+    // 向全服发送
+    let { serverIds, notice } = maintenance;
+    for(let serverId of serverIds) {
+        let chatSid = await getWorldChannelSid(serverId);
+        await pinus.app.rpc.chat.chatRemote.sendServerMaintenance.toServer(chatSid, serverId);
+    }
+
+    // 更新serverlist上的status
+    await ServerlistModel.updateByServerIds(serverIds, { serverStatus: SERVER_STATUS.MAINTENANCE });
+    // 更新notice的isEnable
+    if(notice) {
+        await NoticeModel.updateNotice((<NoticeType>notice).id, { isEnable: true });
+    }
+    
+    // 更新connectorRemote里面的
+    let connectorServers = pinus.app.getServersByType('connector');
+    for(let { id } of connectorServers) {
+        pinus.app.rpc.connector.connectorRemote.setServerMainten.toServer(id, serverIds);
+    }
+    if(startJob) startJob.cancel();
+}