tk.hongkailiu.test.app.xml.impl.dom.DomXMLParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of test-app Show documentation
The newest version!
package tk.hongkailiu.test.app.xml.impl.dom;

import tk.hongkailiu.test.app.xml.XMLParser;
import tk.hongkailiu.test.app.xml.entity.Company;
import tk.hongkailiu.test.app.xml.entity.Staff;
import lombok.extern.log4j.Log4j;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.HashSet;
import java.util.Set;

@Log4j public class DomXMLParser implements XMLParser {

    @Override public Company parseXML2Company(String filename) {
        Company company = null;
        try {

            DocumentBuilderFactory builderFactory = DocumentBuilderFactory.newInstance();
            DocumentBuilder builder = builderFactory.newDocumentBuilder();
            Document document = builder.parse(new FileInputStream(filename));

            // optional but advised
            document.getDocumentElement().normalize();

            Element rootElement = document.getDocumentElement();
            // root is 
            if (rootElement != null && "company".equals(rootElement.getTagName())) {
                company = new Company();
            }
            NodeList nodes = rootElement.getChildNodes();
            Set staffSet = new HashSet();
            company.setStaffSet(staffSet);
            for (int i = 0; i < nodes.getLength(); i++) {
                Node node = nodes.item(i);

                if (node.getNodeType() == Node.ELEMENT_NODE && "staff".equals(((Element) node).getTagName())) {

                    Element element = (Element) node;
                    //if ("staff".equals(element.getTagName())) {
                        Staff s = new Staff();
                        s.setId(Integer.parseInt(element.getAttribute("id")));
                        s.setFirstname(
                            element.getElementsByTagName("firstname").item(0).getTextContent());
                        s.setLastname(
                            element.getElementsByTagName("lastname").item(0).getTextContent());
                        s.setNickname(
                            element.getElementsByTagName("nickname").item(0).getTextContent());
                        s.setSalary(Float.parseFloat(
                            element.getElementsByTagName("salary").item(0).getTextContent()));
                        staffSet.add(s);
                    }

                //}
            }
        } catch (ParserConfigurationException | FileNotFoundException | SAXException e) {
            log.error(e);
        } catch (IOException e) {
            log.error(e);
        }
        return company;
    }

}