io.github.nichetoolkit.rest.helper.XmlHelper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rest-toolkit-utils Show documentation
Rest toolkit utils project for Spring Boot
There is a newer version: 1.1.0
package io.github.nichetoolkit.rest.helper;

import io.github.nichetoolkit.rest.error.often.XmlMarshalException;
import io.github.nichetoolkit.rest.error.often.XmlReadException;
import io.github.nichetoolkit.rest.error.often.XmlWriteException;
import io.github.nichetoolkit.rest.util.GeneralUtils;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import javax.xml.bind.*;
import javax.xml.transform.stream.StreamSource;
import java.io.*;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;

/**
 * XmlHelper
 * @author Cyan ([email protected])
 * @version v1.0.0
 */
public class XmlHelper {

    public static void encode(String filename, HttpServletResponse response) {
        String fileName = new String(filename.trim().getBytes(StandardCharsets.UTF_8), StandardCharsets.UTF_8);
        try {
            fileName = URLEncoder.encode(filename, StandardCharsets.UTF_8.toString());
        } catch (UnsupportedEncodingException ignored) {
        }
        String content = "attachment;filename=".concat(fileName);
        response.setHeader("Content-Disposition", content);
        response.setHeader("ContentType", "application/xml;charset=UTF-8");
        response.setContentType("application/octet-stream");
    }

    public static  Unmarshaller unmarshaller (Class clazz) throws XmlMarshalException {
        try {
            JAXBContext context = JAXBContext.newInstance(clazz);
            return context.createUnmarshaller();
        } catch (JAXBException exception) {
            throw new XmlMarshalException(exception.getMessage());
        }
    }


    public static  Marshaller marshaller(Class clazz) throws XmlMarshalException {
        try {
            JAXBContext context = JAXBContext.newInstance(clazz);
            return context.createMarshaller();
        } catch (JAXBException exception) {
            throw new XmlMarshalException(exception.getMessage());
        }
    }

    public static  T read(MultipartFile xmlFile, Class clazz) throws XmlReadException {
        if (GeneralUtils.isEmpty(xmlFile)) {
            return null;
        }
        try (InputStream inputStream = xmlFile.getInputStream()) {
            return JAXB.unmarshal(inputStream, clazz);
        } catch (DataBindingException | IOException exception) {
            throw new XmlReadException(exception.getMessage());
        }
    }

    public static  T read(File xmlFile, Class clazz) throws XmlReadException {
        if (GeneralUtils.isEmpty(xmlFile) || !xmlFile.exists()) {
            return null;
        }
        try (InputStream inputStream = new FileInputStream(xmlFile)) {
            return JAXB.unmarshal(inputStream, clazz);
        } catch (DataBindingException | IOException exception) {
            throw new XmlReadException(exception.getMessage());
        }
    }

    public static  T read(Unmarshaller unmarshaller, InputStream inputStream, Class clazz) throws XmlReadException {
        if (GeneralUtils.isEmpty(inputStream)) {
            return null;
        }
        try {
            JAXBElement jaxbElement = unmarshaller.unmarshal(new StreamSource(inputStream), clazz);
            return jaxbElement.getValue();
        } catch ( JAXBException | DataBindingException exception) {
            throw new XmlReadException(exception.getMessage());
        }
    }

    public static  T read(InputStream inputStream, Class clazz) throws XmlReadException {
        if (GeneralUtils.isEmpty(inputStream)) {
            return null;
        }
        try {
            return JAXB.unmarshal(inputStream, clazz);
        } catch ( DataBindingException exception) {
            throw new XmlReadException(exception.getMessage());
        }
    }

    public static  void write(Marshaller marshaller, T xmlObject, String filename, HttpServletResponse response) throws XmlWriteException {
        if (GeneralUtils.isEmpty(xmlObject)) {
            return;
        }
        try {
            encode(filename, response);
            ServletOutputStream outputStream = response.getOutputStream();
            marshaller.marshal(xmlObject,outputStream);
        } catch (JAXBException| IOException exception) {
            throw new XmlWriteException(exception.getMessage());
        }
    }

    public static  void write(Marshaller marshaller, T xmlObject, HttpServletResponse response) throws XmlWriteException {
        if (GeneralUtils.isEmpty(xmlObject)) {
            return;
        }
        try {
            ServletOutputStream outputStream = response.getOutputStream();
            marshaller.marshal(xmlObject,outputStream);
        } catch (JAXBException| IOException exception) {
            throw new XmlWriteException(exception.getMessage());
        }
    }

    public static  void write(Marshaller marshaller, T xmlObject, OutputStream outputStream) throws XmlWriteException {
        if (GeneralUtils.isEmpty(xmlObject)) {
            return;
        }
        try {
            marshaller.marshal(xmlObject,outputStream);
        } catch (JAXBException exception) {
            throw new XmlWriteException(exception.getMessage());
        }
    }

    public static  void write(T xmlObject, String filename, HttpServletResponse response) throws XmlWriteException {
        if (GeneralUtils.isEmpty(xmlObject)) {
            return;
        }
        try {
            encode(filename,response);
            ServletOutputStream outputStream = response.getOutputStream();
            JAXB.marshal(xmlObject,outputStream);
        } catch (IOException exception) {
            throw new XmlWriteException(exception.getMessage());
        }
    }

    public static  void write(T xmlObject, HttpServletResponse response) throws XmlWriteException {
        if (GeneralUtils.isEmpty(xmlObject)) {
            return;
        }
        try {
            ServletOutputStream outputStream = response.getOutputStream();
            JAXB.marshal(xmlObject,outputStream);
        } catch (IOException exception) {
            throw new XmlWriteException(exception.getMessage());
        }
    }
}