feat: implement enableCluster

Author: DIYgod

lib/config.ts

@@ -456,7 +456,7 @@ const calculateValue = () => {
     const _value = {
         // app config
         disallowRobot: toBoolean(envs.DISALLOW_ROBOT, false),
-        enableCluster: envs.ENABLE_CLUSTER,
+        enableCluster: toBoolean(envs.ENABLE_CLUSTER, false),
         isPackage: !!envs.IS_PACKAGE,
         nodeName: envs.NODE_NAME,
         puppeteerWSEndpoint: envs.PUPPETEER_WS_ENDPOINT,

lib/index.ts

@@ -3,25 +3,59 @@ import logger from '@/utils/logger';
 import { getLocalhostAddress } from '@/utils/common-utils';
 import { config } from '@/config';
 import app from '@/app';
+import os from 'node:os';
+import cluster from 'node:cluster';
+import process from 'node:process';
 
 const port = config.connect.port;
 const hostIPList = getLocalhostAddress();
 
-logger.info(`🎉 RSSHub is running on port ${port}! Cheers!`);
-logger.info(`🔗 Local: 👉 http://localhost:${port}`);
-if (config.listenInaddrAny) {
-    for (const ip of hostIPList) {
-        logger.info(`🔗 Network: 👉 http://${ip}:${port}`);
+let server;
+if (config.enableCluster) {
+    if (cluster.isPrimary) {
+        logger.info(`🎉 RSSHub is running on port ${port}! Cheers!`);
+        logger.info(`🔗 Local: 👉 http://localhost:${port}`);
+        if (config.listenInaddrAny) {
+            for (const ip of hostIPList) {
+                logger.info(`🔗 Network: 👉 http://${ip}:${port}`);
+            }
+        }
+
+        logger.info(`Primary ${process.pid} is running`);
+
+        const numCPUs = os.availableParallelism();
+
+        for (let i = 0; i < numCPUs; i++) {
+            cluster.fork();
+        }
+    } else {
+        logger.info(`Worker ${process.pid} is running`);
+        serve({
+            fetch: app.fetch,
+            hostname: config.listenInaddrAny ? '::' : '127.0.0.1',
+            port,
+            serverOptions: {
+                maxHeaderSize: 1024 * 32,
+            },
+        });
+    }
+} else {
+    logger.info(`🎉 RSSHub is running on port ${port}! Cheers!`);
+    logger.info(`🔗 Local: 👉 http://localhost:${port}`);
+    if (config.listenInaddrAny) {
+        for (const ip of hostIPList) {
+            logger.info(`🔗 Network: 👉 http://${ip}:${port}`);
+        }
     }
-}
 
-const server = serve({
-    fetch: app.fetch,
-    hostname: config.listenInaddrAny ? '::' : '127.0.0.1',
-    port,
-    serverOptions: {
-        maxHeaderSize: 1024 * 32,
-    },
-});
+    server = serve({
+        fetch: app.fetch,
+        hostname: config.listenInaddrAny ? '::' : '127.0.0.1',
+        port,
+        serverOptions: {
+            maxHeaderSize: 1024 * 32,
+        },
+    });
+}
 
 export default server;