Fix the error of parsing some special XML

autodeployai · Aug 28, 2024 · 2706e8f · 2706e8f
1 parent 1e3fa6c
commit 2706e8f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/main/scala/org/pmml4s/xml/pull.scala b/src/main/scala/org/pmml4s/xml/pull.scala
@@ -16,7 +16,7 @@
 
 package org.pmml4s.xml
 
-import javax.xml.stream.events.{Characters, Comment, EndElement, EntityReference, ProcessingInstruction, StartElement}
+import javax.xml.stream.events.{Characters, Comment, EndElement, EntityReference, ProcessingInstruction, StartElement, EndDocument}
 import javax.xml.stream.{EventFilter, XMLInputFactory, XMLStreamConstants}
 import scala.io.Source
 import XmlImplicits._
@@ -74,6 +74,11 @@ case class EvComment(text: String) extends XMLEvent
  */
 case class NamespaceBinding(prefix: String, uri: String, parent: NamespaceBinding)
 
+/**
+ * The end of document
+ */
+case object EvDocEnd extends XMLEvent
+
 /**
  * Main entry point into creating an event-based XML parser.  Treating this
  * as a [[scala.collection.Iterator]] will provide access to the generated events.
@@ -108,6 +113,7 @@ class XMLEventReader(src: Source)
       EvProcInstr(ev.getTarget, ev.getData)
     case ev: Comment =>
       EvComment(ev.getText)
+    case ev: EndDocument => EvDocEnd
     }
 }