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

feat(adb): buffer writes in sync push

Browse source
This commit is contained in:
Simon Chan 2023-03-05 21:51:59 +08:00
parent 228d2ab7d3
commit 1fb86f5eed
11 changed files with 308 additions and 229 deletions
Download patch file Download diff file Expand all files Collapse all files

83
libraries/adb/src/commands/sync/pull.ts
View file

@ -1,12 +1,10 @@
import type {
BufferedReadableStream,
WritableStreamDefaultWriter,
} from "@yume-chan/stream-extra";
import { ReadableStream } from "@yume-chan/stream-extra";
import type { ReadableStream } from "@yume-chan/stream-extra";
import { PushReadableStream } from "@yume-chan/stream-extra";
import Struct from "@yume-chan/struct";
import { AdbSyncRequestId, adbSyncWriteRequest } from "./request.js";
import { AdbSyncResponseId, adbSyncReadResponses } from "./response.js";
import type { AdbSyncSocket } from "./socket.js";
export const AdbSyncDataResponse = new Struct({ littleEndian: true })
.uint32("dataLength")
@ -16,47 +14,44 @@ export const AdbSyncDataResponse = new Struct({ littleEndian: true })
export type AdbSyncDataResponse =
(typeof AdbSyncDataResponse)["TDeserializeResult"];
export async function* adbSyncPullGenerator(
socket: AdbSyncSocket,
path: string
): AsyncGenerator<Uint8Array, void, void> {
const locked = await socket.lock();
let done = false;
try {
await adbSyncWriteRequest(locked, AdbSyncRequestId.Receive, path);
for await (const packet of adbSyncReadResponses(
locked,
AdbSyncResponseId.Data,
AdbSyncDataResponse
)) {
yield packet.data;
}
done = true;
} finally {
if (!done) {
// sync pull can't be cancelled, so we have to read all data
for await (const packet of adbSyncReadResponses(
locked,
AdbSyncResponseId.Data,
AdbSyncDataResponse
)) {
void packet;
}
}
locked.release();
}
}
export function adbSyncPull(
stream: BufferedReadableStream,
writer: WritableStreamDefaultWriter<Uint8Array>,
socket: AdbSyncSocket,
path: string
): ReadableStream<Uint8Array> {
let generator!: AsyncGenerator<AdbSyncDataResponse, void, void>;
return new ReadableStream<Uint8Array>(
{
async start() {
// TODO: If `ReadableStream.from(AsyncGenerator)` is added to spec, use it instead.
await adbSyncWriteRequest(
writer,
AdbSyncRequestId.Receive,
path
);
generator = adbSyncReadResponses(
stream,
AdbSyncResponseId.Data,
AdbSyncDataResponse
);
},
async pull(controller) {
const { done, value } = await generator.next();
if (done) {
controller.close();
return;
}
controller.enqueue(value.data);
},
cancel() {
generator.return().catch((e) => {
void e;
});
throw new Error(`Sync commands can't be canceled.`);
},
},
{
highWaterMark: 16 * 1024,
size(chunk) {
return chunk.byteLength;
},
return new PushReadableStream(async (controller) => {
for await (const data of adbSyncPullGenerator(socket, path)) {
await controller.enqueue(data);
}
);
});
}