jt56.comm.code.util.WolfXmlUtil.java Source code

Introduction

Here is the source code for jt56.comm.code.util.WolfXmlUtil.java

Source

/**
 * Program  : WolfXmlUtil.java
 * Author   : zhouq
 * Create   : 2014-6-9 ?10:26:51
 *
 * Copyright 2014 by jt56 Technologies Ltd.
 * All rights reserved.
 *
 * This software is the confidential and proprietary information
 * of jt56 Technologies Ltd.("Confidential Information").  
 * You shall not disclose such Confidential Information and shall 
 * use it only in accordance with the terms of the license agreement 
 * you entered into with jt56 Technologies Ltd.
 *
 */

package jt56.comm.code.util;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.InputStreamReader;
import java.io.PrintStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;

/**
 * 
 * @author zhouq
 * @version 1.0.0
 * @2014-6-9 ?10:26:51
 */
public class WolfXmlUtil {
    private void getAddStrutsElemant(String filePath, String nodexPath) throws Exception {
        Document document = getPath(filePath, "utf-8");
        Element element = document.getRootElement();
        Element nextElement = element.element("package");
        Element newElement = nextElement.addElement("action");
        newElement.addComment("");
        newElement.addAttribute("name", "test");
        newElement.addAttribute("class", "");
        newElement.addAttribute("method", "");
        newElement.addText("hello");
    }

    public void getAddNode(String filePath, String xPath, String newNode, Map<String, String> attrMap, String text)
            throws Exception {
        if (getQueryNode(filePath, xPath, newNode, attrMap, text) < 1) {
            Document document = getPath(filePath, "UTF-8");
            List list = document.selectNodes(xPath);
            System.out.println(xPath);
            Element element = (Element) list.get(0);
            Element newElement = element.addElement(newNode);
            for (Map.Entry entry : attrMap.entrySet()) {
                newElement.addAttribute((String) entry.getKey(), (String) entry.getValue());
            }
            if ((text != null) && (text.trim().length() > 0)) {
                newElement.addText(text);
            }
            getXMLWrite(document, filePath);
            System.out.println("" + xPath + "?");
        } else {
            System.out.println("");
        }
    }

    public int getQueryNode(String filePath, String xPath, String newNode, Map<String, String> attrMap, String text)
            throws Exception {
        int i;
        int count = 0;
        Document document = getPath(filePath, "UTF-8");
        StringBuffer sb = new StringBuffer();
        for (Map.Entry entry : attrMap.entrySet()) {
            sb.append("[@" + ((String) entry.getKey()) + "='" + ((String) entry.getValue()) + "']");
        }
        xPath = xPath + "/" + newNode + sb.toString();
        System.out.println("xPath=" + xPath);
        document.selectNodes(xPath);
        List list = document.selectNodes(xPath);
        for (int j = 0; j < list.size(); ++j) {
            Element element = (Element) list.get(j);
            if (element.getText().equals(text)) {
                ++count;
            }

        }

        return count;
    }

    public void getXMLWrite(Document document, String filePath) throws Exception {
        OutputFormat of = new OutputFormat(" ", true);
        of.setEncoding("UTF-8");
        XMLWriter xw = new XMLWriter(new FileWriter(filePath), of);
        xw.setEscapeText(false);
        xw.write(document);
        xw.close();
        System.out.println(document.asXML());
    }

    public void getEditNode(String filePath, String xPath, Map<String, String> attrMap, String text)
            throws Exception {
        int i;
        Document document = getPath(filePath, "UTF-8");
        List list = document.selectNodes(xPath);
        Element element = (Element) list.get(0);
        if (attrMap != null) {
            for (Map.Entry entry : attrMap.entrySet()) {
                element.addAttribute((String) entry.getKey(), (String) entry.getValue());
            }
        }

        List nodelist = element.elements();
        for (int f = 0; f < nodelist.size(); ++f) {
            Element nodeElement = (Element) nodelist.get(f);
            nodeElement.getParent().remove(nodeElement);
        }
        element.setText(text);
        getXMLWrite(document, filePath);
    }

    public Document getPath(String filePath, String coding) {
        SAXReader saxReader = new SAXReader();

        Document document = null;
        try {
            saxReader.setFeature("http://apache.org/xml/features/nonvalidating/load-external-dtd", false);
            BufferedReader read = new BufferedReader(
                    new InputStreamReader(new FileInputStream(new File(filePath)), coding));
            document = saxReader.read(read);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return document;
    }

    public static void main(String[] args) {
        WolfXmlUtil xml = new WolfXmlUtil();
        String filePath1 = "D:\\MyEclipse 8.5\\ssi\\src\\com\\wei\\ssi\\conf\\sqlmap\\ProUserSQL.xml";
        String filePath = "D:\\MyEclipse 8.5\\ssi\\src\\com\\wei\\ssi\\conf\\struts2\\struts2-ssi-proWbType.xml";
        try {
            Map map = new HashMap();
            map.put("file", "no");
            xml.getEditNode(filePath1, "/sqlMap/select[@id='getProUserList']", map, "");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}