使用JDK自帶的Stax操作XML

操作的books.xml

<?xml version="1.0" encoding="UTF-8"?>
<bookstore>
<book category="COOKING">
  <title lang="en">Everyday Italian</title>
  <author>Giada De Laurentiis</author>
  <year>2005</year>
  <price>30.00</price>
</book>

<book category="CHILDREN">
  <title lang="en">Harry Potter</title>
  <author>J K. Rowling</author>
  <year>2005</year>
  <price>29.99</price>
</book>

<book category="WEB">
  <title lang="en">XQuery Kick Start</title>
  <author>James McGovern</author>
  <author>Per Bothner</author>
  <author>Kurt Cagle</author>
  <author>James Linn</author>
  <author>Vaidyanathan Nagarajan</author>
  <year>2003</year>
  <price>49.99</price>
</book>

<book category="WEB">
  <title lang="en">Learning XML</title>
  <author>Erik T. Ray</author>
  <year>2003</year>
  <price>39.95</price>
</book>
</bookstore>

測試類：

import java.io.IOException;
import java.io.InputStream;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.stream.EventFilter;
import javax.xml.stream.XMLEventReader;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLOutputFactory;
import javax.xml.stream.XMLStreamConstants;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import javax.xml.stream.XMLStreamWriter;
import javax.xml.stream.events.XMLEvent;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Result;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import javax.xml.xpath.XPath;
import javax.xml.xpath.XPathConstants;
import javax.xml.xpath.XPathFactory;
import org.junit.Test;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;

public class TestStax {

