UNPKG

@cloudbase/node-sdk

Version:

tencent cloud base server sdk for node.js

97 lines (96 loc) 3.8 kB
"use strict"; var __importDefault = (this && this.__importDefault) || function (mod) { return (mod && mod.__esModule) ? mod : { "default": mod }; }; Object.defineProperty(exports, "__esModule", { value: true }); exports.extraRequest = exports.requestWithTimingsMeasure = void 0; const http_1 = __importDefault(require("http")); const retry_1 = require("./retry"); const request_timings_measurer_1 = require("./request-timings-measurer"); const request_core_1 = require("./request-core"); const SAFE_RETRY_CODE_SET = new Set([ 'ENOTFOUND', 'ENETDOWN', 'EHOSTDOWN', 'ENETUNREACH', 'EHOSTUNREACH', 'ECONNREFUSED' ]); // const RETRY_CODE_SET = new Set(['ECONNRESET', 'ESOCKETTIMEDOUT']) const RETRY_STATUS_CODE_SET = new Set([]); /* istanbul ignore next */ function shouldRetry(e, result, operation) { // 重试的错误码 if (e && SAFE_RETRY_CODE_SET.has(e.code)) { return { retryAble: true, message: e.message }; } // 连接超时 if (e && e.code === 'ETIMEDOUT' && e.connecting === true) { return { retryAble: true, message: e.message }; } // 重试的状态码 if (result && RETRY_STATUS_CODE_SET.has(result.statusCode)) { return { retryAble: true, message: `${result.request.method} ${result.request.href} ${result.statusCode} ${http_1.default.STATUS_CODES[result.statusCode]}` }; } return { retryAble: false, message: '' }; } /* istanbul ignore next */ async function requestWithTimingsMeasure(opts, extraOptions) { return await new Promise((resolve, reject) => { const timingsMeasurerOptions = extraOptions.timingsMeasurerOptions || {}; const { waitingTime = 1000, interval = 200, enable = !!extraOptions.debug } = timingsMeasurerOptions; const timingsMeasurer = request_timings_measurer_1.RequestTimgingsMeasurer.new({ waitingTime, interval, enable }); timingsMeasurer.on('progress', (timings, reason = '') => { const timingsLine = `s:${timings.socket || '-'}|l:${timings.lookup || '-'}|c:${timings.connect || '-'}|r:${timings.ready || '-'}|w:${timings.waiting || '-'}|d:${timings.download || '-'}|e:${timings.end || '-'}|E:${timings.error || '-'}`; console.warn(`[TCB][RequestTimgings][${extraOptions.op || ''}] spent ${Date.now() - timings.start}ms(${timingsLine}) [${extraOptions.seqId}][${extraOptions.attempts || 1}][${reason}]`); }); (function r() { const cRequest = (0, request_core_1.request)(opts, (err, res, body) => { if (err) { reject(err); } else { resolve({ statusCode: res.statusCode, headers: res.headers, body }); } }); if (cRequest instanceof http_1.default.ClientRequest) { timingsMeasurer.measure(cRequest); } }()); }); } exports.requestWithTimingsMeasure = requestWithTimingsMeasure; async function extraRequest(opts, extraOptions) { if (extraOptions && extraOptions.retryOptions) { return await (0, retry_1.withRetry)(async (attempts) => { return await requestWithTimingsMeasure(opts, Object.assign(Object.assign({}, extraOptions), { attempts })); }, Object.assign({ shouldRetry }, extraOptions.retryOptions)); } else { return await requestWithTimingsMeasure(opts, Object.assign(Object.assign({}, extraOptions), { attempts: 1 })); } } exports.extraRequest = extraRequest;