|
1 | 1 | import cluster from 'node:cluster';
|
2 | 2 | import process from 'node:process';
|
3 | 3 | import os from 'node:os';
|
| 4 | +import fs from 'node:fs'; |
4 | 5 | import { createServer } from 'http';
|
5 | 6 | import { parse } from 'url';
|
6 |
| -import path from 'path'; |
7 |
| -import next from 'next'; |
8 | 7 |
|
9 |
| -//console.log("Running custom server.js"); |
10 |
| - |
11 |
| -const numCPUs = os.cpus().length; |
12 |
| -const port = parseInt(process.env.PORT || '3000', 10) |
13 | 8 | const dev = process.env.NODE_ENV !== 'production'
|
14 |
| -const app = next({ dev }) |
15 |
| -const handle = app.getRequestHandler() |
16 | 9 |
|
17 |
| -if (!dev && cluster.isPrimary) { |
18 |
| - console.log(`Primary ${process.pid} is running`); |
19 |
| - |
20 |
| - // Fork workers |
21 |
| - for (let i = 0; i < numCPUs; i++) { |
22 |
| - cluster.fork(); |
| 10 | +// Make sure commands gracefully respect termination signals (e.g. from Docker) |
| 11 | +// Allow the graceful termination to be manually configurable |
| 12 | +if (!process.env.NEXT_MANUAL_SIG_HANDLE) { |
| 13 | + process.on('SIGTERM', () => process.exit(0)) |
| 14 | + process.on('SIGINT', () => process.exit(0)) |
| 15 | +} |
| 16 | + |
| 17 | +if (!dev) { |
| 18 | + const servers = ['./.next/standalone/server.js', './server.js']; |
| 19 | + let server = ''; |
| 20 | + |
| 21 | + for (var i = 0; i < servers.length; i++) { |
| 22 | + if (fs.existsSync(servers[i])) { |
| 23 | + server = fs.realpathSync(servers[i]); |
| 24 | + break; |
| 25 | + } |
23 | 26 | }
|
24 | 27 |
|
25 |
| - cluster.on('exit', (worker, code, signal) => { |
26 |
| - console.log(`worker ${worker.process.pid} stopped`, { code, signal }); |
27 |
| - }); |
| 28 | + if (server === '') { |
| 29 | + console.error('Could not find server.js'); |
| 30 | + process.exit(-1); |
| 31 | + } |
| 32 | + |
| 33 | + //console.log(`server.js found at: ${server}`); |
| 34 | + |
| 35 | + if (cluster.isPrimary) { |
| 36 | + console.log(`Primary ${process.pid} is running`); |
| 37 | + |
| 38 | + const numCPUs = os.cpus().length; |
| 39 | + |
| 40 | + // Fork workers |
| 41 | + for (let i = 0; i < numCPUs; i++) { |
| 42 | + cluster.fork(); |
| 43 | + } |
| 44 | + |
| 45 | + cluster.on('exit', (worker, code, signal) => { |
| 46 | + console.log(`worker ${worker.process.pid} stopped`, { code, signal }); |
| 47 | + }); |
| 48 | + } else { |
| 49 | + require(server); |
| 50 | + } |
28 | 51 | } else {
|
| 52 | + console.log('Development mode'); |
| 53 | + |
| 54 | + const next = require('next'); |
| 55 | + const app = next({ dev }); |
| 56 | + const handle = app.getRequestHandler(); |
| 57 | + const port = parseInt(process.env.PORT || '3000', 10); |
| 58 | + |
29 | 59 | app.prepare().then(() => {
|
30 | 60 | createServer((req, res) => {
|
31 | 61 | const parsedUrl = parse(req.url!, true)
|
32 | 62 | handle(req, res, parsedUrl)
|
33 |
| - }).listen(port) |
| 63 | + }).listen(port); |
34 | 64 |
|
35 | 65 | console.log(
|
36 | 66 | `> Server listening at http://localhost:${port} as ${
|
37 | 67 | dev ? 'development' : process.env.NODE_ENV
|
38 | 68 | }`
|
39 |
| - ) |
40 |
| - }) |
| 69 | + ); |
| 70 | + }); |
41 | 71 | }
|
0 commit comments