@apollo/server/dist/esm/plugin/drainHttpServer/stoppable.js

Core engine for Apollo GraphQL server

import https from 'https';
export class Stopper {
    server;
    requestCountPerSocket = new Map();
    stopped = false;
    constructor(server) {
        this.server = server;
        server.on(server instanceof https.Server ? 'secureConnection' : 'connection', (socket) => {
            this.requestCountPerSocket.set(socket, 0);
            socket.once('close', () => this.requestCountPerSocket.delete(socket));
        });
        server.on('request', (req, res) => {
            this.requestCountPerSocket.set(req.socket, (this.requestCountPerSocket.get(req.socket) ?? 0) + 1);
            res.once('finish', () => {
                const pending = (this.requestCountPerSocket.get(req.socket) ?? 0) - 1;
                this.requestCountPerSocket.set(req.socket, pending);
                if (this.stopped && pending === 0) {
                    req.socket.end();
                }
            });
        });
    }
    async stop(hardDestroyAbortSignal) {
        let gracefully = true;
        await new Promise((resolve) => setImmediate(resolve));
        this.stopped = true;
        const onAbort = () => {
            gracefully = false;
            this.requestCountPerSocket.forEach((_, socket) => socket.end());
            setImmediate(() => {
                this.requestCountPerSocket.forEach((_, socket) => socket.destroy());
            });
        };
        hardDestroyAbortSignal?.addEventListener('abort', onAbort);
        const closePromise = new Promise((resolve) => this.server.close(() => {
            hardDestroyAbortSignal?.removeEventListener('abort', onAbort);
            resolve();
        }));
        this.requestCountPerSocket.forEach((requests, socket) => {
            if (requests === 0)
                socket.end();
        });
        await closePromise;
        return gracefully;
    }
}
//# sourceMappingURL=stoppable.js.map