    @Test
    /**
     * 讀取xml的<book category="COOKING">
     * 獲取book元素的屬性和屬性值
     */
    public void test02() {
        XMLInputFactory factory = XMLInputFactory.newInstance();
        InputStream is = null;
        try {
            is = TestStax.class.getClassLoader().getResourceAsStream("books.xml");
            XMLStreamReader reader = factory.createXMLStreamReader(is);
            while(reader.hasNext()) {
                int type = reader.next();
                //判斷節點類型是否是開始或者結束或者文本節點,之后根據情況及進行處理
                if(type==XMLStreamConstants.START_ELEMENT) {
                    String name = reader.getName().toString();
                    if(name.equals("book")) {
                        System.out.println(reader.getAttributeName(0)+":"+reader.getAttributeValue(0));
                    }
                }
            }
        } catch (XMLStreamException e) {
            e.printStackTrace();
        } finally {
            try {
                if(is!=null) is.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    
    @Test
    /**
     *  讀取title和price標簽的值
        <title lang="en">Everyday Italian</title>
        <price>30.00</price>
     */
    public void test03() {
        XMLInputFactory factory = XMLInputFactory.newInstance();
        InputStream is = null;
        try {
            is = TestStax.class.getClassLoader().getResourceAsStream("books.xml");
            XMLStreamReader reader = factory.createXMLStreamReader(is);
            while(reader.hasNext()) {
                int type = reader.next();
                if(type==XMLStreamConstants.START_ELEMENT) {
                    String name = reader.getName().toString();
                    if(name.equals("title")) {
                        System.out.print(reader.getElementText()+":");
                    }
                    if(name.equals("price")) {
                        System.out.print(reader.getElementText()+"\n");
                    }
                }
            }
        } catch (XMLStreamException e) {
            e.printStackTrace();
        } finally {
            try {
                if(is!=null) is.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    
    @Test
    /**
     * 使用XMLEvent來讀取title和price標簽的值
     * <title lang="en">Everyday Italian</title>
     * <price>30.00</price>
     */
    public void test04() {
        XMLInputFactory factory = XMLInputFactory.newInstance();
        InputStream is = null;
        try {
            is = TestStax.class.getClassLoader().getResourceAsStream("books.xml");
            //基於迭代模型的操作方式
            XMLEventReader reader = factory.createXMLEventReader(is);
            int num = 0;
            while(reader.hasNext()) {
                //通過XMLEvent來獲取是否是某種節點類型
                XMLEvent event = reader.nextEvent();
                if(event.isStartElement()) {
                    //通過event.asxxx轉換節點
                    String name = event.asStartElement().getName().toString();
                    if(name.equals("title")) {
                        System.out.print(reader.getElementText()+":");
                    }
                    if(name.equals("price")) {
                        System.out.print(reader.getElementText()+"\n");
                    }
                }
                num++;
            }
            System.out.println(num);
        } catch (XMLStreamException e) {
            e.printStackTrace();
        } finally {
            try {
                if(is!=null) is.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    
    @Test
    /**
     * 使用Filter過濾的不需要操作的元素，提高效率
     * 同時使用XMLEvent
     */
    public void test05() {
        XMLInputFactory factory = XMLInputFactory.newInstance();
        InputStream is = null;
        try {
            is = TestStax.class.getClassLoader().getResourceAsStream("books.xml");
            //基於Filter的過濾方式，可以有效的過濾掉不用進行操作的節點，效率會高一些
            XMLEventReader reader = factory.createFilteredReader(factory.createXMLEventReader(is),
                    new EventFilter() {
                        @Override
                        public boolean accept(XMLEvent event) {
                            //返回true表示會顯示，返回false表示不顯示
                            if(event.isStartElement()) {
                                String name = event.asStartElement().getName().toString();
                                if(name.equals("title")||name.equals("price"))
                                    return true;
                            }
                            return false;
                        }
                    });
            int num = 0;
            while(reader.hasNext()) {
                //通過XMLEvent來獲取是否是某種節點類型
                XMLEvent event = reader.nextEvent();
                if(event.isStartElement()) {
                    //通過event.asxxx轉換節點
                    String name = event.asStartElement().getName().toString();
                    if(name.equals("title")) {
                        System.out.print(reader.getElementText()+":");
                    }
                    if(name.equals("price")) {
                        System.out.print(reader.getElementText()+"\n");
                    }
                }
                num++;
            }
            System.out.println(num);
        } catch (XMLStreamException e) {
            e.printStackTrace();
        } finally {
            try {
                if(is!=null) is.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    
    @Test
    /**
     * 使用XPath和表達式精確定位元素(基於document的形式，在文檔很大的情況下效率差)
     * 表達式：//book[@category='WEB']
     * 意義：查找book元素的屬性等於WEB的節點
     */
    public void test06() {
        InputStream is = null;
        try {
            is = TestStax.class.getClassLoader().getResourceAsStream("books.xml");
            //創建文檔處理對象
            DocumentBuilder db = DocumentBuilderFactory.newInstance().newDocumentBuilder();
            //通過DocumentBuilder創建doc的文檔對象
            Document doc = db.parse(is);
            //創建XPath
            XPath xpath = XPathFactory.newInstance().newXPath();
            //第一個參數就是xpath,第二參數就是文檔
            NodeList list = (NodeList)xpath.evaluate("//book[@category='WEB']", doc, XPathConstants.NODESET);
            for(int i=0;i<list.getLength();i++) {
                //遍歷輸出相應的結果
                Element e = (Element)list.item(i);
                System.out.println(e.getElementsByTagName("title").item(0).getTextContent());
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if(is!=null) is.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    
    @Test
    /**
     * 使用XMLStreamWriter生成XML
     */
    public void test07() {
        try {
            XMLStreamWriter xsw = XMLOutputFactory.newInstance().createXMLStreamWriter(System.out);
            xsw.writeStartDocument("UTF-8","1.0");
            xsw.writeEndDocument();
            String ns = "http://11:dd";
            xsw.writeStartElement("person");
            xsw.writeAttribute("name", "張三");
            xsw.writeStartElement("age");
            xsw.writeCharacters("1");
            xsw.writeEndElement();
            xsw.writeEndElement();
            xsw.flush();
            xsw.close();
        } catch (Exception e) {
            e.printStackTrace();
        } 
    }
    
    @Test
    /**
     * 使用Xpath定位元素並通過Transformer修改其中元素的值
     */
    public void test08() {
        InputStream is = null;
        try {
            is = TestStax.class.getClassLoader().getResourceAsStream("books.xml");
            //創建文檔處理對象
            DocumentBuilder db = DocumentBuilderFactory.newInstance().newDocumentBuilder();
            //通過DocumentBuilder創建doc的文檔對象
            Document doc = db.parse(is);
            //創建XPath
            XPath xpath = XPathFactory.newInstance().newXPath();
            Transformer tran = TransformerFactory.newInstance().newTransformer();
            tran.setOutputProperty(OutputKeys.ENCODING,"UTF-8");
            tran.setOutputProperty(OutputKeys.INDENT, "yes");
            //第一個參數就是xpath,第二參數就是文檔
            NodeList list = (NodeList)xpath.evaluate("//book[title='Learning XML']", doc,XPathConstants.NODESET);
            //獲取price節點
            Element be = (Element)list.item(0);
            Element e = (Element)(be.getElementsByTagName("price").item(0));
            e.setTextContent("333.9");
            Result result = new StreamResult(System.out);
            //通過tranformer修改節點
            tran.transform(new DOMSource(doc), result);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if(is!=null) is.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}