mirror/ditto
mirror/ditto
1
0
Fork 0
mirror of https://gitlab.com/soapbox-pub/ditto.git synced 2025-12-06 03:19:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Measure performance of uploads

Browse source
This commit is contained in:
Alex Gleason 2025-03-01 13:07:36 -06:00
parent bd4c94852f
commit 9051a59733
No known key found for this signature in database
GPG key ID: 7211D1F99744FBB7
1 changed files with 36 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
packages/ditto/utils/upload.ts
View file

@ -1,4 +1,5 @@
import { analyzeFile, extractVideoFrame, transcodeVideo } from '@ditto/transcode'; import { analyzeFile, extractVideoFrame, transcodeVideo } from '@ditto/transcode';
import { ScopedPerformance } from '@esroyo/scoped-performance';
import { HTTPException } from '@hono/hono/http-exception'; import { HTTPException } from '@hono/hono/http-exception';
import { logi } from '@soapbox/logi'; import { logi } from '@soapbox/logi';
import { crypto } from '@std/crypto'; import { crypto } from '@std/crypto';
@ -22,6 +23,9 @@ export async function uploadFile(
meta: FileMeta, meta: FileMeta,
signal?: AbortSignal, signal?: AbortSignal,
): Promise<DittoUpload> { ): Promise<DittoUpload> {
using perf = new ScopedPerformance();
perf.mark('start');
const { conf, uploader } = c.var; const { conf, uploader } = c.var;
if (!uploader) { if (!uploader) {
@ -38,8 +42,11 @@ export async function uploadFile(
const [baseType] = file.type.split('/'); const [baseType] = file.type.split('/');
perf.mark('probe-start');
const probe = await analyzeFile(file.stream()).catch(() => null); const probe = await analyzeFile(file.stream()).catch(() => null);
perf.mark('probe-end');
perf.mark('transcode-start');
if (baseType === 'video') { if (baseType === 'video') {
let needsTranscode = false; let needsTranscode = false;
@ -60,10 +67,16 @@ export async function uploadFile(
file = new File([transcoded], file.name, file); file = new File([transcoded], file.name, file);
} }
} }
perf.mark('transcode-end');
perf.mark('upload-start');
const tags = await uploader.upload(file, { signal }); const tags = await uploader.upload(file, { signal });
perf.mark('upload-end');
const url = tags[0][1]; const url = tags[0][1];
perf.mark('analyze-start');
if (description) { if (description) {
tags.push(['alt', description]); tags.push(['alt', description]);
} }
@ -154,6 +167,8 @@ export async function uploadFile(
} }
} }
perf.mark('analyze-end');
const upload = { const upload = {
id: crypto.randomUUID(), id: crypto.randomUUID(),
url, url,
@ -164,5 +179,26 @@ export async function uploadFile(
dittoUploads.set(upload.id, upload); dittoUploads.set(upload.id, upload);
perf.mark('end');
const timing = [
perf.measure('probe', 'probe-start', 'probe-end'),
perf.measure('transcode', 'transcode-start', 'transcode-end'),
perf.measure('upload', 'upload-start', 'upload-end'),
perf.measure('analyze', 'analyze-start', 'analyze-end'),
].reduce<Record<string, number>>((acc, m) => {
const name = m.name.split('::')[1]; // ScopedPerformance uses `::` to separate the name.
acc[name] = m.duration / 1000; // Convert to seconds for logging.
return acc;
}, {});
logi({
level: 'info',
ns: 'ditto.upload',
upload: { ...upload, uploadedAt: upload.uploadedAt.toISOString() },
timing,
duration: perf.measure('total', 'start', 'end').duration / 1000,
});
return upload; return upload;
} }