@atomiqlabs/sdk-lib
Version:
Basic SDK functionality library for atomiq
52 lines (51 loc) • 1.74 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.ResponseParamDecoder = void 0;
const ParamDecoder_1 = require("../ParamDecoder");
const buffer_1 = require("buffer");
class ResponseParamDecoder extends ParamDecoder_1.ParamDecoder {
constructor(resp, abortSignal) {
super();
this.abortSignal = abortSignal;
try {
//Read from stream
this.reader = resp.body.getReader();
this.readResponse();
}
catch (e) {
//Read in one piece
resp.arrayBuffer().then(respBuffer => {
super.onData(buffer_1.Buffer.from(respBuffer));
super.onEnd();
}).catch(e => {
super.onError(e);
});
}
if (abortSignal != null)
abortSignal.addEventListener("abort", () => {
super.onError(abortSignal.reason);
if (!this.reader.closed)
this.reader.cancel(abortSignal.reason);
});
}
/**
* Keeps reading the response until the reader closes
* @private
*/
async readResponse() {
while (true) {
const readResp = await this.reader.read().catch(e => {
console.error(e);
return null;
});
if (this.abortSignal != null && this.abortSignal.aborted)
return;
if (readResp == null || readResp.done) {
super.onEnd();
return;
}
super.onData(buffer_1.Buffer.from(readResp.value));
}
}
}
exports.ResponseParamDecoder = ResponseParamDecoder;