Show connected edges instantly

Author: ugouhar

openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/CustomNode.utils.tsx

@@ -236,6 +236,7 @@ export const getColumnContent = (
       data-testid={`column-${fullyQualifiedName}`}
       key={fullyQualifiedName}
       onClick={handleClick}
+      onMouseDown={handleClick}
       onMouseEnter={handleMouseEnter}
       onMouseLeave={handleMouseLeave}>
       {getColumnHandle(

openmetadata-ui/src/main/resources/ui/src/context/LineageProvider/LineageProvider.tsx

@@ -1271,6 +1271,18 @@ const LineageProvider = ({ children }: LineageProviderProps) => {
           .finally(() => {
             setStatus('initial');
             setLoading(false);
+            if (!tracedColumns.includes(params.sourceHandle as string)) {
+              setTracedColumns((prev) => [
+                ...prev,
+                params.sourceHandle as string,
+              ]);
+            }
+            if (!tracedColumns.includes(params.targetHandle as string)) {
+              setTracedColumns((prev) => [
+                ...prev,
+                params.targetHandle as string,
+              ]);
+            }
           });
       }
     },