abstract jpeg encoding away from "WSServer"

Additionally make thumbnail encoding threadpooled as well, just so it (probably) doesn't block as much.
This commit is contained in:
modeco80
2024-04-24 04:18:05 -04:00
parent ddae307874
commit a904f26961
5 changed files with 102 additions and 51 deletions

59
cvmts/src/JPEGEncoder.ts Normal file
View File

@@ -0,0 +1,59 @@
import path from 'node:path';
import Piscina from 'piscina';
import { Size, Rect } from '@cvmts/shared';
const kMaxJpegThreads = 4;
const kIdleTimeout = 25000;
// Thread pool for doing JPEG encoding for rects.
const TheJpegEncoderPool = new Piscina({
filename: path.join(import.meta.dirname + '/JPEGEncoderWorker.js'),
idleTimeout: kIdleTimeout,
maxThreads: kMaxJpegThreads
});
const TheThumbnailEncoderPool = new Piscina({
filename: path.join(import.meta.dirname + '/ThumbnailJPEGEncoderWorker.js'),
idleTimeout: kIdleTimeout,
maxThreads: kMaxJpegThreads
});
// A good balance. TODO: Configurable?
let gJpegQuality = 35;
export class JPEGEncoder {
static SetQuality(quality: number) {
gJpegQuality = quality;
}
static async EncodeJpeg(canvas: Buffer, displaySize: Size, rect: Rect): Promise<Buffer> {
let offset = (rect.y * displaySize.width + rect.x) * 4;
let res = await TheJpegEncoderPool.run({
buffer: canvas.subarray(offset),
width: rect.width,
height: rect.height,
stride: displaySize.width,
quality: gJpegQuality
});
// TODO: There's probably (definitely) a better way to fix this
if (res == undefined) return Buffer.from([]);
// have to manually turn it back into a buffer because
// Piscina for some reason turns it into a Uint8Array
return Buffer.from(res);
}
static async EncodeThumbnail(buffer: Buffer, size: Size) : Promise<Buffer> {
let res = await TheThumbnailEncoderPool.run({
buffer: buffer,
size: size,
quality: gJpegQuality
});
return Buffer.from(res)
}
}