Correct remove method of QualifiedNameMap

ballerina-platform · May 27, 2024 · 98bf3a4 · 98bf3a4
1 parent 326535a
commit 98bf3a4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/native/src/main/java/io/ballerina/lib/data/xmldata/xml/QualifiedNameMap.java b/native/src/main/java/io/ballerina/lib/data/xmldata/xml/QualifiedNameMap.java
@@ -29,12 +29,19 @@ public V remove(QualifiedName qName) {
             if (!fields.containsKey(localName)) {
                 return null;
             }
+
+            List<QualifiedName> qNames = fields.get(localName);
             for (QualifiedName qualifiedName : fields.get(localName)) {
                 if (qualifiedName.getNamespaceURI().equals(Constants.NS_ANNOT_NOT_DEFINED)) {
                     field = this.members.remove(qualifiedName);
+                    qNames.remove(qualifiedName);
                     break;
                 }
             }
+
+            if (qNames.isEmpty()) {
+                fields.remove(localName);
+            }
         }
         return field;
     }