Class GraphReducerImpl

browser/src/service/GraphSizeService.tsx:51–100

}

class GraphReducerImpl implements GraphReducer {
  private classSize: ClassSize;
  private classSizeService: ClassSizeService;

  /**
   * Tracks nodes that have already been reduced. This is necessary to count Dagger modules
   * just once. They may be a common dependency and counted multiple times.
   */
  private reduced: Set<string>;

  constructor(classSizeService: ClassSizeService, initialKey: string) {
    this.classSizeService = classSizeService;
    this.reduced = new Set();

    const classInfo = {
      'field_count': 0,
      'method_count': 0,
      'lambda_count': 0,
      'code_size': 0,
      'inner_class_count': 0
    } as ClassInfo;
    this.classSize = new ClassSize(classInfo);
    const parentClassSize = this.classSizeService.getClassSize(initialKey);
    if (parentClassSize) {
      this.classSize.add(parentClassSize);
    }
  }

  reduce(key: string) {
    if (this.reduced.has(key)) {
      return;
    }

    this.reduced.add(key);
    const childClassSize = this.classSizeService.getClassSize(key);
    if (childClassSize) {
      this.classSize.add(childClassSize);
    }
  }

  result(): Weight {
    return {
      value: Math.round(this.classSize.getMemorySize() / 1024),
      largeThreshold: LARGE_THRESHOLD_KB,
      smallThreshold: SMALL_THRESHOLD_KB,
      summary: this.classSize.getSummary(),
      unit: 'kB'
    } as Weight;
  }
}