-
Notifications
You must be signed in to change notification settings - Fork 532
/
server.js
44 lines (38 loc) · 1.07 KB
/
server.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
'use strict'
const { unlinkSync } = require('node:fs')
const { createServer } = require('node:http')
const os = require('node:os')
const path = require('node:path')
const cluster = require('node:cluster')
const socketPath = path.join(os.tmpdir(), 'undici.sock')
const port = process.env.PORT || socketPath
const timeout = parseInt(process.env.TIMEOUT, 10) || 1
const workers = parseInt(process.env.WORKERS) || os.cpus().length
if (cluster.isPrimary) {
try {
unlinkSync(socketPath)
} catch (_) {
// Do nothing if the socket does not exist
}
for (let i = 0; i < workers; i++) {
cluster.fork()
}
} else {
const buf = Buffer.alloc(64 * 1024, '_')
const headers = {
'Content-Length': `${buf.byteLength}`,
'Content-Type': 'text/plain; charset=UTF-8'
}
let i = 0
const server = createServer((_req, res) => {
i++
setTimeout(() => {
res.writeHead(200, headers)
res.end(buf)
}, timeout)
}).listen(port)
server.keepAliveTimeout = 600e3
setInterval(() => {
console.log(`Worker ${process.pid} processed ${i} requests`)
}, 5000)
}