24ec43eb92
Summary: While looking at the event stream it is useful to know which observer type fired it Reviewed By: lblasa Differential Revision: D39430756 fbshipit-source-id: bc52f085a5497692f8076b12a9c015cc33a19d1e
61 lines
1.6 KiB
TypeScript
61 lines
1.6 KiB
TypeScript
/**
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*
|
|
* @format
|
|
*/
|
|
|
|
import {PluginClient, createState, createDataSource} from 'flipper-plugin';
|
|
import {Id, UINode} from './types';
|
|
|
|
export type PerfStatsEvent = {
|
|
txId: number;
|
|
observerType: string;
|
|
start: number;
|
|
traversalComplete: number;
|
|
serializationComplete: number;
|
|
queuingComplete: number;
|
|
socketComplete: number;
|
|
nodesCount: number;
|
|
};
|
|
|
|
type Events = {
|
|
init: {rootId: string};
|
|
nativeScan: {txId: number; nodes: UINode[]};
|
|
subtreeUpdate: {txId: number; nodes: UINode[]};
|
|
perfStats: PerfStatsEvent;
|
|
};
|
|
|
|
export function plugin(client: PluginClient<Events>) {
|
|
const rootId = createState<Id | undefined>(undefined);
|
|
client.onMessage('init', (root) => rootId.set(root.rootId));
|
|
|
|
const perfEvents = createDataSource<PerfStatsEvent, 'txId'>([], {
|
|
key: 'txId',
|
|
limit: 10 * 1024,
|
|
});
|
|
client.onMessage('perfStats', (event) => {
|
|
perfEvents.append(event);
|
|
});
|
|
|
|
const nodesAtom = createState<Map<Id, UINode>>(new Map());
|
|
client.onMessage('subtreeUpdate', ({nodes}) => {
|
|
nodesAtom.update((draft) => {
|
|
for (const node of nodes) {
|
|
draft.set(node.id, node);
|
|
}
|
|
});
|
|
});
|
|
|
|
client.onMessage('nativeScan', ({nodes}) => {
|
|
//Native scan is a full update so overwrite everything
|
|
nodesAtom.set(new Map(nodes.map((node) => [node.id, node])));
|
|
});
|
|
|
|
return {rootId, nodes: nodesAtom, perfEvents};
|
|
}
|
|
|
|
export {Component} from './components/main';
|