@koeroesi86/node-worker-express 22.3.12-fd2e588 → 23.3.4361459321-main

package/utils/constructWsMessage.js

@@ -1,27 +0,0 @@
-/**
- * TODO: improve
- * @param {String} text
- * @returns {Buffer}
- */
-const constructWsMessage = text => {
-  const jsonByteLength = Buffer.byteLength(text);
-  // Note: we're not supporting > 65535 byte payloads at this stage
-  const lengthByteCount = jsonByteLength < 126 ? 0 : 2;
-  const payloadLength = lengthByteCount === 0 ? jsonByteLength : 126;
-  const buffer = Buffer.alloc(2 + lengthByteCount + jsonByteLength);
-  // Write out the first byte, using opcode `1` to indicate that the message
-  // payload contains text data
-  buffer.writeUInt8(0b10000001, 0);
-  buffer.writeUInt8(payloadLength, 1);
-  // Write the length of the JSON payload to the second byte
-  let payloadOffset = 2;
-  if (lengthByteCount > 0) {
-    buffer.writeUInt16BE(jsonByteLength, 2);
-    payloadOffset += lengthByteCount;
-  }
-  // Write the JSON data to the data buffer
-  buffer.write(text, payloadOffset);
-  return buffer;
-};
-
-module.exports = constructWsMessage;

package/utils/getClientIp.js

@@ -1,71 +0,0 @@
-/**
- * @param {module:http~IncomingMessage} request
- * @returns {string}
- */
-const getClientIp = request => {
-  if (request.headers['x-client-ip']) {
-    return request.headers['x-client-ip'];
-  }
-
-  if (request.headers['x-forwarded-for']) {
-    return request.headers['x-forwarded-for'];
-  }
-
-  if (request.headers['cf-connecting-ip']) {
-    return request.headers['cf-connecting-ip'];
-  }
-
-  if (request.headers['fastly-client-ip']) {
-    return request.headers['fastly-client-ip'];
-  }
-
-  if (request.headers['true-client-ip']) {
-    return request.headers['true-client-ip'];
-  }
-
-  if (request.headers['x-real-ip']) {
-    return request.headers['x-real-ip'];
-  }
-
-  if (request.headers['x-cluster-client-ip']) {
-    return request.headers['x-cluster-client-ip'];
-  }
-
-  if (request.headers['x-forwarded']) {
-    return request.headers['x-forwarded'];
-  }
-
-  if (request.headers['forwarded-for']) {
-    return request.headers['forwarded-for'];
-  }
-
-  if (request.headers.forwarded) {
-    return request.headers.forwarded;
-  }
-
-  if (request.connection) {
-    if (request.connection.remoteAddress) {
-      return request.connection.remoteAddress;
-    }
-
-    if (request.connection.socket && request.connection.socket.remoteAddress) {
-      return request.connection.socket.remoteAddress;
-    }
-  }
-
-  if (request.socket && request.socket.remoteAddress) {
-    return request.socket.remoteAddress;
-  }
-
-  if (request.info && request.info.remoteAddress) {
-    return request.info.remoteAddress;
-  }
-
-  if (request.requestContext && request.requestContext.identity && request.requestContext.identity.sourceIp) {
-    return request.requestContext.identity.sourceIp;
-  }
-
-  return '';
-};
-
-module.exports = getClientIp;

package/utils/isWebSocket.js

@@ -1,16 +0,0 @@
-/**
- * @param {module:http~IncomingMessage} request
- * @returns {boolean}
- */
-const isWebSocket = request => {
-  if (request.method !== 'GET') return false;
-
-  const connection = request.headers.connection || '';
-  const upgrade = request.headers.upgrade || '';
-
-  return request.method === 'GET' &&
-    connection.toLowerCase().split(/ *, */).indexOf('upgrade') >= 0 &&
-    upgrade.toLowerCase() === 'websocket';
-};
-
-module.exports = isWebSocket;

package/utils/parseWsMessage.js

