mirror of
https://github.com/Ark0N/Codeman.git
synced 2026-10-04 14:39:42 +02:00
perf: Operation Lightspeed — 5 parallel performance optimizations
1. Session-scoped SSE subscriptions: server filters events by session ID, clients can subscribe via ?sessions=id1,id2 (backwards-compatible) 2. Lazy xterm.js for subagent windows: terminals created on restore, disposed on minimize — saves ~3.75MB DOM at 50 agents 3. Targeted badge updates: badge count changes update the <span> directly instead of rebuilding the entire session tab sidebar (O(1) vs O(n)) 4. Conditional SSE padding: 8KB Cloudflare padding only on session:terminal and session:needsRefresh, not every event (~70% bandwidth reduction) 5. Canvas renderer on mobile: skip WebGL addon on mobile devices to reduce GPU pressure and prevent context loss on weaker mobile GPUs All 5 implemented in parallel via isolated git worktrees, merged conflict-free. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,445 @@
|
||||
import { describe, it, expect, beforeAll, afterAll } from 'vitest';
|
||||
import { WebServer } from '../src/web/server.js';
|
||||
|
||||
const TEST_PORT = 3212;
|
||||
|
||||
// Helper to parse SSE events from raw text
|
||||
function parseSSEEvents(text: string): Array<{ event: string; data: unknown }> {
|
||||
const events: Array<{ event: string; data: unknown }> = [];
|
||||
const lines = text.split('\n');
|
||||
let currentEvent = '';
|
||||
let currentData = '';
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith('event: ')) {
|
||||
currentEvent = line.substring(7);
|
||||
} else if (line.startsWith('data: ')) {
|
||||
currentData = line.substring(6);
|
||||
} else if (line === '') {
|
||||
if (currentEvent && currentData) {
|
||||
try {
|
||||
events.push({ event: currentEvent, data: JSON.parse(currentData) });
|
||||
} catch {
|
||||
events.push({ event: currentEvent, data: currentData });
|
||||
}
|
||||
}
|
||||
currentEvent = '';
|
||||
currentData = '';
|
||||
}
|
||||
}
|
||||
|
||||
return events;
|
||||
}
|
||||
|
||||
// Helper to collect SSE events for a given duration
|
||||
async function collectSSEEvents(
|
||||
baseUrl: string,
|
||||
queryParams: string,
|
||||
durationMs: number
|
||||
): Promise<Array<{ event: string; data: unknown }>> {
|
||||
const controller = new AbortController();
|
||||
let receivedData = '';
|
||||
|
||||
const fetchPromise = fetch(`${baseUrl}/api/events${queryParams}`, {
|
||||
signal: controller.signal,
|
||||
}).then(async (response) => {
|
||||
const reader = response.body?.getReader();
|
||||
if (reader) {
|
||||
try {
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
receivedData += new TextDecoder().decode(value);
|
||||
}
|
||||
} catch {
|
||||
/* AbortError expected */
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
await new Promise((resolve) => setTimeout(resolve, durationMs));
|
||||
controller.abort();
|
||||
try {
|
||||
await fetchPromise;
|
||||
} catch {
|
||||
/* AbortError expected */
|
||||
}
|
||||
|
||||
return parseSSEEvents(receivedData);
|
||||
}
|
||||
|
||||
describe('SSE Subscription Filtering', () => {
|
||||
let server: WebServer;
|
||||
let baseUrl: string;
|
||||
|
||||
beforeAll(async () => {
|
||||
server = new WebServer(TEST_PORT, false, true);
|
||||
await server.start();
|
||||
baseUrl = `http://localhost:${TEST_PORT}`;
|
||||
});
|
||||
|
||||
afterAll(async () => {
|
||||
await server.stop();
|
||||
}, 60000);
|
||||
|
||||
it('should accept sessions query parameter without error', async () => {
|
||||
const controller = new AbortController();
|
||||
const timeout = setTimeout(() => controller.abort(), 500);
|
||||
|
||||
try {
|
||||
const response = await fetch(`${baseUrl}/api/events?sessions=abc,def`, {
|
||||
signal: controller.signal,
|
||||
});
|
||||
|
||||
expect(response.headers.get('content-type')).toBe('text/event-stream');
|
||||
} catch (err: any) {
|
||||
if (err.name !== 'AbortError') throw err;
|
||||
} finally {
|
||||
clearTimeout(timeout);
|
||||
}
|
||||
});
|
||||
|
||||
it('should send init event regardless of session filter', async () => {
|
||||
const events = await collectSSEEvents(baseUrl, '?sessions=nonexistent-id', 500);
|
||||
|
||||
const initEvent = events.find((e) => e.event === 'init');
|
||||
expect(initEvent).toBeDefined();
|
||||
expect((initEvent?.data as any).sessions).toBeDefined();
|
||||
});
|
||||
|
||||
it('should receive all events when no sessions param is provided (backwards-compatible)', async () => {
|
||||
// Start two SSE listeners: one with no filter, one with a filter for a nonexistent session
|
||||
const controller1 = new AbortController();
|
||||
const controller2 = new AbortController();
|
||||
let unfilteredData = '';
|
||||
let filteredData = '';
|
||||
|
||||
const fetch1 = fetch(`${baseUrl}/api/events`, {
|
||||
signal: controller1.signal,
|
||||
}).then(async (response) => {
|
||||
const reader = response.body?.getReader();
|
||||
if (reader) {
|
||||
try {
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
unfilteredData += new TextDecoder().decode(value);
|
||||
}
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
const fetch2 = fetch(`${baseUrl}/api/events?sessions=nonexistent-session`, {
|
||||
signal: controller2.signal,
|
||||
}).then(async (response) => {
|
||||
const reader = response.body?.getReader();
|
||||
if (reader) {
|
||||
try {
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
filteredData += new TextDecoder().decode(value);
|
||||
}
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// Wait for connections to establish
|
||||
await new Promise((resolve) => setTimeout(resolve, 200));
|
||||
|
||||
// Create a session — this emits session:created (a global-ish event that has id but
|
||||
// the nonexistent filter won't match it)
|
||||
const createRes = await fetch(`${baseUrl}/api/sessions`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ workingDir: '/tmp' }),
|
||||
});
|
||||
const createData = await createRes.json();
|
||||
const sessionId = createData.session.id;
|
||||
|
||||
// Wait for events to arrive
|
||||
await new Promise((resolve) => setTimeout(resolve, 300));
|
||||
|
||||
// Stop listening
|
||||
controller1.abort();
|
||||
controller2.abort();
|
||||
try {
|
||||
await fetch1;
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
try {
|
||||
await fetch2;
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
|
||||
const unfilteredEvents = parseSSEEvents(unfilteredData);
|
||||
const filteredEvents = parseSSEEvents(filteredData);
|
||||
|
||||
// Unfiltered client should receive the session:created event
|
||||
const unfilteredCreated = unfilteredEvents.find((e) => e.event === 'session:created');
|
||||
expect(unfilteredCreated).toBeDefined();
|
||||
expect((unfilteredCreated?.data as any).id).toBe(sessionId);
|
||||
|
||||
// Filtered client (subscribed to nonexistent-session) should NOT receive session:created
|
||||
// because session:created has an `id` field that doesn't match the filter
|
||||
const filteredCreated = filteredEvents.find((e) => e.event === 'session:created');
|
||||
expect(filteredCreated).toBeUndefined();
|
||||
|
||||
// Both should have received the init event (it has no sessionId)
|
||||
expect(unfilteredEvents.find((e) => e.event === 'init')).toBeDefined();
|
||||
expect(filteredEvents.find((e) => e.event === 'init')).toBeDefined();
|
||||
|
||||
// Cleanup
|
||||
await fetch(`${baseUrl}/api/sessions/${sessionId}`, { method: 'DELETE' });
|
||||
});
|
||||
|
||||
it('should deliver session events to a client subscribed to that session', async () => {
|
||||
// First create a session so we know the ID
|
||||
const createRes = await fetch(`${baseUrl}/api/sessions`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ workingDir: '/tmp' }),
|
||||
});
|
||||
const createData = await createRes.json();
|
||||
const sessionId = createData.session.id;
|
||||
|
||||
// Now connect SSE with the session filter
|
||||
const controller = new AbortController();
|
||||
let receivedData = '';
|
||||
|
||||
const fetchPromise = fetch(`${baseUrl}/api/events?sessions=${sessionId}`, {
|
||||
signal: controller.signal,
|
||||
}).then(async (response) => {
|
||||
const reader = response.body?.getReader();
|
||||
if (reader) {
|
||||
try {
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
receivedData += new TextDecoder().decode(value);
|
||||
}
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// Wait for connection
|
||||
await new Promise((resolve) => setTimeout(resolve, 200));
|
||||
|
||||
// Delete the session — this emits session:deleted with { id: sessionId }
|
||||
await fetch(`${baseUrl}/api/sessions/${sessionId}`, { method: 'DELETE' });
|
||||
|
||||
// Wait for events
|
||||
await new Promise((resolve) => setTimeout(resolve, 300));
|
||||
|
||||
controller.abort();
|
||||
try {
|
||||
await fetchPromise;
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
|
||||
const events = parseSSEEvents(receivedData);
|
||||
|
||||
// Should receive session:deleted because we're subscribed to this session
|
||||
const deletedEvent = events.find((e) => e.event === 'session:deleted');
|
||||
expect(deletedEvent).toBeDefined();
|
||||
expect((deletedEvent?.data as any).id).toBe(sessionId);
|
||||
});
|
||||
|
||||
it('should filter out events for sessions not in the subscription', async () => {
|
||||
// Create two sessions
|
||||
const createRes1 = await fetch(`${baseUrl}/api/sessions`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ workingDir: '/tmp' }),
|
||||
});
|
||||
const session1 = (await createRes1.json()).session;
|
||||
|
||||
const createRes2 = await fetch(`${baseUrl}/api/sessions`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ workingDir: '/tmp' }),
|
||||
});
|
||||
const session2 = (await createRes2.json()).session;
|
||||
|
||||
// Connect SSE subscribed ONLY to session1
|
||||
const controller = new AbortController();
|
||||
let receivedData = '';
|
||||
|
||||
const fetchPromise = fetch(`${baseUrl}/api/events?sessions=${session1.id}`, {
|
||||
signal: controller.signal,
|
||||
}).then(async (response) => {
|
||||
const reader = response.body?.getReader();
|
||||
if (reader) {
|
||||
try {
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
receivedData += new TextDecoder().decode(value);
|
||||
}
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// Wait for connection
|
||||
await new Promise((resolve) => setTimeout(resolve, 200));
|
||||
|
||||
// Delete both sessions — emits session:deleted for each
|
||||
await fetch(`${baseUrl}/api/sessions/${session1.id}`, { method: 'DELETE' });
|
||||
await fetch(`${baseUrl}/api/sessions/${session2.id}`, { method: 'DELETE' });
|
||||
|
||||
// Wait for events
|
||||
await new Promise((resolve) => setTimeout(resolve, 300));
|
||||
|
||||
controller.abort();
|
||||
try {
|
||||
await fetchPromise;
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
|
||||
const events = parseSSEEvents(receivedData);
|
||||
|
||||
// Should receive session:deleted for session1
|
||||
const deleted1 = events.find((e) => e.event === 'session:deleted' && (e.data as any).id === session1.id);
|
||||
expect(deleted1).toBeDefined();
|
||||
|
||||
// Should NOT receive session:deleted for session2
|
||||
const deleted2 = events.find((e) => e.event === 'session:deleted' && (e.data as any).id === session2.id);
|
||||
expect(deleted2).toBeUndefined();
|
||||
});
|
||||
|
||||
it('should support subscribing to multiple sessions', async () => {
|
||||
// Create two sessions
|
||||
const createRes1 = await fetch(`${baseUrl}/api/sessions`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ workingDir: '/tmp' }),
|
||||
});
|
||||
const session1 = (await createRes1.json()).session;
|
||||
|
||||
const createRes2 = await fetch(`${baseUrl}/api/sessions`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ workingDir: '/tmp' }),
|
||||
});
|
||||
const session2 = (await createRes2.json()).session;
|
||||
|
||||
// Connect SSE subscribed to both sessions
|
||||
const controller = new AbortController();
|
||||
let receivedData = '';
|
||||
|
||||
const fetchPromise = fetch(`${baseUrl}/api/events?sessions=${session1.id},${session2.id}`, {
|
||||
signal: controller.signal,
|
||||
}).then(async (response) => {
|
||||
const reader = response.body?.getReader();
|
||||
if (reader) {
|
||||
try {
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
receivedData += new TextDecoder().decode(value);
|
||||
}
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// Wait for connection
|
||||
await new Promise((resolve) => setTimeout(resolve, 200));
|
||||
|
||||
// Delete both sessions
|
||||
await fetch(`${baseUrl}/api/sessions/${session1.id}`, { method: 'DELETE' });
|
||||
await fetch(`${baseUrl}/api/sessions/${session2.id}`, { method: 'DELETE' });
|
||||
|
||||
// Wait for events
|
||||
await new Promise((resolve) => setTimeout(resolve, 300));
|
||||
|
||||
controller.abort();
|
||||
try {
|
||||
await fetchPromise;
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
|
||||
const events = parseSSEEvents(receivedData);
|
||||
|
||||
// Should receive session:deleted for both sessions
|
||||
const deleted1 = events.find((e) => e.event === 'session:deleted' && (e.data as any).id === session1.id);
|
||||
const deleted2 = events.find((e) => e.event === 'session:deleted' && (e.data as any).id === session2.id);
|
||||
expect(deleted1).toBeDefined();
|
||||
expect(deleted2).toBeDefined();
|
||||
});
|
||||
|
||||
it('should deliver global events (no sessionId) to filtered clients', async () => {
|
||||
// Connect with a filter — global events like case:created should still arrive
|
||||
const controller = new AbortController();
|
||||
let receivedData = '';
|
||||
|
||||
const fetchPromise = fetch(`${baseUrl}/api/events?sessions=some-session-id`, {
|
||||
signal: controller.signal,
|
||||
}).then(async (response) => {
|
||||
const reader = response.body?.getReader();
|
||||
if (reader) {
|
||||
try {
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
receivedData += new TextDecoder().decode(value);
|
||||
}
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// Wait for connection
|
||||
await new Promise((resolve) => setTimeout(resolve, 200));
|
||||
|
||||
// Create a case — case:created is a global event (no sessionId or id matching a session)
|
||||
const caseName = `test-filter-case-${Date.now()}`;
|
||||
await fetch(`${baseUrl}/api/cases`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ name: caseName }),
|
||||
});
|
||||
|
||||
// Wait for events
|
||||
await new Promise((resolve) => setTimeout(resolve, 300));
|
||||
|
||||
controller.abort();
|
||||
try {
|
||||
await fetchPromise;
|
||||
} catch {
|
||||
/* expected */
|
||||
}
|
||||
|
||||
const events = parseSSEEvents(receivedData);
|
||||
|
||||
// case:created should arrive because it has no sessionId — it's a global event
|
||||
const caseCreated = events.find((e) => e.event === 'case:created');
|
||||
expect(caseCreated).toBeDefined();
|
||||
expect((caseCreated?.data as any).name).toBe(caseName);
|
||||
|
||||
// Cleanup
|
||||
const { rmSync } = await import('node:fs');
|
||||
const { join } = await import('node:path');
|
||||
const { homedir } = await import('node:os');
|
||||
try {
|
||||
rmSync(join(homedir(), 'codeman-cases', caseName), { recursive: true });
|
||||
} catch {
|
||||
/* may not exist */
|
||||
}
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user