Memoize component

Author: KoolADE85

src/lib/fragments/AgGrid.react.js

@@ -272,7 +272,6 @@ export function DashAgGrid(props) {
     const [openGroups, setOpenGroups] = useState({});
     const [columnState_push, setColumnState_push] = useState(true);
     const [rowTransactionState, setRowTransactionState] = useState(null);
-    const [parentState] = useState(props.parentState || {});
 
     const components = useMemo(
         () => ({
@@ -1706,4 +1705,22 @@ export const defaultProps = DashAgGrid.defaultProps;
 var dagfuncs = (window.dash_ag_grid = window.dash_ag_grid || {});
 dagfuncs.useGridFilter = useGridFilter;
 
-export default DashAgGrid;
+const MemoizedAgGrid = React.memo(DashAgGrid, (prevProps, nextProps) => {
+  // Check if props are equal (excluding render-specific props)
+  const relevantNextProps = { ...omit(OMIT_PROP_RENDER, nextProps) };
+  const relevantPrevProps = { ...omit(OMIT_PROP_RENDER, prevProps) };
+
+  const isInternalChange = nextProps?.dashRenderType === 'internal';
+  const propsHaveChanged = !equals(relevantNextProps, relevantPrevProps);
+  const rowDataChanged = !equals(nextProps.rowData, prevProps.rowData);
+  const selectedRowsChanged = !equals(nextProps.selectedRows, prevProps.selectedRows);
+
+  if (propsHaveChanged && (!isInternalChange || rowDataChanged || selectedRowsChanged)) {
+    return false; // Props changed, re-render
+  }
+
+  return true;
+
+});
+
+export default MemoizedAgGrid;

src/lib/utils/propCategories.js

@@ -317,10 +317,12 @@ export const PASSTHRU_PROPS = ['rowData'];
  * in the render() method, so they don't need to be listed here
  */
 export const PROPS_NOT_FOR_AG_GRID = [
+    'children',
     'setProps',
     'loading_state',
     'enableEnterpriseModules',
     'parentState',
+    'persistence',
     'persisted_props',
     'persistence_type',
     'virtualRowData',
@@ -335,6 +337,7 @@ export const PROPS_NOT_FOR_AG_GRID = [
     'alignedGrids',
     'resetColumnState',
     'exportDataAsCsv',
+    'selectedRows',
     'selectAll',
     'deselectAll',
     'deleteSelectedRows',
@@ -357,7 +360,6 @@ export const OMIT_PROP_RENDER = [
     'virtualRowData',
     'columnState',
     'filterModel',
-    'selectedRows',
     'getRowRequest',
     'getDetailRequest',
     'cellValueChanged',