Make finalization part of the stream
This commit is contained in:
parent
c93a211595
commit
86b4da1ec2
3 changed files with 122 additions and 55 deletions
22
ts/util/finalStream.ts
Normal file
22
ts/util/finalStream.ts
Normal file
|
@ -0,0 +1,22 @@
|
|||
// Copyright 2024 Signal Messenger, LLC
|
||||
// SPDX-License-Identifier: AGPL-3.0-only
|
||||
|
||||
import { Transform } from 'node:stream';
|
||||
|
||||
export function finalStream(finalizer: () => Promise<void> | void): Transform {
|
||||
return new Transform({
|
||||
transform(data, enc, callback) {
|
||||
this.push(Buffer.from(data, enc));
|
||||
callback();
|
||||
},
|
||||
async final(callback) {
|
||||
try {
|
||||
await finalizer();
|
||||
} catch (error) {
|
||||
callback(error);
|
||||
return;
|
||||
}
|
||||
callback(null);
|
||||
},
|
||||
});
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue