minor change

xmlobjects · Dec 10, 2023 · 9d17d36 · 9d17d36
1 parent 5eb517b
commit 9d17d36
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 10 deletions.
diff --git a/src/main/java/org/xmlobjects/stream/XMLOutput.java b/src/main/java/org/xmlobjects/stream/XMLOutput.java
@@ -35,6 +35,8 @@ public abstract class XMLOutput<T extends XMLOutput<?>> extends SAXFilter implem
     protected boolean writeXMLDeclaration = true;
     protected String[] headerComment;
 
+    protected abstract T self();
+
     public XMLOutput() {
         super(new DefaultHandler());
     }
@@ -57,11 +59,10 @@ String createPrefix(String namespaceURI) {
         return prefixMapping.createPrefix(namespaceURI);
     }
 
-    @SuppressWarnings("unchecked")
     public T withPrefix(String prefix, String namespaceURI) {
         prefixMapping.pushContext();
         prefixMapping.declarePrefix(prefix, namespaceURI);
-        return (T) this;
+        return self();
     }
 
     public String getNamespaceURI(String prefix) {
@@ -76,46 +77,42 @@ public String getSchemaLocation(String namespaceURI) {
         return schemaLocations.get(namespaceURI);
     }
 
-    @SuppressWarnings("unchecked")
     public T withSchemaLocation(String namespaceURI, String schemaLocation) {
         if (namespaceURI != null && schemaLocation != null) {
             schemaLocations.put(namespaceURI, schemaLocation);
             withPrefix("xsi", XMLConstants.W3C_XML_SCHEMA_INSTANCE_NS_URI);
         }
 
-        return (T) this;
+        return self();
     }
 
     public String getIndent() {
         return indent;
     }
 
-    @SuppressWarnings("unchecked")
     public T withIndent(String indent) {
         this.indent = indent;
-        return (T) this;
+        return self();
     }
 
     public boolean isWriteXMLDeclaration() {
         return writeXMLDeclaration;
     }
 
-    @SuppressWarnings("unchecked")
     public T writeXMLDeclaration(boolean writeXMLDeclaration) {
         this.writeXMLDeclaration = writeXMLDeclaration;
-        return (T) this;
+        return self();
     }
 
     public String[] getHeaderComment() {
         return headerComment;
     }
 
-    @SuppressWarnings("unchecked")
     public T withHeaderComment(String... headerMessage) {
         if (headerMessage != null) {
             this.headerComment = headerMessage;
         }
 
-        return (T) this;
+        return self();
     }
 }
diff --git a/src/main/java/org/xmlobjects/util/xml/SAXOutputHandler.java b/src/main/java/org/xmlobjects/util/xml/SAXOutputHandler.java
@@ -105,4 +105,9 @@ public void endElement(String uri, String localName, String qName) throws SAXExc
 
         prefixMapping.popContext();
     }
+
+    @Override
+    protected SAXOutputHandler self() {
+        return this;
+    }
 }
diff --git a/src/main/java/org/xmlobjects/util/xml/SAXWriter.java b/src/main/java/org/xmlobjects/util/xml/SAXWriter.java
@@ -517,4 +517,9 @@ private void writeAttributeContent(String content) throws IOException {
 
         writer.write(content, pos, end - pos);
     }
+
+    @Override
+    protected SAXWriter self() {
+        return this;
+    }
 }