wtf.ts 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159
  1. /**
  2. * @license
  3. * Copyright Google Inc. All Rights Reserved.
  4. *
  5. * Use of this source code is governed by an MIT-style license that can be
  6. * found in the LICENSE file at https://angular.io/license
  7. */
  8. /**
  9. * @fileoverview
  10. * @suppress {missingRequire}
  11. */
  12. (function(global: any) {
  13. interface Wtf {
  14. trace: WtfTrace;
  15. }
  16. interface WtfScope {}
  17. interface WtfRange {}
  18. interface WtfTrace {
  19. events: WtfEvents;
  20. leaveScope(scope: WtfScope, returnValue?: any): void;
  21. beginTimeRange(rangeType: string, action: string): WtfRange;
  22. endTimeRange(range: WtfRange): void;
  23. }
  24. interface WtfEvents {
  25. createScope(signature: string, flags?: any): WtfScopeFn;
  26. createInstance(signature: string, flags?: any): WtfEventFn;
  27. }
  28. type WtfScopeFn = (...args: any[]) => WtfScope;
  29. type WtfEventFn = (...args: any[]) => any;
  30. // Detect and setup WTF.
  31. let wtfTrace: WtfTrace|null = null;
  32. let wtfEvents: WtfEvents|null = null;
  33. const wtfEnabled: boolean = (function(): boolean {
  34. const wtf: Wtf = global['wtf'];
  35. if (wtf) {
  36. wtfTrace = wtf.trace;
  37. if (wtfTrace) {
  38. wtfEvents = wtfTrace.events;
  39. return true;
  40. }
  41. }
  42. return false;
  43. })();
  44. class WtfZoneSpec implements ZoneSpec {
  45. name: string = 'WTF';
  46. static forkInstance =
  47. wtfEnabled ? wtfEvents!.createInstance('Zone:fork(ascii zone, ascii newZone)') : null;
  48. static scheduleInstance: {[key: string]: WtfEventFn} = {};
  49. static cancelInstance: {[key: string]: WtfEventFn} = {};
  50. static invokeScope: {[key: string]: WtfEventFn} = {};
  51. static invokeTaskScope: {[key: string]: WtfEventFn} = {};
  52. onFork(parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, zoneSpec: ZoneSpec):
  53. Zone {
  54. const retValue = parentZoneDelegate.fork(targetZone, zoneSpec);
  55. WtfZoneSpec.forkInstance!(zonePathName(targetZone), retValue.name);
  56. return retValue;
  57. }
  58. onInvoke(
  59. parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, delegate: Function,
  60. applyThis: any, applyArgs?: any[], source?: string): any {
  61. const src = source || 'unknown';
  62. let scope = WtfZoneSpec.invokeScope[src];
  63. if (!scope) {
  64. scope = WtfZoneSpec.invokeScope[src] =
  65. wtfEvents!.createScope(`Zone:invoke:${source}(ascii zone)`);
  66. }
  67. return wtfTrace!.leaveScope(
  68. scope(zonePathName(targetZone)),
  69. parentZoneDelegate.invoke(targetZone, delegate, applyThis, applyArgs, source));
  70. }
  71. onHandleError(parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, error: any):
  72. boolean {
  73. return parentZoneDelegate.handleError(targetZone, error);
  74. }
  75. onScheduleTask(parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, task: Task):
  76. any {
  77. const key = task.type + ':' + task.source;
  78. let instance = WtfZoneSpec.scheduleInstance[key];
  79. if (!instance) {
  80. instance = WtfZoneSpec.scheduleInstance[key] =
  81. wtfEvents!.createInstance(`Zone:schedule:${key}(ascii zone, any data)`);
  82. }
  83. const retValue = parentZoneDelegate.scheduleTask(targetZone, task);
  84. instance(zonePathName(targetZone), shallowObj(task.data, 2));
  85. return retValue;
  86. }
  87. onInvokeTask(
  88. parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, task: Task,
  89. applyThis?: any, applyArgs?: any[]): any {
  90. const source = task.source;
  91. let scope = WtfZoneSpec.invokeTaskScope[source];
  92. if (!scope) {
  93. scope = WtfZoneSpec.invokeTaskScope[source] =
  94. wtfEvents!.createScope(`Zone:invokeTask:${source}(ascii zone)`);
  95. }
  96. return wtfTrace!.leaveScope(
  97. scope(zonePathName(targetZone)),
  98. parentZoneDelegate.invokeTask(targetZone, task, applyThis, applyArgs));
  99. }
  100. onCancelTask(parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, task: Task):
  101. any {
  102. const key = task.source;
  103. let instance = WtfZoneSpec.cancelInstance[key];
  104. if (!instance) {
  105. instance = WtfZoneSpec.cancelInstance[key] =
  106. wtfEvents!.createInstance(`Zone:cancel:${key}(ascii zone, any options)`);
  107. }
  108. const retValue = parentZoneDelegate.cancelTask(targetZone, task);
  109. instance(zonePathName(targetZone), shallowObj(task.data, 2));
  110. return retValue;
  111. }
  112. }
  113. function shallowObj(obj: {[k: string]: any}|undefined, depth: number): any {
  114. if (!obj || !depth) return null;
  115. const out: {[k: string]: any} = {};
  116. for (const key in obj) {
  117. if (obj.hasOwnProperty(key)) {
  118. let value = obj[key];
  119. switch (typeof value) {
  120. case 'object':
  121. const name = value && value.constructor && (<any>value.constructor).name;
  122. value = name == (<any>Object).name ? shallowObj(value, depth - 1) : name;
  123. break;
  124. case 'function':
  125. value = value.name || undefined;
  126. break;
  127. }
  128. out[key] = value;
  129. }
  130. }
  131. return out;
  132. }
  133. function zonePathName(zone: Zone) {
  134. let name: string = zone.name;
  135. let localZone = zone.parent;
  136. while (localZone != null) {
  137. name = localZone.name + '::' + name;
  138. localZone = localZone.parent;
  139. }
  140. return name;
  141. }
  142. (Zone as any)['wtfZoneSpec'] = !wtfEnabled ? null : new WtfZoneSpec();
  143. })(typeof window === 'object' && window || typeof self === 'object' && self || global);