angular-docs-cn/packages/zone.js/test/performance/performance_setup.js

/**
 * @license
 * Copyright Google LLC 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
 */
(function(_global) {
var allTasks = _global['__zone_symbol__performance_tasks'];
if (!allTasks) {
  allTasks = _global['__zone_symbol__performance_tasks'] = [];
}

var mark = _global['__zone_symbol__mark'] = function(name) {
  performance && performance['mark'] && performance['mark'](name);
};

var measure = _global['__zone_symbol__measure'] = function(name, label) {
  performance && performance['measure'] && performance['measure'](name, label);
};

var getEntries = _global['__zone_symbol__getEntries'] = function() {
  performance && performance['getEntries'] && performance['getEntries']();
};

var getEntriesByName = _global['__zone_symbol__getEntriesByName'] = function(name) {
  return performance && performance['getEntriesByName'] && performance['getEntriesByName'](name);
};

var clearMarks = _global['__zone_symbol__clearMarks'] = function(name) {
  return performance && performance['clearMarks'] && performance['clearMarks'](name);
};

var clearMeasures = _global['__zone_symbol__clearMeasures'] = function(name) {
  return performance && performance['clearMeasures'] && performance['clearMeasures'](name);
};

var averageMeasures = _global['__zone_symbol__averageMeasures'] = function(name, times) {
  var sum = _global['__zone_symbol__getEntriesByName'](name)
                .filter(function(m) {
                  return m.entryType === 'measure';
                })
                .map(function(m) {
                  return m.duration
                })
                .reduce(function(sum, d) {
                  return sum + d;
                });
  return sum / times;
};

var serialPromise = _global['__zone_symbol__serialPromise'] =
    function(promiseFactories) {
  let lastPromise;
  for (var i = 0; i < promiseFactories.length; i++) {
    var promiseFactory = promiseFactories[i];
    if (!lastPromise) {
      lastPromise = promiseFactory.factory(promiseFactory.context).then(function(value) {
        return {value, idx: 0};
      });
    } else {
      lastPromise = lastPromise.then(function(ctx) {
        var idx = ctx.idx + 1;
        var promiseFactory = promiseFactories[idx];
        return promiseFactory.factory(promiseFactory.context).then(function(value) {
          return {value, idx};
        });
      });
    }
  }
  return lastPromise;
}

var callbackContext = _global['__zone_symbol__callbackContext'] = {};
var zone = _global['__zone_symbol__callbackZone'] = Zone.current.fork({
  name: 'callback',
  onScheduleTask: function(delegate, curr, target, task) {
    delegate.scheduleTask(target, task);
    if (task.type === callbackContext.type && task.source.indexOf(callbackContext.source) !== -1) {
      if (task.type === 'macroTask' || task.type === 'eventTask') {
        var invoke = task.invoke;
        task.invoke = function() {
          mark(callbackContext.measureName);
          var result = invoke.apply(this, arguments);
          measure(callbackContext.measureName, callbackContext.measureName);
          return result;
        };
      } else if (task.type === 'microTask') {
        var callback = task.callback;
        task.callback = function() {
          mark(callbackContext.measureName);
          var result = callback.apply(this, arguments);
          measure(callbackContext.measureName, callbackContext.measureName);
          return result;
        };
      }
    }
    return task;
  }
});

var runAsync = _global['__zone_symbol__runAsync'] = function(testFn, times, _delay) {
  var delay = _delay | 100;
  const fnPromise = function() {
    return new Promise(function(res, rej) {
      // run test with a setTimeout
      // several times to decrease measurement error
      setTimeout(function() {
        testFn().then(function() {
          res();
        });
      }, delay);
    });
  };
  var promiseFactories = [];
  for (var i = 0; i < times; i++) {
    promiseFactories.push({factory: fnPromise, context: {}});
  }

  return serialPromise(promiseFactories);
};

var getNativeMethodName = function(nativeWithSymbol) {
  return nativeWithSymbol.replace('__zone_symbol__', 'native_');
};

function testAddRemove(api, count) {
  var timerId = [];

  var name = api.method;
  mark(name);
  for (var i = 0; i < count; i++) {
    timerId.push(api.run());
  }
  measure(name, name);

  if (api.supportClear) {
    var clearName = api.clearMethod;
    mark(clearName);
    for (var i = 0; i < count; i++) {
      api.runClear(timerId[i]);
    }
    measure(clearName, clearName);
  }

  timerId = [];

  var nativeName = getNativeMethodName(api.nativeMethod);
  mark(nativeName);
  for (var i = 0; i < count; i++) {
    timerId.push(api.nativeRun());
  }
  measure(nativeName, nativeName);

  if (api.supportClear) {
    var nativeClearName = getNativeMethodName(api.nativeClearMethod);
    mark(nativeClearName);
    for (var i = 0; i < count; i++) {
      api.nativeRunClear(timerId[i]);
    }
    measure(nativeClearName, nativeClearName);
  }

  return Promise.resolve(1);
}

function testCallback(api, count) {
  var promises = [Promise.resolve(1)];
  for (var i = 0; i < count; i++) {
    var r = api.run();
    if (api.isAsync) {
      promises.push(r);
    }
  }

  for (var i = 0; i < count; i++) {
    var r = api.nativeRun();
    if (api.isAsync) {
      promises.push(r);
    }
  }
  return Promise.all(promises);
}

function measureCallback(api, ops) {
  var times = ops.times;
  var displayText = ops.displayText;
  var rawData = ops.rawData;
  var summary = ops.summary;

  var name = api.method;
  var nativeName = getNativeMethodName(api.nativeMethod);
  var measure = averageMeasures(name, times);
  var nativeMeasure = averageMeasures(nativeName, times);
  displayText += `- ${name} costs ${measure} ms\n`;
  displayText += `- ${nativeName} costs ${nativeMeasure} ms\n`;
  var absolute = Math.floor(1000 * (measure - nativeMeasure)) / 1000;
  displayText += `# ${name} is ${absolute}ms slower than ${nativeName}\n`;
  rawData[name + '_measure'] = measure;
  rawData[nativeName + '_measure'] = nativeMeasure;
  summary[name] = absolute + 'ms';
}

function measureAddRemove(api, ops) {
  var times = ops.times;
  var displayText = ops.displayText;
  var rawData = ops.rawData;
  var summary = ops.summary;

  var name = api.method;
  var nativeName = getNativeMethodName(api.nativeMethod);

  var measure = averageMeasures(name, times);
  var nativeMeasure = averageMeasures(nativeName, times);
  displayText += `- ${name} costs ${measure} ms\n`;
  displayText += `- ${nativeName} costs ${nativeMeasure} ms\n`;
  var percent = Math.floor(100 * (measure - nativeMeasure) / nativeMeasure);
  displayText += `# ${name} is ${percent}% slower than ${nativeName}\n`;
  rawData[name + '_measure'] = measure;
  rawData[nativeName + '_measure'] = nativeMeasure;
  summary[name] = percent + '%';
  if (api.supportClear) {
    var clearName = api.clearMethod;
    var nativeClearName = getNativeMethodName(api.nativeClearMethod);
    var clearMeasure = averageMeasures(clearName, times);
    var nativeClearMeasure = averageMeasures(nativeClearName, times);
    var clearPercent = Math.floor(100 * (clearMeasure - nativeClearMeasure) / nativeClearMeasure);
    displayText += `- ${clearName} costs ${clearMeasure} ms\n`;
    displayText += `- ${nativeClearName} costs ${nativeClearMeasure} ms\n`;
    displayText += `# ${clearName} is ${clearPercent}% slower than ${nativeClearName}\n`;
    rawData[clearName + '_measure'] = clearMeasure;
    rawData[nativeClearName + '_measure'] = nativeClearMeasure;
    summary[clearName] = clearPercent + '%';
  }
}

var testRunner = _global['__zone_symbol__testRunner'] = function(testTarget) {
  var title = testTarget.title;
  var apis = testTarget.apis;
  var methods = apis.reduce(function(acc, api) {
    return acc.concat([api.method, api.nativeMethod]
                          .concat(api.supportClear ? [api.clearMethod, api.nativeClearMethod] : [])
                          .concat[api.method + '_callback', api.nativeMethod + '_callback']);
  }, []);
  var times = testTarget.times;

  allTasks.push({
    title: title,
    cleanFn: function() {
      methods.forEach(function(m) {
        clearMarks(m);
        clearMeasures(m);
      });
    },
    before: function() {
      testTarget.before && testTarget.before();
    },
    after: function() {
      testTarget.after && testTarget.after();
    },
    testFn: function() {
      var count = typeof testTarget.count === 'number' ? testTarget.count : 10000;
      var times = typeof testTarget.times === 'number' ? testTarget.times : 5;

      var testFunction = function() {
        var promises = [];
        apis.forEach(function(api) {
          if (api.isCallback) {
            var r = testCallback(api, count / 100);
            promises.push(api.isAsync ? r : Promise.resolve(1));
          } else {
            var r = testAddRemove(api, count);
            promises.push[api.isAsync ? r : Promise.resolve(1)];
          }
        });
        return Promise.all(promises);
      };

      return runAsync(testFunction, times).then(function() {
        var displayText = `running ${count} times\n`;
        var rawData = {};
        var summary = {};
        apis.forEach(function(api) {
          if (api.isCallback) {
            measureCallback(api, {times, displayText, rawData, summary});
          } else {
            measureAddRemove(api, {times, displayText, rawData, summary});
          }
        });
        return Promise.resolve({displayText: displayText, rawData: rawData, summary: summary});
      });
    }
  });
};
}(typeof window === 'undefined' ? global : window));