Class ProfilerService

packages/editor-app/src/services/ProfilerService.ts:35–546

type AdvancedProfilerDataListener = (data: AdvancedProfilerDataPayload) => void;

export class ProfilerService implements IProfilerService {
    private ws: WebSocket | null = null;
    private isServerRunning = false;
    private wsPort: number;
    private listeners: Set<ProfilerDataListener> = new Set();
    private advancedListeners: Set<AdvancedProfilerDataListener> = new Set();
    private currentData: ProfilerData | null = null;
    private lastRawData: AdvancedProfilerDataPayload | null = null;
    private checkServerInterval: NodeJS.Timeout | null = null;
    private reconnectTimeout: NodeJS.Timeout | null = null;
    private clientIdMap: Map<string, string> = new Map();
    private autoStart: boolean;

    constructor() {
        const settings = SettingsService.getInstance();
        this.wsPort = settings.get('profiler.port', 8080);
        this.autoStart = settings.get('profiler.autoStart', true);

        this.startServerCheck();
        this.listenToSettingsChanges();

        // 如果设置了自动启动，则启动服务器
        if (this.autoStart) {
            this.manualStartServer();
        }
    }

    private listenToSettingsChanges(): void {
        window.addEventListener('settings:changed', ((event: CustomEvent) => {
            const newPort = event.detail['profiler.port'];
            if (newPort !== undefined && Number(newPort) !== this.wsPort) {
                console.log(`[ProfilerService] Port changed from ${this.wsPort} to ${newPort}`);
                this.wsPort = Number(newPort);
                this.reconnectWithNewPort();
            }
        }) as EventListener);
    }

    private async reconnectWithNewPort(): Promise<void> {
        this.disconnect();

        if (this.checkServerInterval) {
            clearInterval(this.checkServerInterval);
            this.checkServerInterval = null;
        }

        try {
            await invoke('stop_profiler_server');
            this.isServerRunning = false;
        } catch (error) {
            console.error('[ProfilerService] Failed to stop server:', error);
        }

        this.startServerCheck();
    }

    public subscribe(listener: ProfilerDataListener): () => void {
        this.listeners.add(listener);