feat: 基础版不再限制流水线数量

2026-05-17 13:57:31 +08:00 · 2024-12-23 23:33:13 +08:00
parent bb4910f4e5
commit cb27d4b490
16 changed files with 158 additions and 62 deletions
@@ -0,0 +1,85 @@
+import { Autoload, Config, Init, Inject, Scope, ScopeEnum } from '@midwayjs/core';
+import { PipelineService } from '../pipeline/service/pipeline-service.js';
+import { logger } from '@certd/basic';
+import { SysSettingsService } from '@certd/lib-server';
+import { SiteInfoService } from '../monitor/index.js';
+import { Cron } from '../cron/cron.js';
+
+@Autoload()
+@Scope(ScopeEnum.Request, { allowDowngrade: true })
+export class AutoCRegisterCron {
+  @Inject()
+  pipelineService: PipelineService;
+
+  @Config('cron.onlyAdminUser')
+  private onlyAdminUser: boolean;
+
+  // @Inject()
+  // echoPlugin: EchoPlugin;
+  @Config('cron.immediateTriggerOnce')
+  private immediateTriggerOnce = false;
+
+  @Config('cron.immediateTriggerSiteMonitor')
+  private immediateTriggerSiteMonitor = false;
+
+  @Inject()
+  sysSettingsService: SysSettingsService;
+
+  @Inject()
+  siteInfoService: SiteInfoService;
+
+  @Inject()
+  cron: Cron;
+
+  @Init()
+  async init() {
+    logger.info('加载定时trigger开始');
+    await this.pipelineService.onStartup(this.immediateTriggerOnce, this.onlyAdminUser);
+    logger.info('加载定时trigger完成');
+    //
+    // const meta = getClassMetadata(CLASS_KEY, this.echoPlugin);
+    // console.log('meta', meta);
+    // const metas = listPropertyDataFromClass(CLASS_KEY, this.echoPlugin);
+    // console.log('metas', metas);
+    this.registerSiteMonitorCron();
+  }
+
+  registerSiteMonitorCron() {
+    const job = async () => {
+      logger.info('站点证书检查开始执行');
+
+      let offset = 0;
+      const limit = 50;
+      while (true) {
+        const res = await this.siteInfoService.page({
+          query: { disabled: false },
+          page: { offset, limit },
+        });
+        const { records } = res;
+
+        if (records.length === 0) {
+          break;
+        }
+        offset += records.length;
+        for (const record of records) {
+          try {
+            await this.siteInfoService.doCheck(record, true);
+          } catch (e) {
+            logger.error(`站点${record.name}检查出错：`, e);
+          }
+        }
+      }
+
+      logger.info('站点证书检查完成');
+    };
+
+    this.cron.register({
+      name: 'siteMonitor',
+      cron: '0 0 0 * * *',
+      job,
+    });
+    if (this.immediateTriggerSiteMonitor) {
+      job();
+    }
+  }
+}