From 3648ce8492e23c0cb0583d866771909ec2d849f9 Mon Sep 17 00:00:00 2001
From: csirius <85753828+csirius@users.noreply.github.com>
Date: Fri, 15 Oct 2021 19:27:08 -0400
Subject: [PATCH] fix: add pagination for child node expansion

Signed-off-by: csirius <85753828+csirius@users.noreply.github.com>
---
 .../Tables/NodeExecutionChildren.tsx          | 63 ++++++++++++++++---
 1 file changed, 53 insertions(+), 10 deletions(-)

diff --git a/src/components/Executions/Tables/NodeExecutionChildren.tsx b/src/components/Executions/Tables/NodeExecutionChildren.tsx
index eeb9a0164..256a2e29e 100644
--- a/src/components/Executions/Tables/NodeExecutionChildren.tsx
+++ b/src/components/Executions/Tables/NodeExecutionChildren.tsx
@@ -1,4 +1,5 @@
-import { Typography } from '@material-ui/core';
+import { Button, Typography } from '@material-ui/core';
+import { Theme, makeStyles } from '@material-ui/core/styles';
 import * as classnames from 'classnames';
 import { getCacheKey } from 'components/Cache/utils';
 import { useTheme } from 'components/Theme/useTheme';
@@ -15,12 +16,24 @@ export interface NodeExecutionChildrenProps {
     level: number;
 }
 
+const PAGE_SIZE = 50;
+
+const useStyles = makeStyles((theme: Theme) => ({
+    loadMoreContainer: {
+        display: 'flex',
+        justifyContent: 'center',
+        marginTop: theme.spacing(1),
+        marginBottom: theme.spacing(1)
+    }
+}));
+
 /** Renders a nested list of row items for children of a NodeExecution */
 export const NodeExecutionChildren: React.FC<NodeExecutionChildrenProps> = ({
     abortMetadata,
     childGroups,
     level
 }) => {
+    const styles = useStyles();
     const showNames = childGroups.length > 1;
     const tableStyles = useExecutionTableStyles();
     const theme = useTheme();
@@ -31,18 +44,45 @@ export const NodeExecutionChildren: React.FC<NodeExecutionChildrenProps> = ({
             theme.spacing
         )}px`
     };
+    const [loadedNodes, setLoadedNodes] = React.useState(
+        new Array(childGroups.length).fill(PAGE_SIZE)
+    );
+
+    const loadMoreRows = React.useCallback(
+        (which: number) => () => {
+            const newLoadedNodes = [...loadedNodes];
+            newLoadedNodes[which] += PAGE_SIZE;
+            setLoadedNodes(newLoadedNodes);
+        },
+        [loadedNodes]
+    );
+
+    const loadMoreButton = (which: number) => (
+        <div className={styles.loadMoreContainer}>
+            <Button
+                onClick={loadMoreRows(which)}
+                size="small"
+                variant="outlined"
+            >
+                Load More
+            </Button>
+        </div>
+    );
+
     return (
         <>
             {childGroups.map(({ name, nodeExecutions }, groupIndex) => {
-                const rows = nodeExecutions.map((nodeExecution, index) => (
-                    <NodeExecutionRow
-                        abortMetadata={abortMetadata}
-                        key={getCacheKey(nodeExecution.id)}
-                        index={index}
-                        execution={nodeExecution}
-                        level={level}
-                    />
-                ));
+                const rows = nodeExecutions
+                    .slice(0, loadedNodes[groupIndex])
+                    .map((nodeExecution, index) => (
+                        <NodeExecutionRow
+                            abortMetadata={abortMetadata}
+                            key={getCacheKey(nodeExecution.id)}
+                            index={index}
+                            execution={nodeExecution}
+                            level={level}
+                        />
+                    ));
                 const key = `group-${name}`;
                 return showNames ? (
                     <div key={key} role="list">
@@ -63,10 +103,13 @@ export const NodeExecutionChildren: React.FC<NodeExecutionChildrenProps> = ({
                             </Typography>
                         </div>
                         <div>{rows}</div>
+                        {loadMoreButton(groupIndex)}
                     </div>
                 ) : (
                     <div key={key} role="list">
                         {rows}
+                        {nodeExecutions.length > loadedNodes[groupIndex] &&
+                            loadMoreButton(groupIndex)}
                     </div>
                 );
             })}