me.chanjar.weixin.util.xml.XmlTransformer Maven / Gradle / Ivy

Go to download
package me.chanjar.weixin.util.xml;

import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.io.StringWriter;
import java.io.Writer;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;

import org.xml.sax.InputSource;

import com.sun.xml.bind.marshaller.CharacterEscapeHandler;

public class XmlTransformer {

  /**
   * xml -> pojo
   * @param clazz
   * @param object
   * @return
   * @throws JAXBException 
   */
  public static  T fromXml(Class clazz, String xml) throws JAXBException {
    JAXBContext context = JAXBContext.newInstance(clazz);
    Unmarshaller um = context.createUnmarshaller();
    T object = (T) um.unmarshal(new StringReader(xml));
    return object;
  }
  
  public static  T fromXml(Class clazz, InputStream is) throws JAXBException {
    JAXBContext context = JAXBContext.newInstance(clazz);
    Unmarshaller um = context.createUnmarshaller();
    InputSource inputSource = new InputSource(is);
    inputSource.setEncoding("utf-8");
    T object = (T) um.unmarshal(inputSource);
    return object;
  }
  
  /**
   * pojo -> xml
   * @param clazz
   * @return
   * @throws JAXBException 
   */
  public static  String toXml(Class clazz, T object) throws JAXBException {
    StringWriter stringWriter = new StringWriter();
    toXml(clazz, object, stringWriter);
    return stringWriter.getBuffer().toString();
  }
  
  public static  void toXml(Class clazz, T object, Writer writer) throws JAXBException {
    JAXBContext context = JAXBContext.newInstance(clazz);
    Marshaller m = context.createMarshaller();
    m.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
    m.setProperty(CharacterEscapeHandler.class.getName(), characterUnescapeHandler);
    m.marshal(object, writer);
  }
  
  protected static CharacterEscapeHandler characterUnescapeHandler = new CharacterUnescapeHandler();
  
  protected static class CharacterUnescapeHandler implements CharacterEscapeHandler {
    public void escape(char[] ac, int i, int j, boolean flag, Writer writer) throws IOException {
      writer.write(ac, i, j);
    }
  }
  
}