mirror of
https://github.com/denoland/deno.git
synced 2024-10-30 09:08:00 -04:00
39 lines
1.1 KiB
TypeScript
39 lines
1.1 KiB
TypeScript
|
// Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
|
||
|
import { Reader, ReadResult } from "deno";
|
||
|
import { encode } from "../strings/strings.ts";
|
||
|
|
||
|
/** Reader utility for strings */
|
||
|
export class StringReader implements Reader {
|
||
|
private offs = 0;
|
||
|
private buf = new Uint8Array(encode(this.s));
|
||
|
|
||
|
constructor(private readonly s: string) {}
|
||
|
|
||
|
async read(p: Uint8Array): Promise<ReadResult> {
|
||
|
const n = Math.min(p.byteLength, this.buf.byteLength - this.offs);
|
||
|
p.set(this.buf.slice(this.offs, this.offs + n));
|
||
|
this.offs += n;
|
||
|
return { nread: n, eof: this.offs === this.buf.byteLength };
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/** Reader utility for combining multiple readers */
|
||
|
export class MultiReader implements Reader {
|
||
|
private readonly readers: Reader[];
|
||
|
private currentIndex = 0;
|
||
|
|
||
|
constructor(...readers: Reader[]) {
|
||
|
this.readers = readers;
|
||
|
}
|
||
|
|
||
|
async read(p: Uint8Array): Promise<ReadResult> {
|
||
|
const r = this.readers[this.currentIndex];
|
||
|
if (!r) return { nread: 0, eof: true };
|
||
|
const { nread, eof } = await r.read(p);
|
||
|
if (eof) {
|
||
|
this.currentIndex++;
|
||
|
}
|
||
|
return { nread, eof: false };
|
||
|
}
|
||
|
}
|