使用JDK自带类库操作XML二 JAXB

最新推荐文章于 2021-02-28 07:54:38 发布

iteye_8851

最新推荐文章于 2021-02-28 07:54:38 发布

阅读量127

点赞数

CC 4.0 BY-SA版权

文章标签： XML JDK F#

本文链接：https://blog.youkuaiyun.com/iteye_8851/article/details/81825011

1. 环境

jdk1.6

2. 代码

public class XmlException extends RuntimeException {

    private static final long serialVersionUID = 1L;

    public XmlException(String message){
        super(message);
    }

    public XmlException(String message, Throwable cause){
        super(message, cause);
    }
}

import org.xml.sax.ErrorHandler;

import org.xml.sax.SAXException;
import org.xml.sax.SAXParseException;

public class ForgivingErrorHandler implements ErrorHandler {

    public void error(SAXParseException exception) throws SAXException {
        System.out.println(exception.getMessage());
    }

    public void fatalError(SAXParseException exception) throws SAXException {
        System.out.println(exception.getMessage());
    }

    public void warning(SAXParseException exception) throws SAXException {
        System.out.println(exception.getMessage());
    }

}

import java.io.File;

import java.io.FileReader;
import java.io.FileWriter;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.io.StringReader;

import javax.xml.XMLConstants;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import javax.xml.transform.Source;
import javax.xml.transform.stream.StreamSource;
import javax.xml.validation.Schema;
import javax.xml.validation.SchemaFactory;
import javax.xml.validation.Validator;

public class Jaxb {

    public static <T extends Object> T parseFromString(Class<T> clazz, String xmlStr)throws XmlException{
        return parse(clazz, new StringReader(xmlStr));
    }
    public static <T extends Object> T parse(Class<T> clazz, final String xmlFile)throws XmlException{
        return parse(clazz, xmlFile, false);
    }

    public static <T extends Object> T parse(Class<T> clazz, final String xmlFile, final boolean isValidate)throws XmlException{
        File f = new File(xmlFile);
        if(!f.exists()){
            return null;
        }
        if(isValidate && !validator(f)){
            throw new XmlException("文件"+xmlFile+"验证失败");
        }
        T obj = null;
        try {
            obj = parse(clazz, new FileReader(f));
        } catch (Exception e) {
            throw new XmlException(e.getMessage(), e);
        }
        return obj;
    }

    public static <T extends Object> T parse(Class<T> clazz, InputStream is)throws XmlException{
        return parse(clazz, is, "UTF-8");
    }
    public static <T extends Object> T parse(Class<T> clazz, InputStream is, final String encoding)throws XmlException{
        T obj = null;
        try {
            obj = parse(clazz, new InputStreamReader(is, encoding));
            is.close();
        } catch (Exception e) {
            throw new XmlException(e.getMessage(), e);
        }
        return obj;
    }
    public static <T extends Object> T parse(Class<T> clazz, Reader reader)throws XmlException{
        T obj = null;
        try {
            JAXBContext context = JAXBContext.newInstance(clazz);
            Unmarshaller um = context.createUnmarshaller();
            obj = (T)um.unmarshal(reader);
            reader.close();
        } catch (Exception e) {
            throw new XmlException(e.getMessage(), e);
        }
        return obj;
    }

    /**
    * 验证文档是否有效
    * @param f File
    * @return
    */
    public static boolean validator(final File f){
        //加载一个模式工厂
        SchemaFactory schemaFactory = SchemaFactory.newInstance(XMLConstants.W3C_XML_SCHEMA_NS_URI);
        try {
            //编译源文件中的模式
            Schema schema = schemaFactory.newSchema();
            //用编译后的模式创建一个验证程序
            Validator validator = schema.newValidator();
            validator.setErrorHandler(new ForgivingErrorHandler());
            //为需要验证的文档创建Source对象。StreamSource通常最简单
            Source source = new StreamSource(f);
            //验证输入的源文档。如果文档无效，则抛出SAXException.否则什么也不显示
            validator.validate(source);
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
        return true;
    }

    public static void write(Object obj , final String xmlFile)throws XmlException{
        File f = new File(xmlFile);
        try {
            if(!f.exists()){
                f.createNewFile();
            }
            JAXBContext context = JAXBContext.newInstance(obj.getClass());
            Marshaller m = context.createMarshaller();
            m.setProperty(Marshaller.JAXB_ENCODING, "UTF-8"); //指定已编组 XML 数据中的输出编码,默认为UTF-8
            m.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true); //使用换行和缩排对已编组 XML 数据进行格式化,默认为false
            FileWriter fw = new FileWriter(f);
            m.marshal(obj, fw);
            fw.close();
        } catch (Exception e) {
            throw new XmlException(e.getMessage(), e);
        }
    }

}

3. 使用的XSD文件books.xsd

<?xml version="1.0" encoding="UTF-8"?>

<xs:schema xmlns:xs="http://www.w3.org/2001/XMLSchema" elementFormDefault="qualified" attributeFormDefault="unqualified">
    <xs:element name="ENTER_NAME_OF_ROOT_ELEMENT_HERE">
        <xs:annotation>
            <xs:documentation>Comment describing your root element</xs:documentation>
        </xs:annotation>
    </xs:element>
    <xs:element name="book">
        <xs:complexType>
            <xs:sequence>
                <xs:element name="id" type="xs:string"/>
                <xs:element name="name" type="xs:string"/>
                <xs:element name="price" type="xs:int"/>
            </xs:sequence>
        </xs:complexType>
    </xs:element>
    <xs:element name="books">
        <xs:complexType>
            <xs:sequence>
                <xs:element ref="book" maxOccurs="unbounded"/>
            </xs:sequence>
        </xs:complexType>
    </xs:element>
</xs:schema>