Fixed review comments

1. Used hash function in place of JSON.stringify(row)
2. Storing row identifiers in place of complete row object.

Author: anilsahoo20

web/pgadmin/tools/sqleditor/static/js/components/sections/GeometryViewer.jsx

@@ -198,26 +198,35 @@ function findPkColumn(columns) {
   return columns.find(c => PK_COLUMN_NAMES.includes(c.name));
 }
 
+// Hash function for row objects
+function hashRow(row) {
+  const str = Object.keys(row).sort().map(k => `${k}:${row[k]}`).join('|');
+  let hash = 0;
+  for (let i = 0; i < str.length; i++) {
+    const char = str.charCodeAt(i);
+    hash = ((hash << 5) - hash) + char;
+    hash = hash & hash;
+  }
+  return `hash_${hash}`;
+}
+
 // Get unique row identifier using PK column or first column
 function getRowIdentifier(row, pkColumn, columns) {
   if (pkColumn?.key && row[pkColumn.key] !== undefined) {
     return row[pkColumn.key];
   }
   const firstKey = columns[0]?.key;
-  return firstKey && row[firstKey] !== undefined ? row[firstKey] : JSON.stringify(row);
-}
-
-// Create Set of row identifiers
-function createIdentifierSet(rowData, pkColumn, columns) {
-  return new Set(rowData.map(row => getRowIdentifier(row, pkColumn, columns)));
+  if (firstKey && row[firstKey] !== undefined) {
+    return row[firstKey];
+  }
+  return hashRow(row);
 }
 
 // Match rows from previous selection to current rows
-function matchRowSelection(prevRowData, currentRows, pkColumn, columns) {
-  if (prevRowData.length === 0) return [];
+function matchRowSelection(prevIdentifiers, currentRows, pkColumn, columns) {
+  if (prevIdentifiers.size === 0) return [];
 
-  const prevIdSet = createIdentifierSet(prevRowData, pkColumn, columns);
-  return currentRows.filter(row => prevIdSet.has(getRowIdentifier(row, pkColumn, columns)));
+  return currentRows.filter(row => prevIdentifiers.has(getRowIdentifier(row, pkColumn, columns)));
 }
 
 function PopupTable({data}) {
@@ -492,7 +501,7 @@ export function GeometryViewer({rows, columns, column}) {
   const prevStateRef = React.useRef({
     columnKey: null,
     columnNames: null,
-    selectedRowData: [],
+    selectedRowIdentifiers: new Set(),
   });
 
   const [mapKey, setMapKey] = React.useState(0);
@@ -513,7 +522,7 @@ export function GeometryViewer({rows, columns, column}) {
       prevStateRef.current = {
         columnKey: null,
         columnNames: null,
-        selectedRowData: [],
+        selectedRowIdentifiers: new Set(),
       };
       return;
     }
@@ -524,32 +533,46 @@ export function GeometryViewer({rows, columns, column}) {
       prevStateRef.current = {
         columnKey: currentColumnKey,
         columnNames: currentColumnNames,
-        selectedRowData: rows,
+        selectedRowIdentifiers: new Set(rows.map(r => getRowIdentifier(r, pkColumn, columns))),
       };
       return;
     }
 
     if (currentColumnKey === prevState.columnKey && 
         currentColumnNames === prevState.columnNames &&
         rows.length > 0) {
-      prevStateRef.current.selectedRowData = displayRows;
+      prevStateRef.current.selectedRowIdentifiers = new Set(
+        displayRows.map(r => getRowIdentifier(r, pkColumn, columns))
+      );
     }
   }, [currentColumnKey, currentColumnNames, rows, pkColumn, columns]);
 
   // Get rows to display based on selection
   const displayRows = React.useMemo(() => {
+    // No geometry column selected or no rows available - nothing to display
     if (!currentColumnKey || rows.length === 0) return [];
     const prevState = prevStateRef.current;
+
+    // Column context changed (different geometry column or different query schema)
+    // Show all new rows since previous selection is no longer valid
     if (currentColumnKey !== prevState.columnKey || currentColumnNames !== prevState.columnNames) {
       return rows;
     }
 
-    const prevSelected = prevState.selectedRowData;
-    if (prevSelected.length === 0) return rows;
-    if (prevSelected.length < rows.length) {
-      const matched = matchRowSelection(prevSelected, rows, pkColumn, columns);
+    const prevIdentifiers = prevState.selectedRowIdentifiers;
+    // No previous selection recorded - show all rows
+    if (prevIdentifiers.size === 0) return rows;
+
+    // Previous selection was a subset of total rows, meaning user had specific rows selected.
+    // Try to match those previously selected rows in the new result set using stable
+    // row identifiers (PK value, first column value, or hash fallback).
+    // This handles the case where same query reruns with more/fewer rows
+    if (prevIdentifiers.size < rows.length) {
+      const matched = matchRowSelection(prevIdentifiers, rows, pkColumn, columns);
+      // If matched rows found, show only those; otherwise fall back to all rows
       return matched.length > 0 ? matched : rows;
     }
+    // Previous selection covered all rows (or same count) - show all current rows
     return rows;
   }, [rows, currentColumnKey, currentColumnNames, pkColumn, columns]);