angular-cn/packages/service-worker/worker/src/idle.ts

/**
 * @license
 * Copyright Google Inc. All Rights Reserved.
 *
 * Use of this source code is governed by an MIT-style license that can be
 * found in the LICENSE file at https://angular.io/license
 */

import {Adapter} from './adapter';

export interface IdleTask {
  run: () => Promise<void>;
  desc: string;
}

interface ScheduledRun {
  cancel: boolean;
}

export class IdleScheduler {
  private queue: IdleTask[] = [];
  private scheduled: ScheduledRun|null = null;
  empty: Promise<void> = Promise.resolve();
  private emptyResolve: Function|null = null;
  lastTrigger: number|null = null;
  lastRun: number|null = null;

  constructor(private adapter: Adapter, private threshold: number) {}

  async trigger(): Promise<void> {
    this.lastTrigger = this.adapter.time;
    if (this.queue.length === 0) {
      return;
    }

    if (this.scheduled !== null) {
      this.scheduled.cancel = true;
    }

    this.scheduled = {
      cancel: false,
    };

    await this.adapter.timeout(this.threshold);

    if (this.scheduled !== null && this.scheduled.cancel) {
      this.scheduled = null;
      return;
    }

    this.scheduled = null;

    await this.execute();
  }

  async execute(): Promise<void> {
    this.lastRun = this.adapter.time;
    while (this.queue.length > 0) {
      const queue = this.queue.map(task => {
        try {
          return task.run();
        } catch (e) {
          // Ignore errors, for now.
          return Promise.resolve();
        }
      });

      this.queue = [];

      await Promise.all(queue);
      if (this.emptyResolve !== null) {
        this.emptyResolve();
        this.emptyResolve = null;
      }
      this.empty = Promise.resolve();
    }
  }

  schedule(desc: string, run: () => Promise<void>): void {
    this.queue.push({desc, run});
    if (this.emptyResolve === null) {
      this.empty = new Promise(resolve => { this.emptyResolve = resolve; });
    }
  }

  get size(): number { return this.queue.length; }

  get taskDescriptions(): string[] { return this.queue.map(task => task.desc); }
}