yetangitu/ya-webadb
1
0
Fork 0
mirror of https://github.com/yume-chan/ya-webadb.git synced 2025-10-03 09:49:24 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix(adb): minimize AbortSignal "abort" event handlers

Browse source
This commit is contained in:
Simon Chan 2024-06-18 00:09:56 +08:00
parent 68ef8d62d0
commit 1c14eb69e7
No known key found for this signature in database
GPG key ID: A8B69F750B9BCEDD
1 changed files with 13 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

26
libraries/adb/src/daemon/socket.ts
View file

@ -1,7 +1,6 @@
import { PromiseResolver } from "@yume-chan/async";
import type { Disposable } from "@yume-chan/event";
import type {
AbortSignal,
PushReadableStreamController,
ReadableStream,
WritableStream,
@ -67,12 +66,12 @@ export class AdbDaemonSocketController
#availableWriteBytesChanged: PromiseResolver<void> | undefined;
/**
* When delayed ack is disabled, can be `Infinity` if the socket is ready to write.
* Exactly one packet can be written no matter how large it is. Or `-1` if the socket
* is waiting for ack.
* When delayed ack is disabled, returns `Infinity` if the socket is ready to write
* (exactly one packet can be written no matter how large it is), or `-1` if the socket
* is waiting for ack message.
*
* When delayed ack is enabled, a non-negative finite number indicates the number of
* bytes that can be written to the socket before receiving an ack.
* When delayed ack is enabled, returns a non-negative finite number indicates the number of
* bytes that can be written to the socket before waiting for ack message.
*/
#availableWriteBytes = 0;
@ -90,8 +89,13 @@ export class AdbDaemonSocketController
this.writable = new MaybeConsumable.WritableStream<Uint8Array>({
start: (controller) => {
this.#writableController = controller;
controller.signal.addEventListener("abort", () => {
this.#availableWriteBytesChanged?.reject(
controller.signal.reason,
);
});
},
write: async (data, controller) => {
write: async (data) => {
const size = data.length;
const chunkSize = this.#dispatcher.options.maxPayloadSize;
for (
@ -100,7 +104,7 @@ export class AdbDaemonSocketController
start = end, end += chunkSize
) {
const chunk = data.subarray(start, end);
await this.#writeChunk(chunk, controller.signal);
await this.#writeChunk(chunk);
}
},
});
@ -109,16 +113,12 @@ export class AdbDaemonSocketController
this.#availableWriteBytes = options.availableWriteBytes;
}
async #writeChunk(data: Uint8Array, signal: AbortSignal) {
async #writeChunk(data: Uint8Array) {
const length = data.length;
while (this.#availableWriteBytes < length) {
// Only one lock is required because Web Streams API guarantees
// that `write` is not reentrant.
const resolver = new PromiseResolver<void>();
signal.addEventListener("abort", () => {
resolver.reject(signal.reason);
});
this.#availableWriteBytesChanged = resolver;
await resolver.promise;
}