chore: added caching layer for table widget data

app/client/src/widgets/TableWidgetV2/component/Table.tsx

@@ -349,7 +349,11 @@ export function Table(props: TableProps) {
         {isHeaderVisible && <TableHeader />}
         <div className={getTableWrapClassName} ref={tableWrapperRef}>
           <div {...getTableProps()} className="table column-freeze">
-            {shouldUseVirtual ? <VirtualTable /> : <StaticTable />}
+            {shouldUseVirtual ? (
+              <VirtualTable ref={scrollBarRef} />
+            ) : (
+              <StaticTable />
+            )}
           </div>
         </div>
       </TableWrapper>

app/client/src/widgets/TableWidgetV2/component/TableContext.test.tsx

@@ -99,6 +99,7 @@ const mockTableProviderProps = {
   showConnectDataOverlay: false,
   onConnectData: jest.fn(),
   isInfiniteScrollEnabled: false,
+  endOfData: false,
 };
 
 // Test components
@@ -206,6 +207,7 @@ describe("TableContext", () => {
         "showConnectDataOverlay",
         "onConnectData",
         "isInfiniteScrollEnabled",
+        "endOfData",
       ].sort();
 
       const result = Object.keys(

app/client/src/widgets/TableWidgetV2/component/VirtualTable/BaseVirtualList.tsx

@@ -147,7 +147,7 @@ const BaseVirtualList = React.memo(function BaseVirtualList({
         rowNeedsMeasurement={rowNeedsMeasurement}
       />
     );
-  }, [listRef, rowHeights, rowNeedsMeasurement]);
+  }, [listRef, rowHeights, rowNeedsMeasurement, hasMoreData]);
 
   return (
     <VariableSizeList

app/client/src/widgets/TableWidgetV2/component/VirtualTable/InifiniteScrollBody/index.tsx

@@ -1,10 +1,9 @@
-import React, { type Ref } from "react";
+import React, { useMemo, type Ref } from "react";
 import { type ReactElementType } from "react-window";
 import type SimpleBar from "simplebar-react";
 import { LoadingIndicator } from "../../LoadingIndicator";
 import { VariableInfiniteVirtualList } from "../../TableBodyCoreComponents/VirtualList";
 import { useAppsmithTable } from "../../TableContext";
-import { useInfiniteVirtualization } from "./useInfiniteVirtualization";
 
 interface InfiniteScrollBodyProps {
   innerElementType: ReactElementType;
@@ -13,6 +12,7 @@ interface InfiniteScrollBodyProps {
 const InfiniteScrollBodyComponent = React.forwardRef(
   (props: InfiniteScrollBodyProps, ref: Ref<SimpleBar>) => {
     const {
+      endOfData,
       height,
       isLoading,
       nextPageClick,
@@ -21,24 +21,24 @@ const InfiniteScrollBodyComponent = React.forwardRef(
       tableSizes,
       totalRecordsCount,
     } = useAppsmithTable();
-    const { cachedRows, hasMoreData, itemCount } = useInfiniteVirtualization({
-      rows,
-      totalRecordsCount,
-      loadMore: nextPageClick,
-      pageSize,
-    });
+
+    const itemCount = useMemo(() => {
+      return totalRecordsCount && totalRecordsCount > 0
+        ? totalRecordsCount
+        : rows.length;
+    }, [totalRecordsCount, rows]);
 
     return (
       <div className="simplebar-content-wrapper">
         <VariableInfiniteVirtualList
-          hasMoreData={hasMoreData}
+          hasMoreData={!endOfData}
           height={height}
           innerElementType={props.innerElementType}
           itemCount={itemCount}
           loadMore={nextPageClick}
           outerRef={ref}
           pageSize={pageSize}
-          rows={cachedRows}
+          rows={rows}
           tableSizes={tableSizes}
         />
         {isLoading && <LoadingIndicator />}