com.hframework.common.util.message.XmlUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hframe-common Show documentation
There is a newer version: 1.0.2
package com.hframework.common.util.message;

import com.hframework.common.util.file.FileUtils;
import com.hframework.common.util.StringUtils;
import com.thoughtworks.xstream.XStream;
import com.thoughtworks.xstream.io.xml.DomDriver;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;

/**
 * User: zhangqh6
 * Date: 2016/1/21 13:28:28
 */
public class XmlUtils {
    private static final Logger logger = LoggerFactory.getLogger(XmlUtils.class);

    public static  T readValue(String content, Class valueType) {
        XStream xstream = new XStream(new DomDriver());
        xstream.processAnnotations(valueType);
        return (T) xstream.fromXML(content);
    }

    /**
     * 读取文件-类加载器中文件
     * @param filePath
     * @param valueType
     * @param 
     * @return
     * @throws IOException
     */
    public static  T readValueFromFile(String filePath, Class valueType) throws IOException {
        URL fileResource = XmlUtils.class.getResource("/" + filePath);
        if(fileResource != null) {
            return readValueFromAbsoluteFilePath(fileResource.getPath(), valueType);
        }else {
            return readValueFromAbsoluteFilePath(XmlUtils.class.getResource("/").getPath() + "/" + filePath, valueType);
        }
    }

    /**
     * 读取文件-根路径 +相对路径
     * @param rootPath
     * @param filePath
     * @param valueType
     * @param 
     * @return
     * @throws IOException
     */
    public static  T readValueFromFile(String rootPath, String filePath, Class valueType) throws IOException {
        if(new File(rootPath).exists()) {
            return readValueFromAbsoluteFilePath(rootPath + "/" + filePath,valueType);
        }else {
            return readValueFromFile(filePath, valueType);
        }
    }

    /**
     * 读取文件-绝对路径
     * @param filePath
     * @param valueType
     * @param 
     * @return
     * @throws IOException
     */
    public static  T readValueFromAbsoluteFilePath( String filePath, Class valueType) throws IOException {
        logger.debug("入参：{}|{}",filePath,valueType);
        String xmlString = FileUtils.readFile(filePath);
        logger.debug("报文：{}", xmlString);
        T t = readValue(xmlString, valueType);
        logger.debug("对象：{}", t);
        return t;
    }

    /**
     * 读取文件列表-类加载器中资源目录
     * @param directory
     * @param valueType
     * @param format
     * @param 
     * @return
     * @throws IOException
     */
    public static  List readValuesFromDirectory(String directory, Class valueType, String format) throws IOException {
        URL fileResource = XmlUtils.class.getResource("/" + directory);
        if(fileResource != null) {
            return readValueFromAbsoluteDirectoryPath(fileResource.getPath(), valueType, format);
        }else {
            return readValueFromAbsoluteDirectoryPath(XmlUtils.class.getResource("/").getPath() + "/" + directory, valueType, format);
        }
    }


    public static  List readValuesFromDirectory(String rootPath, String directory, Class valueType, String format) throws IOException {
        if(new File(rootPath).exists()) {
            return readValueFromAbsoluteDirectoryPath(rootPath + "/" + directory, valueType, format);
        }else {
            return readValuesFromDirectory(directory, valueType, format);
        }

    }

    public static  List readValuesFromDirectory(String directory, Class valueType) throws IOException {
        return readValuesFromDirectory(directory,valueType,null);
    }

    public static  List readValueFromAbsoluteDirectoryPath(String directory, Class valueType, String format) throws IOException {
        List result = new ArrayList();
        logger.debug("入参：{}|{}",directory,valueType);
        File[] fileList = FileUtils.getFileList(new File(directory));
        if(fileList == null) {
            logger.warn("file not exists !");
            return result;
        }
        for (File file : fileList) {
            if(StringUtils.isBlank(format) || file.getName().endsWith(format)) {
                String xmlString = FileUtils.readFile(file.getAbsolutePath());
                logger.debug("报文：{}", xmlString);
                T t = readValue(xmlString, valueType);
                result.add(t);
                logger.debug("对象：{}", t);
            }
        }
        return result;
    }



    public static   String writeValueAsString(T t) throws IOException {
        XStream xstream = new XStream(new DomDriver("utf8"));
//        XStream xstream = new XStream(new DomDriver("UTF-8", new NoNameCoder()));

        xstream.processAnnotations(t.getClass());// 识别obj类中的注解
        /*
         // 以压缩的方式输出XML
         StringWriter sw = new StringWriter();
         xstream.marshal(obj, new CompactWriter(sw));
         return sw.toString();
         */
        // 以格式化的方式输出XML
        return xstream.toXML(t);
    }

    public static void main(String[] args) throws IOException {
        String rootClassPath = Thread.currentThread().getContextClassLoader ().getResource("").getPath();
        String xmlString = readFile(rootClassPath + "test.xml");
        System.out.println(xmlString);
//        Menu menu = readValue(jsonString, Menu.class);
//        Map map = readValue(xmlString, Map.class);
//        System.out.println(map);




    }

    private static String readFile(String fileName) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(new FileInputStream(new File(fileName))));
        String retStr = "";
        String str = br.readLine();
        while(str != null) {
            retStr += str;
            str = br.readLine();
        }
        return retStr;
    }
}