在 node.js 中通过节点获取重用 TCP 连接

Reuse TCP connection with node-fetch in node.js

我正在使用这个函数调用外部 API

const fetch = require('node-fetch');

fetchdata= async function (result = {}) {
  var start_time = new Date().getTime();

    let response = await fetch('{API endpoint}', {
      method: 'post',
      body: JSON.stringify(result),
      headers: { 'Content-Type': 'application/json' },
      keepalive: true

    });

  console.log(response) 
  var time = { 'Respone time': + (new Date().getTime() - start_time) + 'ms' };
  console.log(time)
  return [response.json(), time];
  
}

问题是我不确定每次使用此功能时 node.js 是否重新使用到 API 的 TCP 连接,即使我定义了保活 属性。

重用 TCP 连接可以显着缩短响应时间
欢迎任何建议。

https://github.com/node-fetch/node-fetch#custom-agent

中所述
const fetch = require('node-fetch');

const http = require('http');
const https = require('https');

const httpAgent = new http.Agent({ keepAlive: true });
const httpsAgent = new https.Agent({ keepAlive: true });
const agent = (_parsedURL) => _parsedURL.protocol == 'http:' ? httpAgent : httpsAgent;

const fetchdata = async function (result = {}) {
    var start_time = new Date().getTime();

    let response = await fetch('{API endpoint}', {
        method: 'post',
        body: JSON.stringify(result),
        headers: { 'Content-Type': 'application/json' },
        agent
    });

    console.log(response)
    var time = { 'Respone time': + (new Date().getTime() - start_time) + 'ms' };
    console.log(time)
    return [response.json(), time];

}

Keep-alive 不是 enabled for the default used agent and is not currently implemented into node-fetch directly, but you can easily specify a custom-agent 您启用 keep-alive 选项的地方:

const keepAliveAgent = new http.Agent({
    keepAlive: true
});

fetch('{API endpoint}', {
     ...
     agent: keepAliveAgent         
});

这里是 node-fetch 的包装器,用于添加 keepAlive 选项,基于 Ilan Frumer 的回答

// fetch: add option keepAlive with default true
const fetch = (function getFetchWithKeepAlive() {
  const node_fetch = require('node-fetch');
  const http = require('http');
  const https = require('https');
  const httpAgent = new http.Agent({ keepAlive: true });
  const httpsAgent = new https.Agent({ keepAlive: true });
  return async function (url, userOptions) {
    const options = { keepAlive: true };
    Object.assign(options, userOptions);
    if (options.keepAlive == true)
      options.agent = (parsedUrl => parsedUrl.protocol == 'http:' ? httpAgent : httpsAgent);
    delete options.keepAlive;
    return await node_fetch(url, options);
  }
})();

const response = await fetch('https://github.com/');
const response = await fetch('https://github.com/', { keepAlive: false });

这是一个基于他们的文档的 node-fetch 包装器:

import nodeFetch, { RequestInfo, RequestInit } from "node-fetch";
import http from "http";
import https from "https";

const httpAgent = new http.Agent({
  keepAlive: true
});

const httpsAgent = new https.Agent({
  keepAlive: true
});

export const fetch = (url: RequestInfo, options: RequestInit = {}) => {
  return nodeFetch(url, {
    agent: (parsedURL) => {
      if (parsedURL.protocol === "http:") {
        return httpAgent;
      } else {
        return httpsAgent;
      }
    },
    ...options
  });
};