From 8f0eca6179c75629f167e8f60cbebf3255b792d8 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Wed, 14 May 2025 20:49:07 +0800
Subject: [PATCH] core: add disableWorker option

---
 packages/hydrooj/src/service/worker.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/packages/hydrooj/src/service/worker.ts b/packages/hydrooj/src/service/worker.ts
index 8144bac68..202d1d4e8 100644
--- a/packages/hydrooj/src/service/worker.ts
+++ b/packages/hydrooj/src/service/worker.ts
@@ -1,3 +1,4 @@
+import cac from 'cac';
 import moment from 'moment-timezone';
 import { sleep } from '@hydrooj/utils';
 import { Context, Service } from '../context';
@@ -9,6 +10,8 @@ declare module '../context' {
     }
 }
 
+const argv = cac().parse();
+
 export default class WorkerService extends Service {
     private handlers: Record<string, Function> = {};
     consuming = true;
@@ -20,7 +23,8 @@ export default class WorkerService extends Service {
         super(ctx, 'worker');
     }
 
-    async *[Context.init]() {
+    *[Service.init]() {
+        if (argv.options.disableWorker) return;
         this.consume();
         yield () => {
             this.consuming = false;