143 lines
3.5 KiB
JavaScript
143 lines
3.5 KiB
JavaScript
/**
|
|
* Copyright (c) 2015-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*
|
|
* @providesModule BridgeProfiling
|
|
* @flow
|
|
*/
|
|
'use strict';
|
|
|
|
type RelayProfiler = {
|
|
attachProfileHandler(
|
|
name: string,
|
|
handler: (name: string, state?: any) => () => void
|
|
): void
|
|
};
|
|
|
|
var GLOBAL = GLOBAL || this;
|
|
var TRACE_TAG_REACT_APPS = 1 << 17;
|
|
|
|
var _enabled;
|
|
var _ReactPerf = null;
|
|
function ReactPerf() {
|
|
if (!_ReactPerf) {
|
|
_ReactPerf = require('ReactPerf');
|
|
}
|
|
return _ReactPerf;
|
|
}
|
|
|
|
var BridgeProfiling = {
|
|
setEnabled(enabled: boolean) {
|
|
_enabled = enabled;
|
|
|
|
ReactPerf().enableMeasure = enabled;
|
|
},
|
|
|
|
profile(profileName?: any) {
|
|
if (_enabled) {
|
|
profileName = typeof profileName === 'function' ?
|
|
profileName() : profileName;
|
|
global.nativeTraceBeginSection(TRACE_TAG_REACT_APPS, profileName);
|
|
}
|
|
},
|
|
|
|
profileEnd() {
|
|
if (_enabled) {
|
|
global.nativeTraceEndSection(TRACE_TAG_REACT_APPS);
|
|
}
|
|
},
|
|
|
|
reactPerfMeasure(objName: string, fnName: string, func: any): any {
|
|
return function (component) {
|
|
if (!_enabled) {
|
|
return func.apply(this, arguments);
|
|
}
|
|
|
|
var name = objName === 'ReactCompositeComponent' && this.getName() || '';
|
|
BridgeProfiling.profile(`${objName}.${fnName}(${name})`);
|
|
var ret = func.apply(this, arguments);
|
|
BridgeProfiling.profileEnd();
|
|
return ret;
|
|
};
|
|
},
|
|
|
|
swizzleReactPerf() {
|
|
ReactPerf().injection.injectMeasure(BridgeProfiling.reactPerfMeasure);
|
|
},
|
|
|
|
attachToRelayProfiler(relayProfiler: RelayProfiler) {
|
|
relayProfiler.attachProfileHandler('*', (name) => {
|
|
BridgeProfiling.profile(name);
|
|
return () => {
|
|
BridgeProfiling.profileEnd();
|
|
};
|
|
});
|
|
},
|
|
|
|
/* This is not called by default due to perf overhead but it's useful
|
|
if you want to find traces which spend too much time in JSON. */
|
|
swizzleJSON() {
|
|
BridgeProfiling.measureMethods(JSON, 'JSON', [
|
|
'parse',
|
|
'stringify'
|
|
]);
|
|
},
|
|
|
|
/**
|
|
* Measures multiple methods of a class. For example, you can do:
|
|
* BridgeProfiling.measureMethods(JSON, 'JSON', ['parse', 'stringify']);
|
|
*
|
|
* @param object
|
|
* @param objectName
|
|
* @param methodNames Map from method names to method display names.
|
|
*/
|
|
measureMethods(object: any, objectName: string, methodNames: Array<string>): void {
|
|
if (!__DEV__) {
|
|
return;
|
|
}
|
|
|
|
methodNames.forEach(methodName => {
|
|
object[methodName] = BridgeProfiling.measure(
|
|
objectName,
|
|
methodName,
|
|
object[methodName]
|
|
);
|
|
});
|
|
},
|
|
|
|
/**
|
|
* Returns an profiled version of the input function. For example, you can:
|
|
* JSON.parse = BridgeProfiling.measure('JSON', 'parse', JSON.parse);
|
|
*
|
|
* @param objName
|
|
* @param fnName
|
|
* @param {function} func
|
|
* @return {function} replacement function
|
|
*/
|
|
measure(objName: string, fnName: string, func: any): any {
|
|
if (!__DEV__) {
|
|
return func;
|
|
}
|
|
|
|
var profileName = `${objName}.${fnName}`;
|
|
return function() {
|
|
if (!_enabled) {
|
|
return func.apply(this, arguments);
|
|
}
|
|
|
|
BridgeProfiling.profile(profileName);
|
|
var ret = func.apply(this, arguments);
|
|
BridgeProfiling.profileEnd();
|
|
return ret;
|
|
};
|
|
},
|
|
};
|
|
|
|
BridgeProfiling.setEnabled(global.__RCTProfileIsProfiling || false);
|
|
|
|
module.exports = BridgeProfiling;
|