detect-port

299 阅读2分钟

检查端口号是否被占用,被占用自动加1,最大加10

/**
 * 1. require('detect-port')({hostname: '0.0.0.0', port: 3000}).then(res => res)
 * 2. 判断传递进来的参数是否有回调。没有回调就使用promise,回调第一个参数是报错信息第二个是可以使用的端口,promise返回的直接就是一个可以使用的端口。
      把端口和ip赋值给变量。弄一个最大端口号变量maxPort = port+10,maxPort最大为65535,port - maxPort都被占用了就直接报错
 * 3. 调用tryListen方法,校验端口是否会报错,有ip就使用,没有就给null,走null没有报错就走主机的ip,在检查端口的时候报错就走handleError方法
 * 4. handleError方法用来端口号+1继续调用tryListen来校验
 */

/**
 * 1.  hostname: ip
 * 2. port: 端口
 * 3. net.Server(): 创建一个服务器
 * 4. server.on: 服务器状态监听
 * 5. server.listen: 开启服务器
 * 6. server.close(): 关闭服务器
 * 7. server.address(): 获取服务器地址
 */

// 严格模式
'use strict';

// 网络模块
const net = require('net');

// 用来获取ip,ipv6, mac
const address = require('address');

// 用来调试
const debug = require('debug')('detect-port');

module.exports = (port, callback) => {
  let hostname = '';

  // 根据类型赋值
  if (typeof port === 'object' && port) {
    hostname = port.hostname;
    callback = port.callback;
    port = port.port;
  } else {
    if (typeof port === 'function') {
      callback = port;
      port = null;
    }
  }

  port = parseInt(port) || 0;

  let maxPort = port + 10;
  if (maxPort > 65535) {
    maxPort = 65535;
  }

  // 建立一个debug实例,并且打印 检测之间的空闲端口 [3000, 3010)
  debug('检测之间的空闲端口 [%s, %s)', port, maxPort);

  // 如果是回调
  if (typeof callback === 'function') {
    return tryListen(port, maxPort, hostname, callback);
  }

  // promise
  return new Promise(resolve => {
    tryListen(port, maxPort, hostname, (_, realPort) => {
      resolve(realPort);
    });
  });
};

function tryListen(port, maxPort, hostname, callback) {
  // 处理报错,端口号添加一位
  function handleError() {
    port++;
    if (port >= maxPort) {
      debug('port: %s >= maxPort: %s, 放弃使用随机端口', port, maxPort);
      port = 0;
      maxPort = 0;
    }

    // 再走一遍检查,找到合适的port
    tryListen(port, maxPort, hostname, callback);
  }

  // 有ip
  if (hostname) {
    listen(port, hostname, (err, realPort) => {
      // 启动服务器报错
      if (err) {
        if (err.code === 'EADDRNOTAVAIL') {
          return callback(new Error('在机器上不是未知的IP'));
        }

        return handleError();
      }

      callback(null, realPort);
    });
  } else {
    // 1. 没有ip
    listen(port, null, (err, realPort) => {
      // 忽略随机听
      if (port === 0) {
        return callback(err, realPort);
      }

      if (err) {
        return handleError(err);
      }

      // 2. check 0.0.0.0
      listen(port, '0.0.0.0', err => {
        if (err) {
          return handleError(err);
        }

        // 3. 检查主机
        listen(port, 'localhost', err => {
          //  如果localhost在机器上引用的ip不是未知的,您将看到错误EADDRNOTAVAIL
          // https://stackoverflow.com/questions/10809740/listen-eaddrnotavail-error-in-node-js
          if (err && err.code !== 'EADDRNOTAVAIL') {
            return handleError(err);
          }

          // 4. 检查当前ip
          listen(port, address.ip(), (err, realPort) => {
            if (err) {
              return handleError(err);
            }

            callback(null, realPort);
          });
        });
      });
    });
  }
}

function listen(port, hostname, callback) {
  const server = new net.Server();

  // 监听server的报错事件
  server.on('error', err => {
    debug('监听 %s:%s 报错: %s', hostname, port, err);

    // 服务器停止接收新的连接
    server.close();

    if (err.code === 'ENOTFOUND') {
      debug('忽略dns ENOTFOUND错误, 免除 %s:%s', hostname, port);
      return callback(null, port);
    }
    return callback(err);
  });

  // 启动监听连接的服务器
  server.listen(port, hostname, () => {
    // 操作系统返回绑定的地址,协议族名和服务器端口。
    port = server.address().port;

    // 服务器停止接收新的连接
    server.close();

    debug('免除 %s:%s', hostname, port);

    return callback(null, port);
  });
}