immutable data structures in tables 5/n: create table plugin

Summary:
Migrating tables' row collection to Immutable.js List:
1. Migrate createTablePlugin to ManagedTable_immutable

-----
Current implementation of tables forces to copy arrays on new data arrival which causes O(N^2) complexity ->
tables can't handle a lot of new rows in short period of time -> tables freeze and become unresponsive for a few seconds.
Immutable data structures will bring us to O(N) complexity

Reviewed By: jknoxville

Differential Revision: D16416867

fbshipit-source-id: 20890aa851cd2e34e33fd2ed69c5d6048af14cbb

src/__tests__/createTablePlugin.node.js

@@ -6,15 +6,30 @@
  */
 
 import {createTablePlugin} from '../createTablePlugin.js';
+
+import type {TableRows_immutable} from 'flipper';
+
+//import type {PersistedState, RowData} from '../createTablePlugin.js';
 import {FlipperPlugin} from '../plugin.js';
 
+import {List, Map as ImmutableMap} from 'immutable';
+
 const PROPS = {
   method: 'method',
   resetMethod: 'resetMethod',
   columns: {},
   columnSizes: {},
-  renderSidebar: () => {},
-  buildRow: () => {},
+  renderSidebar: (r: {id: string}) => {},
+  buildRow: (r: {id: string}) => {},
+};
+
+type PersistedState<T> = {|
+  rows: TableRows_immutable,
+  datas: ImmutableMap<string, T>,
+|};
+
+type RowData = {
+  id: string,
 };
 
 test('createTablePlugin returns FlipperPlugin', () => {
@@ -24,29 +39,27 @@ test('createTablePlugin returns FlipperPlugin', () => {
 
 test('persistedStateReducer is resetting data', () => {
   const resetMethod = 'resetMethod';
-  const tablePlugin = createTablePlugin({...PROPS, resetMethod});
+  const tablePlugin = createTablePlugin<RowData>({...PROPS, resetMethod});
 
-  // $FlowFixMe persistedStateReducer exists for createTablePlugin
-  const {rows, datas} = tablePlugin.persistedStateReducer(
-    {
-      datas: {'1': {id: '1'}},
-      rows: [
-        {
-          key: '1',
-          columns: {
-            id: {
-              value: '1',
-            },
+  const ps: PersistedState<RowData> = {
+    datas: ImmutableMap({'1': {id: '1'}}),
+    rows: List([
+      {
+        key: '1',
+        columns: {
+          id: {
+            value: '1',
           },
         },
-      ],
-    },
-    resetMethod,
-    {},
-  );
+      },
+    ]),
+  };
 
-  expect(datas).toEqual({});
-  expect(rows).toEqual([]);
+  // $FlowFixMe persistedStateReducer exists for createTablePlugin
+  const {rows, datas} = tablePlugin.persistedStateReducer(ps, resetMethod, {});
+
+  expect(datas.toJSON()).toEqual({});
+  expect(rows.size).toBe(0);
 });
 
 test('persistedStateReducer is adding data', () => {
@@ -56,14 +69,11 @@ test('persistedStateReducer is adding data', () => {
 
   // $FlowFixMe persistedStateReducer exists for createTablePlugin
   const {rows, datas} = tablePlugin.persistedStateReducer(
-    {
-      datas: {},
-      rows: [],
-    },
+    tablePlugin.defaultPersistedState,
     method,
     {id},
   );
 
-  expect(rows.length).toBe(1);
-  expect(Object.keys(datas)).toEqual([id]);
+  expect(rows.size).toBe(1);
+  expect([...datas.keys()]).toEqual([id]);
 });

src/createTablePlugin.js

@@ -7,7 +7,7 @@
 
 import type {
   TableHighlightedRows,
-  TableRows,
+  TableRows_immutable,
   TableColumnSizes,
   TableColumns,
 } from 'flipper';
@@ -16,13 +16,15 @@ import FlexColumn from './ui/components/FlexColumn';
 import Button from './ui/components/Button';
 import DetailSidebar from './chrome/DetailSidebar';
 import {FlipperPlugin} from './plugin';
-import SearchableTable from './ui/components/searchable/SearchableTable';
+import SearchableTable_immutable from './ui/components/searchable/SearchableTable_immutable';
 import textContent from './utils/textContent.js';
 import createPaste from './fb-stubs/createPaste.js';
 
+import {List, Map as ImmutableMap} from 'immutable';
+
 type ID = string;
 
-type RowData = {
+export type RowData = {
   id: ID,
 };
 
@@ -35,9 +37,9 @@ type Props<T> = {|
   buildRow: (row: T) => any,
 |};
 
-type PersistedState<T> = {|
-  rows: TableRows,
-  datas: {[key: ID]: T},
+export type PersistedState<T> = {|
+  rows: TableRows_immutable,
+  datas: ImmutableMap<ID, T>,
 |};
 
 type State = {|
@@ -58,41 +60,36 @@ type State = {|
  * the client in an unknown state.
  */
 export function createTablePlugin<T: RowData>(props: Props<T>) {
-  return class extends FlipperPlugin<State, *, PersistedState<T>> {
+  return class extends FlipperPlugin<State, *, PersistedState<*>> {
     static keyboardActions = ['clear', 'createPaste'];
 
-    static defaultPersistedState: PersistedState<T> = {
-      rows: [],
-      datas: {},
+    static defaultPersistedState = {
+      rows: List(),
+      datas: ImmutableMap<ID, T>(),
     };
 
     static persistedStateReducer = (
-      persistedState: PersistedState<T>,
+      persistedState: PersistedState<*>,
       method: string,
       payload: T | Array<T>,
-    ): $Shape<PersistedState<RowData>> => {
+    ): $Shape<PersistedState<T>> => {
       if (method === props.method) {
-        const newRows = [];
-        const newData = {};
-        const datas = Array.isArray(payload) ? payload : [payload];
-
-        for (const data of datas.reverse()) {
-          if (data.id == null) {
-            console.warn('The data sent did not contain an ID.', data);
-          }
-          if (persistedState.datas[data.id] == null) {
-            newData[data.id] = data;
-            newRows.push(props.buildRow(data));
-          }
-        }
-        return {
-          datas: {...persistedState.datas, ...newData},
-          rows: [...persistedState.rows, ...newRows],
-        };
+        return List(Array.isArray(payload) ? payload : [payload]).reduce(
+          (ps: PersistedState<*>, data: T) => {
+            if (data.id == null) {
+              console.warn('The data sent did not contain an ID.', data);
+            }
+            return {
+              datas: ps.datas.set(data.id, data),
+              rows: ps.rows.push(props.buildRow(data)),
+            };
+          },
+          persistedState,
+        );
       } else if (method === props.resetMethod) {
         return {
-          rows: [],
-          datas: {},
+          rows: List(),
+          datas: ImmutableMap(),
         };
       } else {
         return {};
@@ -113,8 +110,8 @@ export function createTablePlugin<T: RowData>(props: Props<T>) {
 
     clear = () => {
       this.props.setPersistedState({
-        rows: [],
-        datas: {},
+        rows: List(),
+        datas: ImmutableMap(),
       });
       this.setState({
         selectedIds: [],
@@ -154,7 +151,8 @@ export function createTablePlugin<T: RowData>(props: Props<T>) {
       const selectedId = selectedIds.length !== 1 ? null : selectedIds[0];
 
       if (selectedId != null) {
-        return renderSidebar(datas[selectedId]);
+        const data = datas.get(selectedId);
+        return data != null ? renderSidebar(data) : null;
       } else {
         return null;
       }
@@ -166,7 +164,7 @@ export function createTablePlugin<T: RowData>(props: Props<T>) {
 
       return (
         <FlexColumn grow={true}>
-          <SearchableTable
+          <SearchableTable_immutable
             key={this.constructor.id}
             rowLineHeight={28}
             floating={false}