Refactored a bit the code; the original code was too experimental in style.

This commit is contained in:
didier amyot committed 2023-06-13 08:00:01 -04:00
1 parent f7a8d494f5
commit bbd705ee3b
3 files changed
+55 -60

No files matched your search

+33 -40
View File
@@ -14,18 +14,16 @@ export interface FileService {
list(): AsyncIterable<FileMetaData>; list(): AsyncIterable<FileMetaData>;
createFromFullBody( createFromFullContent(
path: string, path: string,
body: string | Uint8Array | Deno.Reader, content: string | Uint8Array,
): Promise<FileMetaData>; ): Promise<FileMetaData>;
createFromReader(path: string, body: Deno.Reader): Promise<FileMetaData>;
createFromReader(path: string, content: Deno.Reader): Promise<FileMetaData>;
getFullBodyByPath(path: string): Promise<string | null>; getRawContentByPath(path: string): Promise<Uint8Array | null>;
getRawBodyByPath(path: string): Promise<Uint8Array | null>; getContentAsReader(path: string): Promise<Deno.Reader | null>;
getBodyAsReader(path: string): Promise<Deno.Reader | null>;
getMetaDataByID(id: string): Promise<FileMetaData | null>; getMetaDataByID(id: string): Promise<FileMetaData | null>;
@@ -42,14 +40,14 @@ export interface FileService {
interface SerializedFile { interface SerializedFile {
id: string; id: string;
path: string; path: string;
bodySize: number; contentSize: number;
compression: string; compression: string;
} }
export interface FileMetaData { export interface FileMetaData {
path: string; path: string;
id: string; id: string;
bodySize: number; contentSize: number;
} }
interface SerializedFilePathIndex { interface SerializedFilePathIndex {
@@ -73,36 +71,36 @@ export class DenoKVBlobService {
.join(""); // convert bytes to hex string .join(""); // convert bytes to hex string
} }
private *splitText(body: Uint8Array): Generator<Uint8Array> { private *splitText(content: Uint8Array): Generator<Uint8Array> {
let blobCount = 0; let blobCount = 0;
let txt = body.slice( let txt = content.slice(
blobCount * this.blobSize, blobCount * this.blobSize,
(blobCount + 1) * this.blobSize, (blobCount + 1) * this.blobSize,
); );
while (txt.length > 0) { while (txt.length > 0) {
yield txt; yield txt;
blobCount += 1; blobCount += 1;
txt = body.slice( txt = content.slice(
blobCount * this.blobSize, blobCount * this.blobSize,
(blobCount + 1) * this.blobSize, (blobCount + 1) * this.blobSize,
); );
} }
} }
private async unsafePersistBlob(mainIndex: string, index: number, body: Uint8Array): Promise<void> { private async unsafePersistBlob(mainIndex: string, index: number, content: Uint8Array): Promise<void> {
if(body.length > this.blobSize){ if(content.length > this.blobSize){
throw new Error('Persisted a too large blob'); throw new Error('Persisted a too large blob');
} }
const blob_id = await this.sha1(body); const blob_id = await this.sha1(content);
await this.kv.set(['blob_index', mainIndex, index], blob_id); await this.kv.set(['blob_index', mainIndex, index], blob_id);
await this.kv.set(['blob', blob_id], body); await this.kv.set(['blob', blob_id], content);
} }
async persistBlob(index: string, body: Uint8Array): Promise<number> { async persistBlob(index: string, content: Uint8Array): Promise<number> {
let blobCount = 0; let blobCount = 0;
let size = 0; let size = 0;
for (const txt of this.splitText(body)) { for (const txt of this.splitText(content)) {
await this.unsafePersistBlob(index, blobCount, txt); await this.unsafePersistBlob(index, blobCount, txt);
size += txt.length; size += txt.length;
blobCount += 1; blobCount += 1;
@@ -110,10 +108,10 @@ export class DenoKVBlobService {
return size; return size;
} }
async persistReader(index: string, body: Deno.Reader): Promise<number>{ async persistReader(index: string, content: Deno.Reader): Promise<number>{
let i = 0; let i = 0;
let size = 0; let size = 0;
for await (const chunk of iterateReader(body, {bufSize: this.blobSize})){ for await (const chunk of iterateReader(content, {bufSize: this.blobSize})){
await this.unsafePersistBlob(index, i, chunk); await this.unsafePersistBlob(index, i, chunk);
size += chunk.length; size += chunk.length;
i++; i++;
@@ -181,14 +179,14 @@ export class DenoKVFileService implements FileService {
this.blobService = new DenoKVBlobService(kv, 32 * 1024); this.blobService = new DenoKVBlobService(kv, 32 * 1024);
} }
createFromFullBody( createFromFullContent(
path: string, path: string,
body: string | Uint8Array, content: string | Uint8Array,
): Promise<FileMetaData> { ): Promise<FileMetaData> {
if (body instanceof Uint8Array){ if (content instanceof Uint8Array){
return this.createFromReader(path, new Buffer(body)); return this.createFromReader(path, new Buffer(content));
} }
return this.createFromReader(path, new StringReader(body)); return this.createFromReader(path, new StringReader(content));
} }
async createFromReader(path: string, content: Deno.Reader): Promise<FileMetaData>{ async createFromReader(path: string, content: Deno.Reader): Promise<FileMetaData>{
@@ -199,14 +197,14 @@ export class DenoKVFileService implements FileService {
const serializedFile: SerializedFile = { const serializedFile: SerializedFile = {
id: newId, id: newId,
path, path,
bodySize: size, contentSize: size,
compression: 'gzip' compression: 'gzip'
}; };
const idx: SerializedFilePathIndex = { fileId: newId }; const idx: SerializedFilePathIndex = { fileId: newId };
await this.kv.set(['file', newId], serializedFile); await this.kv.set(['file', newId], serializedFile);
await this.kv.set(['file_index_path', path], idx); await this.kv.set(['file_index_path', path], idx);
return { id: newId, path, bodySize: size }; return { id: newId, path, contentSize: size };
} }
@@ -224,7 +222,7 @@ export class DenoKVFileService implements FileService {
yield { yield {
path: ret.value.path, path: ret.value.path,
id: ret.value.id, id: ret.value.id,
bodySize: ret.value.bodySize, contentSize: ret.value.contentSize,
}; };
} }
} }
@@ -252,7 +250,7 @@ export class DenoKVFileService implements FileService {
return ret.value return ret.value
} }
private decompressBody(serializedFile: SerializedFile, content: Deno.Reader): Deno.Reader { private decompressContent(serializedFile: SerializedFile, content: Deno.Reader): Deno.Reader {
if(serializedFile.compression == 'gzip'){ if(serializedFile.compression == 'gzip'){
const ret = readableStreamFromReader(content).pipeThrough(new DecompressionStream('gzip')).getReader(); const ret = readableStreamFromReader(content).pipeThrough(new DecompressionStream('gzip')).getReader();
return readerFromStreamReader(ret); return readerFromStreamReader(ret);
@@ -262,18 +260,13 @@ export class DenoKVFileService implements FileService {
} }
async getFullBodyByPath(path: string): Promise<string | null> { async getRawContentByPath(path: string): Promise<Uint8Array | null> {
const ret = await this.getRawBodyByPath(path); const ret = await this.getContentAsReader(path);
return ret == null ? null : new TextDecoder().decode(ret);
}
async getRawBodyByPath(path: string): Promise<Uint8Array | null> {
const ret = await this.getBodyAsReader(path);
return ret === null ? null : readAll(ret); return ret === null ? null : readAll(ret);
} }
async getBodyAsReader(path: string): Promise<Deno.Reader | null> { async getContentAsReader(path: string): Promise<Deno.Reader | null> {
const idx = await this.kv.get<SerializedFilePathIndex>([ const idx = await this.kv.get<SerializedFilePathIndex>([
'file_index_path', 'file_index_path',
path, path,
@@ -285,7 +278,7 @@ export class DenoKVFileService implements FileService {
if(ser.value === null){ if(ser.value === null){
return null; return null;
} }
return this.decompressBody(ser.value, this.blobService.getBlob(idx.value.fileId)); return this.decompressContent(ser.value, this.blobService.getBlob(idx.value.fileId));
} }
@@ -297,7 +290,7 @@ export class DenoKVFileService implements FileService {
return { return {
path: fileObject.path, path: fileObject.path,
id: fileObject.id, id: fileObject.id,
bodySize: fileObject.bodySize, contentSize: fileObject.contentSize,
}; };
} }
@@ -313,7 +306,7 @@ export class DenoKVFileService implements FileService {
const newFile: SerializedFile = { const newFile: SerializedFile = {
path: newPath, path: newPath,
id: fileObject.value.id, id: fileObject.value.id,
bodySize: fileObject.value.bodySize, contentSize: fileObject.value.contentSize,
compression: fileObject.value.compression, compression: fileObject.value.compression,
}; };
await this.kv.atomic() await this.kv.atomic()
+1 -1
View File
@@ -30,7 +30,7 @@ function getFileHandlerFactory(fileService: FileService): Handler {
return async (ctx: Context) => { return async (ctx: Context) => {
const untypedCtx: any = ctx; const untypedCtx: any = ctx;
const fileId = untypedCtx.params.path; const fileId = untypedCtx.params.path;
const reader = await fileService.getBodyAsReader(fileId); const reader = await fileService.getContentAsReader(fileId);
if(reader == null){ if(reader == null){
ctx.response.status = 404; ctx.response.status = 404;
return; return;
+21 -19
View File
@@ -24,8 +24,8 @@ Deno.test("DenoKVBlobService happy path", async (t) => {
await t.step("read write small text", async () => { await t.step("read write small text", async () => {
const size = await bs.persistBlob("0002", toUint8Array("hello world")); const size = await bs.persistBlob("0002", toUint8Array("hello world"));
const body = await bs.readAllBlob("0002"); const content = await bs.readAllBlob("0002");
assertEquals(body, toUint8Array("hello world")); assertEquals(content, toUint8Array("hello world"));
assertEquals(await bs.getNumberOfBlobs("0002"), 1); assertEquals(await bs.getNumberOfBlobs("0002"), 1);
assertEquals(size, "hello world".length); assertEquals(size, "hello world".length);
}); });
@@ -33,8 +33,8 @@ Deno.test("DenoKVBlobService happy path", async (t) => {
await t.step("read write large text", async () => { await t.step("read write large text", async () => {
const largeText = toUint8Array(generateLargeText(33, 100)); const largeText = toUint8Array(generateLargeText(33, 100));
const size = await bs.persistBlob("0001", largeText); const size = await bs.persistBlob("0001", largeText);
const body = await bs.readAllBlob("0001"); const content = await bs.readAllBlob("0001");
assertEquals(body, largeText); assertEquals(content, largeText);
assertEquals(await bs.getNumberOfBlobs("0001"), 21); assertEquals(await bs.getNumberOfBlobs("0001"), 21);
assertEquals(size, largeText.length); assertEquals(size, largeText.length);
}); });
@@ -61,15 +61,15 @@ Deno.test("DenoKVFileService happy path xxx ", async (t) => {
let file2: FileMetaData | undefined; let file2: FileMetaData | undefined;
await t.step("Create 2 files xyzabc", async () => { await t.step("Create 2 files xyzabc", async () => {
file1 = await fs.createFromFullBody(fileName1, "bob le chef"); file1 = await fs.createFromFullContent(fileName1, "bob le chef");
file2 = await fs.createFromFullBody(fileName2, "le geant vert ☀️"); file2 = await fs.createFromFullContent(fileName2, "le geant vert ☀️");
assertEquals(file1.path, fileName1); assertEquals(file1.path, fileName1);
assertEquals(file2.bodySize, 47); assertEquals(file2.contentSize, 47);
}); });
await t.step("Get file metadata", async () => { await t.step("Get file metadata", async () => {
const ret = await fs.getMetaDataByID(file1?.id ?? "404"); const ret = await fs.getMetaDataByID(file1!.id);
assertEquals(ret, file1); assertEquals(ret, file1);
}); });
@@ -80,11 +80,13 @@ Deno.test("DenoKVFileService happy path xxx ", async (t) => {
}); });
await t.step("Get File by path", async () => { await t.step("Get File by path", async () => {
assertEquals(await fs.getFullBodyByPath(fileName1), "bob le chef"); const ret = await fs.getRawContentByPath(fileName1);
assertEquals(new TextDecoder().decode(ret!), "bob le chef");
}); });
await t.step("Get File by Id", async () => { await t.step("Get File by Id", async () => {
assertEquals(await fs.getFullBodyByPath(fileName1), "bob le chef"); const ret = await fs.getRawContentByPath(fileName1);
assertEquals(new TextDecoder().decode(ret!), "bob le chef");
}); });
await t.step("Delete File", async () => { await t.step("Delete File", async () => {
@@ -102,19 +104,19 @@ Deno.test("DenoKVFileService happy path xxx ", async (t) => {
const newPath = "boris/streamed-file.txt"; const newPath = "boris/streamed-file.txt";
const ret = await fs.createFromReader(newPath, new StringReader("hello world 18")); const ret = await fs.createFromReader(newPath, new StringReader("hello world 18"));
assertEquals(ret.bodySize, 40); assertEquals(ret.contentSize, 40);
const s = await fs.getRawBodyByPath(newPath); const s = await fs.getRawContentByPath(newPath);
assertEquals(new TextDecoder().decode(s!), "hello world 18") assertEquals(new TextDecoder().decode(s!), "hello world 18")
}); });
await t.step("Rename file", async () => { await t.step("Rename file", async () => {
const newPath = "boris/file2.txt"; const newPath = "boris/file2.txt";
const ret = await fs.renameById(file2?.id ?? "404", newPath); const ret = await fs.renameById(file2!.id, newPath);
assertEquals(ret?.path, newPath); assertEquals(ret?.path, newPath);
assertEquals( assertEquals(
await fs.getFullBodyByPath(newPath), new TextDecoder().decode((await fs.getRawContentByPath(newPath))!),
'le geant vert ☀️', 'le geant vert ☀️',
); );
}); });
@@ -126,14 +128,14 @@ Deno.test("DenoKVFileService happy path with byte array", async (t) => {
const kv = await Deno.openKv(":memory:"); const kv = await Deno.openKv(":memory:");
const fs: FileService = new DenoKVFileService(kv); const fs: FileService = new DenoKVFileService(kv);
await t.step("Create and read 1 binary file", async () => { await t.step("Create and read 1 binary file", async () => {
const inputBody = new Uint8Array([240]); const inputContent = new Uint8Array([240]);
const file = await fs.createFromFullBody("hello", inputBody); const file = await fs.createFromFullContent("hello", inputContent);
assertEquals(file.path, "hello"); assertEquals(file.path, "hello");
assertEquals(file.bodySize, 27); assertEquals(file.contentSize, 27);
const bodyString = await fs.getFullBodyByPath("hello"); const persistedContent = await fs.getRawContentByPath("hello");
assertEquals(bodyString, new TextDecoder().decode(inputBody)); assertEquals(persistedContent, inputContent);
}); });
kv.close(); kv.close();