fix: business attribute empty bubble selection issue (#11720)

datahub-project · Oct 28, 2024 · a8af347 · a8af347
1 parent 0e4b3c1
commit a8af347
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/datahub-web-react/src/app/shared/businessAttribute/AddBusinessAttributeModal.tsx b/datahub-web-react/src/app/shared/businessAttribute/AddBusinessAttributeModal.tsx
@@ -155,7 +155,9 @@ export default function EditBusinessAttributeModal({
             event.stopPropagation();
         };
         /* eslint-disable-next-line react/prop-types */
-        const selectedItem = displayedAttributes.find((attribute) => attribute.urn === value)?.component;
+        const selectedItem = selectedAttribute
+            ? selectedAttribute?.props?.component
+            : displayedAttributes.find((attribute) => attribute.urn === value)?.component;
         return (
             <StyleTag onMouseDown={onPreventMouseDown} closable={closable} onClose={onClose}>
                 {selectedItem}
@@ -172,7 +174,7 @@ export default function EditBusinessAttributeModal({
     const onSelectValue = (selectedUrn: string) => {
         const selectedSearchOption = attributeSearchOptions?.find((option) => option.props.value === selectedUrn);
         setUrn(selectedUrn);
-        if (!selectedAttribute) {
+        if (selectedAttribute?.selectedUrn !== selectedUrn) {
             setSelectedAttribute({
                 selectedUrn,
                 component: (