@@ -1,25 +0,0 @@
-/**
- * TODO: improve
- * @param {Buffer} data
- * @returns {string}
- */
-const parseWsMessage = data => {
-  const dl = data[1] & 127;
-  let ifm = 2;
-  if (dl === 126) {
-    ifm = 4;
-  } else if (dl === 127) {
-    ifm = 10;
-  }
-  let i = ifm + 4;
-  const masks = data.slice(ifm, i);
-  let index = 0;
-  let output = "";
-  const l = data.length;
-  while (i < l) {
-    output += String.fromCharCode(data[i++] ^ masks[index++ % 4]);
-  }
-  return output;
-};
-
-module.exports = parseWsMessage;

package/utils/workerPool.js

@@ -1,101 +0,0 @@
-const { v4: uuid } = require('uuid');
-const path = require('path');
-const Worker = require('@koeroesi86/node-worker');
-
-/** @var {WorkerPool[]} */
-const pools = [];
-
-process.once('exit', () => {
-  pools.forEach(pool => pool.onClose());
-});
-
-const createWorkerCommand = workerPath => {
-  const ext = path.extname(workerPath);
-  switch (ext) {
-    case '.js':
-    default:
-      return `node --expose-gc ${path.resolve(__dirname, '../middleware/workerInvoke.js')} ${workerPath}`;
-  }
-};
-
-class WorkerPool {
-  constructor({
-                overallLimit = 0,
-                idleCheckTimeout = 5,
-                onExit = () => {
-                }
-              }) {
-    this.overallLimit = overallLimit;
-    this.idleCheckTimeout = idleCheckTimeout;
-    this.onExit = onExit;
-    this.workers = {};
-    pools.push(this);
-
-    this._creating = false;
-
-    this.getWorker = this.getWorker.bind(this);
-    this.onClose = this.onClose.bind(this);
-    this.getNonBusyId = this.getNonBusyId.bind(this);
-    this.getWorkerCount = this.getWorkerCount.bind(this);
-  }
-
-  onClose() {
-    Object.keys(this.workers).forEach(workerPath => {
-      const current = this.workers[workerPath];
-      Object.keys(current).forEach(id => {
-        current[id].terminate();
-      });
-    });
-  }
-
-  getNonBusyId(workerPath) {
-    return this.workers[workerPath] ? Object.keys(this.workers[workerPath] || {}).find(id => {
-      return !this.workers[workerPath][id].busy;
-    }) : undefined;
-  }
-
-  getWorkerCountForPath(p) {
-    return this.workers[p] ? Object.keys(this.workers[p]).length : 0;
-  }
-
-  getWorkerCount() {
-    return Object.keys(this.workers).reduce((result, current) => this.getWorkerCountForPath(current) + result, 0);
-  }
-
-  isBeyondLimit(workerPath, limit) {
-    return (this.workers[workerPath] && limit > 0 && this.getWorkerCountForPath(workerPath) >= limit)
-      || (this.overallLimit > 0 && this.getWorkerCount() >= this.overallLimit);
-  }
-
-  async getWorker(workerPath, options = {}, limit = 0) {
-    const nonBusyId = this.getNonBusyId(workerPath);
-    // TODO: tidy up
-    if (nonBusyId !== undefined) {
-      return this.workers[workerPath][nonBusyId];
-    } else if (this.isBeyondLimit(workerPath, limit) || this._creating) {
-      await new Promise(r => setTimeout(r, this.idleCheckTimeout));
-      return this.getWorker(workerPath, options, limit);
-    }
-
-    this._creating = true;
-    const id = uuid();
-    const instance = new Worker(createWorkerCommand(workerPath), options);
-
-    if (!this.workers[workerPath]) {
-      this.workers[workerPath] = {};
-    }
-
-    instance.addEventListenerOnce('close', code => {
-      this.workers[workerPath][id] = null;
-      delete this.workers[workerPath][id];
-      this.onExit(code, workerPath, id);
-    });
-
-    this.workers[workerPath][id] = instance;
-
-    this._creating = false;
-    return instance;
-  }
-}
-
-module.exports = WorkerPool;