Merge Hibernate Tools into ORM : Fix failing test on Oracle and Graal platforms

Author: koentsje

tooling/hibernate-reveng/src/main/java/org/hibernate/tool/reveng/internal/xml/AbstractXMLPrettyPrinterStrategy.java

@@ -22,10 +22,10 @@
 
 public abstract class AbstractXMLPrettyPrinterStrategy implements XMLPrettyPrinterStrategy {
 
+	private static DocumentBuilderFactory DOCUMENT_BUILDER_FACTORU = null;
+
 	protected Document newDocument(String xml, String encoding) throws SAXException, IOException, ParserConfigurationException {
-		final DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
-		dbf.setFeature("http://apache.org/xml/features/nonvalidating/load-external-dtd", false);
-		final Document document = dbf
+		final Document document = getDocumentBuilderFactory()
 				.newDocumentBuilder()
 				.parse(new InputSource(new ByteArrayInputStream(xml.getBytes(encoding))));
 		document.normalize();
@@ -43,4 +43,39 @@ protected void removeWhitespace(final Document document) throws XPathExpressionE
 			node.getParentNode().removeChild(node);
 		}
 	}
+
+	private static DocumentBuilderFactory getDocumentBuilderFactory() {
+		if (DOCUMENT_BUILDER_FACTORU == null) {
+			DOCUMENT_BUILDER_FACTORU = createDocumentBuilderFactory();
+		}
+		return DOCUMENT_BUILDER_FACTORU;
+	}
+
+	private static DocumentBuilderFactory createDocumentBuilderFactory() {
+		DocumentBuilderFactory result = null;
+		// Save the documentBuilderFactoryClass and set it to the default jre version
+		String documentBuilderFactoryClass = System.getProperty( "javax.xml.parsers.DocumentBuilderFactory" );
+		System.setProperty(
+				"javax.xml.parsers.DocumentBuilderFactory",
+				"com.sun.org.apache.xerces.internal.jaxp.DocumentBuilderFactoryImpl");
+		try {
+			result = DocumentBuilderFactory.newInstance();
+			result.setFeature("http://apache.org/xml/features/nonvalidating/load-external-dtd", false);
+		}
+		catch (ParserConfigurationException e) {
+			throw new RuntimeException(e);
+		}
+		finally {
+			// Restore the documentBuilderFactory class
+			if (documentBuilderFactoryClass != null) {
+				System.setProperty(
+						"javax.xml.parsers.DocumentBuilderFactory",
+						documentBuilderFactoryClass );
+			}
+			else {
+				System.clearProperty( "javax.xml.parsers.DocumentBuilderFactory" );
+			}
+		}
+		return result;
+	}
 }

tooling/hibernate-reveng/src/main/java/org/hibernate/tool/reveng/internal/xml/TrAXPrettyPrinterStrategy.java

@@ -14,8 +14,15 @@
 import javax.xml.transform.dom.DOMSource;
 import javax.xml.transform.stream.StreamResult;
 import java.io.StringWriter;
+import java.util.logging.Logger;
 
 public class TrAXPrettyPrinterStrategy extends AbstractXMLPrettyPrinterStrategy {
+
+	private static final Logger LOGGER = Logger.getLogger( TrAXPrettyPrinterStrategy.class.getName() );
+	private static final TransformerFactory TRANSFORMER_FACTORY = TransformerFactory.newInstance(
+			"com.sun.org.apache.xalan.internal.xsltc.trax.TransformerFactoryImpl",
+			null);
+
 	private int indent = 4;
 	private boolean omitXmlDeclaration;
 
@@ -32,17 +39,18 @@ public String prettyPrint(String xml) throws Exception {
 	}
 
 	protected Transformer newTransformer(final Document document) throws TransformerConfigurationException {
-		final TransformerFactory transformerFactory = newTransformerFactory();
 
-		final Transformer transformer = transformerFactory.newTransformer();
+		final Transformer transformer = TRANSFORMER_FACTORY.newTransformer();
 		transformer.setOutputProperty(OutputKeys.METHOD, "xml");
 		transformer.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
 		transformer.setOutputProperty(OutputKeys.INDENT, "yes");
 		transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, isOmitXmlDeclaration() ? "yes" : "no");
 		try {
 			transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", String.valueOf(getIndent()));
 		}
-		catch (IllegalArgumentException ignored) {
+		catch (IllegalArgumentException e) {
+			LOGGER.severe( "An IllegalArgumentException happened while adding the 'indent' property." );
+			throw new RuntimeException(e);
 		}
 
 		final DocumentType doctype = document.getDoctype();
@@ -54,17 +62,6 @@ protected Transformer newTransformer(final Document document) throws Transformer
 		return transformer;
 	}
 
-	protected TransformerFactory newTransformerFactory() {
-		final TransformerFactory transformerFactory = TransformerFactory.newInstance();
-		try {
-			transformerFactory.setAttribute("indent-number", getIndent());
-		}
-		catch (IllegalArgumentException ignored) {
-		}
-
-		return transformerFactory;
-	}
-
 	public int getIndent() {
 		return indent;
 	}
@@ -80,4 +77,5 @@ public boolean isOmitXmlDeclaration() {
 	public void setOmitXmlDeclaration(boolean omitXmlDeclaration) {
 		this.omitXmlDeclaration = omitXmlDeclaration;
 	}
+
 }