com.aliyuncs.utils.XmlUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of discover-plug-alidayu Show documentation
There is a newer version: 1.8.10
Show newest version
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package com.aliyuncs.utils;

import java.io.Closeable;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.util.ArrayList;
import java.util.List;

import javax.xml.XMLConstants;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.Source;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamSource;
import javax.xml.validation.Schema;
import javax.xml.validation.SchemaFactory;
import javax.xml.validation.Validator;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;

public final class XmlUtils {

	public static Document getDocument(String payload) 
			throws ParserConfigurationException,SAXException,IOException {
		if (payload == null || payload.length() < 1) 
			return null;

		StringReader sr = new StringReader(payload);
		InputSource source = new InputSource(sr);
		
		return getDocument(source, null);
	}
	
	public static Document getDocument(InputSource xml, InputStream xsd) 
			throws ParserConfigurationException,SAXException,IOException {
		Document doc = null;

		try {
			DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
			if (xsd != null) {
				dbf.setNamespaceAware(true);
			}

			DocumentBuilder builder = dbf.newDocumentBuilder();
			doc = builder.parse(xml);

			if (xsd != null)
				validateXml(doc, xsd);
		} finally {
			closeStream(xml.getByteStream());
		}
		
		return doc;
	}

	public static Element getRootElementFromString(String payload) 
			throws ParserConfigurationException, SAXException, IOException {
		
		return getDocument(payload).getDocumentElement();
	}

	
	public static List getChildElements(Element parent, String tagName) {
		if (null == parent)
			return null;
		NodeList nodes = parent.getElementsByTagName(tagName);
		List elements = new ArrayList();

		for (int i = 0; i < nodes.getLength(); i++) {
			Node node = nodes.item(i);
			if (node.getParentNode() == parent)
				elements.add((Element) node);
		}

		return elements;
	}

	
	public static List getChildElements(Element parent) {
		if (null == parent)
			return null;
		
		NodeList nodes = parent.getChildNodes();
		List elements = new ArrayList();

		for (int i = 0; i < nodes.getLength(); i++) {
			Node node = nodes.item(i);
			if (node.getNodeType() == Node.ELEMENT_NODE) 
				elements.add((Element) node);
		}

		return elements;
	}
	
	public static void validateXml(InputStream xml, InputStream xsd) 
			throws SAXException, IOException, ParserConfigurationException {
		try {
			DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
			dbf.setNamespaceAware(true);
			Document doc = dbf.newDocumentBuilder().parse(xml);
			validateXml(doc, xsd);
		} finally {
			closeStream(xml);
			closeStream(xsd);
		}
	}

	public static void validateXml(Node root, InputStream xsd) 
			throws SAXException, IOException {
		try {
			Source source = new StreamSource(xsd);
			Schema schema = SchemaFactory.newInstance(
					XMLConstants.W3C_XML_SCHEMA_NS_URI).newSchema(source);

			Validator validator = schema.newValidator();
			validator.validate(new DOMSource(root));
		} finally {
			closeStream(xsd);
		}
	}
	
	private static void closeStream(Closeable stream) {
		if (stream != null) {
			try {
				stream.close();
			} catch (IOException e) {
			}
		}
	}
}