feat(http): support server listening on multi-hosts

config.example.yml

@@ -2,7 +2,7 @@ name: 'hitokoto' # 服务名称，例如：hitokoto
 url: 'https://v1.hitokoto.cn' # 服务地址，例如：https://v1.hitokoto.cn
 api_name: 'demo_prprpr' # 服务标识，取个好区分的标识吧，例如：cd-01-demo
 server: # 配置 HTTP 服务的信息
-  host: 127.0.0.1 # 监听的地址
+  hosts: 0.0.0.0 # 监听的地址
   port: 8000 # 监听的端口
   compress_body: true # 是否使用 GZIP 压缩
 redis: # 配置 Redis

package.json

@@ -1,14 +1,14 @@
 {
   "name": "hitokoto",
-  "version": "1.7.1",
+  "version": "1.7.2-dev",
   "description": "Fast & Powerful Hitokoto API Framework.",
   "main": "core.js",
   "engines": {
     "node": ">=14"
   },
   "scripts": {
     "start": "node core",
-    "dev": "yarn nodemon core.js -D",
+    "dev": "yarn nodemon -e js,mjs,cjs,json,yml,yaml core.js -- -D",
     "pnpify": "yarn dlx @yarnpkg/pnpify",
     "lint": "eslint ./src/**/*.js ./adapter/**/*.js ./test/**/*.js core.js",
     "watch": "yarn dlx supervisor --watch \".\" --extensions \"js,json\" --exec \"yarn\" -- \"start\"",

src/http/primary.js

@@ -11,7 +11,9 @@ module.exports = exports = {}
 
 const workersNumber = nconf.get('workers') || os.cpus().length
 const listeningPort = nconf.get('server:port') || 8000
-const workers = new Workers({ workersNumber, listeningPort })
+const listeningHosts =
+  nconf.get('server:host') || nconf.get('server:hosts') || '0.0.0.0'
+const workers = new Workers({ workersNumber, listeningPort, listeningHosts })
 const { requestsCounter } = require('../utils/master/requestsCounterMerge')
 
 exports.startWorkersPool = async () => {

src/http/worker.js

@@ -1,3 +1,4 @@
+const http = require('node:http')
 const Koa = require('koa')
 const chalk = require('chalk')
 const nconf = require('nconf')
@@ -22,7 +23,9 @@ async function registerRoutes(routes) {
     process.exit(1)
   }
 }
-async function StartWebServer(isDev, port) {
+async function StartWebServer(isDev, port, hosts) {
+  const { logger } = require('../logger')
+  logger.verbose(hosts)
   await require('../middleware').register(app, isDev) // Register middlewares
   const Routes = require('../route')
   await registerRoutes(new Routes().routes())
@@ -39,7 +42,17 @@ async function StartWebServer(isDev, port) {
       })
     })
   }
-  app.listen(port)
+  if (!hosts) {
+    http.createServer(app.callback()).listen(port)
+    return
+  }
+  hosts = typeof hosts === 'string' ? [hosts] : hosts
+  for (const host of hosts) {
+    logger.verbose(
+      `[http.Worker.StartWebServer] start http server on ${host}:${port}`,
+    )
+    http.createServer(app.callback()).listen(port, host)
+  }
 }
 
 function notifyMasterWorkerStarted() {
@@ -81,7 +94,7 @@ process.on('message', ({ key, data }) => {
       .then(async () => {
         const { recoverAB } = require('../utils')
         await recoverAB()
-        await StartWebServer(isDev, data.port)
+        await StartWebServer(isDev, data.port, data.hosts)
       })
       .finally(() => {
         const { logger } = require('../logger')

src/prestart.js

@@ -96,7 +96,7 @@ function loadConfig(configFile, next, isChild = false, isDev = false) {
       url: 'https://v1.hitokoto.cn',
       api_name: 'undefined',
       server: {
-        host: '127.0.0.1',
+        hosts: '0.0.0.0',
         port: 8000,
         compress_body: true,
       },

src/utils/master/workers.js

@@ -20,6 +20,7 @@ function setupCluster(params) {
 
 class Workers {
   workersNumbers = 0 // workers numbers
+  listeningHosts = '0.0.0.0'
   listeningPort = 8000 // workers listening port
   configFile = ''
   isDev = false // debug mode
@@ -36,8 +37,9 @@ class Workers {
    * @param {WorkersSettings} params
    * @returns {boolean|any}
    */
-  constructor({ workersNumber, listeningPort }) {
+  constructor({ workersNumber, listeningPort, listeningHosts }) {
     this.workersNumbers = workersNumber
+    this.listeningHosts = listeningHosts
     this.listeningPort = listeningPort
     this.configFile = nconf.get('config_file')
     this.isDev = !!nconf.get('dev')
@@ -77,10 +79,18 @@ class Workers {
       '[core.http.primary.Workers] %d workers will be spawned.',
       this.workersNumbers,
     )
-    logger.info(
-      '[core.http.primary.Workers] web server will listen on port: ' +
-        chalk.yellow(this.listeningPort),
-    )
+    const hosts =
+      typeof this.listeningHosts === 'string'
+        ? [this.listeningHosts]
+        : this.listeningHosts
+    for (const host of hosts) {
+      logger.info(
+        `[core.http.primary.Workers] web server will listen on ${chalk.yellow(
+          host + ':' + this.listeningPort,
+        )}`,
+      )
+    }
+
     for (let i = 0; i < this.workersNumbers; i++) {
       this.spawnWorker()
     }
@@ -116,6 +126,7 @@ class Workers {
       key: 'start_server',
       data: {
         port: this.listeningPort,
+        hosts: this.listeningHosts,
       },
     })
   }