初始化环境文件
This commit is contained in:
60
node_modules/@smithy/util-stream/dist-es/checksum/ChecksumStream.js
generated
vendored
Normal file
60
node_modules/@smithy/util-stream/dist-es/checksum/ChecksumStream.js
generated
vendored
Normal file
@@ -0,0 +1,60 @@
|
||||
import { toBase64 } from "@smithy/util-base64";
|
||||
import { Duplex } from "stream";
|
||||
export class ChecksumStream extends Duplex {
|
||||
expectedChecksum;
|
||||
checksumSourceLocation;
|
||||
checksum;
|
||||
source;
|
||||
base64Encoder;
|
||||
pendingCallback = null;
|
||||
constructor({ expectedChecksum, checksum, source, checksumSourceLocation, base64Encoder, }) {
|
||||
super();
|
||||
if (typeof source.pipe === "function") {
|
||||
this.source = source;
|
||||
}
|
||||
else {
|
||||
throw new Error(`@smithy/util-stream: unsupported source type ${source?.constructor?.name ?? source} in ChecksumStream.`);
|
||||
}
|
||||
this.base64Encoder = base64Encoder ?? toBase64;
|
||||
this.expectedChecksum = expectedChecksum;
|
||||
this.checksum = checksum;
|
||||
this.checksumSourceLocation = checksumSourceLocation;
|
||||
this.source.pipe(this);
|
||||
}
|
||||
_read(size) {
|
||||
if (this.pendingCallback) {
|
||||
const callback = this.pendingCallback;
|
||||
this.pendingCallback = null;
|
||||
callback();
|
||||
}
|
||||
}
|
||||
_write(chunk, encoding, callback) {
|
||||
try {
|
||||
this.checksum.update(chunk);
|
||||
const canPushMore = this.push(chunk);
|
||||
if (!canPushMore) {
|
||||
this.pendingCallback = callback;
|
||||
return;
|
||||
}
|
||||
}
|
||||
catch (e) {
|
||||
return callback(e);
|
||||
}
|
||||
return callback();
|
||||
}
|
||||
async _final(callback) {
|
||||
try {
|
||||
const digest = await this.checksum.digest();
|
||||
const received = this.base64Encoder(digest);
|
||||
if (this.expectedChecksum !== received) {
|
||||
return callback(new Error(`Checksum mismatch: expected "${this.expectedChecksum}" but received "${received}"` +
|
||||
` in response header "${this.checksumSourceLocation}".`));
|
||||
}
|
||||
}
|
||||
catch (e) {
|
||||
return callback(e);
|
||||
}
|
||||
this.push(null);
|
||||
return callback();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user