openreplay/frontend/app/player/mobile/managers/IOSPerformanceTrackManager.ts

import ListWalker from "Player/common/ListWalker";
import type { IosPerformanceEvent } from "Player/web/messages";

const performanceEvTypes = {
  MemoryUsage: 'memoryUsage',
  MainThreadCPU: 'mainThreadCPU',
  Background: 'background',
}

export type PerformanceChartPoint = {
  time: number,
  cpu: number | null,
  memory: number | null,
  isBackground: boolean,
}

export default class IOSPerformanceTrackManager extends ListWalker<IosPerformanceEvent> {
  private chart: Array<PerformanceChartPoint> = [];
  private isInBg = false;

  lastData: { cpu: number | null, memory: number | null } = { cpu: null, memory: null };
  append(msg: IosPerformanceEvent): void {
    if (!Object.values(performanceEvTypes).includes(msg.name)) {
      return console.log('Unsupported performance event type', msg.name)
    }

    let cpu: number | null = null;
    let memory: number | null = null;
    if (msg.time < 0) msg.time = 1;
    if (msg.name === performanceEvTypes.Background) {
      // @ts-ignore
      const isBackground = msg.value === 1;
      if (isBackground === this.isInBg) return;
      this.isInBg = isBackground;
      this.chart.push({
        time: msg.time,
        cpu: null,
        memory: null,
        isBackground,
      })
      return super.append(msg);
    }
    if (msg.name === performanceEvTypes.MemoryUsage) {
      memory = msg.value;
      cpu = this.lastData.cpu;
      this.lastData.memory = memory;
    }
    if (msg.name === performanceEvTypes.MainThreadCPU) {
      cpu = msg.value;
      memory = this.lastData.memory;
      this.lastData.cpu = cpu;
    }

    this.chart.push({
      time: msg.time,
      cpu,
      memory,
      isBackground: false,
    });
    super.append(msg);
  }

  get chartData(): Array<PerformanceChartPoint> {
    return this.chart;
  }
}