com.gitee.yongzhuzl.commonutil.util.wx.pay.WxXmlUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of commonutil Show documentation
commonutil是一个功能丰富的Java工具类库，通过封装静态方法，降低学习成本，提升工作效率。
There is a newer version: 4.4.8
package com.gitee.yongzhuzl.commonutil.util.wx.pay;

import com.gitee.yongzhuzl.commonutil.util.empty.EmptyUtil;
import org.jdom.Document;
import org.jdom.Element;
import org.jdom.input.SAXBuilder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.*;

/**
 * description: 微信XML解析工具类
 * @author 朱正磊
 * @date 2019-12-05 16:58:51
 */
@SuppressWarnings("ALL")
public class WxXmlUtil {

    private static final Logger log = LoggerFactory.getLogger(WxXmlUtil.class);

    /**
     * description: XML数据解析成Map（会进行签名校验）
     * @author 朱正磊
     * @date 2019-10-23 09:51:25
     * @param xmlStr xml格式的字符串
     * @return 返回TreeMap对象
     */
    public static SortedMap xmlParse(String xmlStr) {
        log.info("XML原始待解析数据：{}", xmlStr);
        xmlStr = xmlStr.replaceFirst("encoding=\".*\"", "encoding=\"UTF-8\"");
        log.info("XML正常待解析数据：{}", xmlStr);
        // TreeMap默认字母序
        SortedMap map = new TreeMap<>();
        if (EmptyUtil.isEmpty(xmlStr)) {
            return map;
        }
        try {
            InputStream in = new ByteArrayInputStream(xmlStr.getBytes("UTF-8"));
            SAXBuilder builder = new SAXBuilder();
            Document doc = builder.build(in);
            Element root = doc.getRootElement();
            List list = root.getChildren();
            for (Object aList : list) {
                Element e = (Element) aList;
                String k = e.getName();
                String v;
                List children = e.getChildren();
                if (children.isEmpty()) {
                    v = e.getTextNormalize();
                } else {
                    v = getChildrenText(children);
                }
                map.put(k, v);
            }
            //关闭流
            in.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
        log.info("XML解析结果：{}", map);
        return map;
    }

    /**
     * description: 获取子结点的XML
     * @author 朱正磊
     * @date 2019-10-23 09:52:35
     * @param children 子节点列表
     * @return 返回子结点的xml
     */
    private static String getChildrenText(List children) {
        StringBuilder sb = new StringBuilder();
        if (EmptyUtil.isNotEmpty(children)) {
            for (Object aChildren : children) {
                Element e = (Element) aChildren;
                String name = e.getName();
                String value = e.getTextNormalize();
                sb.append("<").append(name).append(">");
                List list = e.getChildren();
                if (EmptyUtil.isNotEmpty(list)) {
                    sb.append(getChildrenText(list));
                }
                sb.append(value);
                sb.append("");
            }
        }
        return sb.toString();
    }

    /**
     * description: 将请求参数封装的Map对象转换成XML字符串
     * @author 朱正磊
     * @date 2019-12-05 17:55:07
     * @param paramsMap 参数封装对象
     * @return 返回XML字符串
     */
    public static String paramsMapToXml(SortedMap paramsMap) {
        log.info("待转换成XML的请求参数：{}", paramsMap);
        StringBuilder stringBuilder = new StringBuilder();
        stringBuilder.append("");
        Set> entries = paramsMap.entrySet();
        for (Map.Entry entry : entries) {
            String k = entry.getKey();
            Object v = entry.getValue();
            if ("attach".equalsIgnoreCase(k) || "body".equalsIgnoreCase(k) || "sign".equalsIgnoreCase(k)) {
                stringBuilder.append("<").append(k).append(">").append("");
            } else {
                stringBuilder.append("<").append(k).append(">").append(v).append("");
            }
        }
        stringBuilder.append("");
        String xmlStr = stringBuilder.toString();
        log.info("生成的XML数据：{}", xmlStr);
        return xmlStr;
    }

}