@gravity-ui/gateway 4.11.0 → 4.11.1

package/build/components/rest.js

@@ -11,6 +11,7 @@ const constants_1 = require("../constants");
 const axios_1 = require("../utils/axios");
 const common_1 = require("../utils/common");
 const parse_error_1 = require("../utils/parse-error");
+const pipe_through_byte_counter_1 = require("../utils/pipe-through-byte-counter");
 const redact_sensitive_headers_1 = require("../utils/redact-sensitive-headers");
 const validate_1 = require("../utils/validate");
 function getRestResponseSize(data, ctx, ErrorConstructor) {
@@ -306,7 +307,16 @@ function createRestAction(endpoints, config, serviceKey, actionName, options, Er
                 }
             }
             if (options === null || options === void 0 ? void 0 : options.sendStats) {
-                options.sendStats(Object.assign(Object.assign({}, requestData), { responseSize: getRestResponseSize(response === null || response === void 0 ? void 0 : response.data, ctx, ErrorConstructor), restStatus: 200 }), (0, redact_sensitive_headers_1.redactSensitiveHeaders)(parentCtx, headers), parentCtx, { debugHeaders: (0, common_1.sanitizeDebugHeaders)(debugHeaders) });
+                const emitStats = (responseSize, restStatus) => {
+                    var _a;
+                    return (_a = options.sendStats) === null || _a === void 0 ? void 0 : _a.call(options, Object.assign(Object.assign({}, requestData), { responseSize, restStatus }), (0, redact_sensitive_headers_1.redactSensitiveHeaders)(parentCtx, headers), parentCtx, { debugHeaders: (0, common_1.sanitizeDebugHeaders)(debugHeaders) });
+                };
+                if (response.data && typeof response.data.pipe === 'function') {
+                    response.data = (0, pipe_through_byte_counter_1.pipeThroughByteCounter)(response.data, (streamedBytes) => emitStats(streamedBytes, 200));
+                }
+                else {
+                    emitStats(getRestResponseSize(response === null || response === void 0 ? void 0 : response.data, ctx, ErrorConstructor), 200);
+                }
             }
             else {
                 ctx.stats(Object.assign(Object.assign({}, requestData), { responseStatus: 200 }));

package/build/utils/pipe-through-byte-counter.d.ts

@@ -0,0 +1,4 @@
+/// <reference types="node" />
+/// <reference types="node" />
+import type { Readable } from 'stream';
+export declare function pipeThroughByteCounter(source: NodeJS.ReadableStream, onFlush: (streamedBytes: number) => void): Readable;

package/build/utils/pipe-through-byte-counter.js

@@ -0,0 +1,24 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.pipeThroughByteCounter = void 0;
+const stream_1 = require("stream");
+function pipeThroughByteCounter(source, onFlush) {
+    let streamedBytes = 0;
+    const counter = new stream_1.Transform({
+        transform(chunk, encoding, cb) {
+            if (Buffer.isBuffer(chunk) || chunk instanceof Uint8Array) {
+                streamedBytes += chunk.length;
+            }
+            else if (typeof chunk === 'string') {
+                streamedBytes += Buffer.byteLength(chunk, typeof encoding === 'string' ? encoding : undefined);
+            }
+            cb(null, chunk);
+        },
+        flush(cb) {
+            onFlush(streamedBytes);
+            cb();
+        },
+    });
+    return source.pipe(counter);
+}
+exports.pipeThroughByteCounter = pipeThroughByteCounter;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gravity-ui/gateway",
-  "version": "4.11.0",
+  "version": "4.11.1",
   "description": "",
   "license": "MIT",
   "main": "build/index.js",