Java实现XML格式化输出

这个是我在CSDN找到的一段XML格式化代码，但在XML包含版本编码的时候格式化第一个节点会走样，我稍微判断了一下。

 

public class XMLFormat {

 

    //格式化XML字符串

    public static String format(String originStr) {

        String XMLInfo = null;

        StringBuilder sb = new StringBuilder();

 

        try {

            if (originStr != null) {

                if (originStr.trim().startsWith("<?")) {

                    //包含XML版本编码

                    XMLInfo = originStr.trim().substring(0, originStr.trim().indexOf("?>") + 2);

                    originStr = originStr.trim().substring(originStr.trim().indexOf("?>") + 2, originStr.trim().length());

 

                }

                char[] charArray = originStr.toCharArray();

                int openCount = 0;

                int closeCount = 0;

                int lastNode = -1;//0 <> 1 </>

                for (int i = 0; i < charArray.length; i++) {

 

                    // < open and 换行

                    if (charArray[i] == '<' && i != charArray.length - 1 && charArray[i + 1] != '/' && charArray[i + 1] != '!') {

                        lastNode = 0;

                        enterAndTabs(sb, openCount - closeCount);

                        sb.append(charArray[i]);

                        openCount++;

                        continue;

                    }

                    // </ close and 换行

                    if (charArray[i] == '<' && i != charArray.length - 1 && charArray[i + 1] == '/') {

                        // 0 表示 开-闭 ,不换行

                        if (lastNode != 0) {

                            enterAndTabs(sb, openCount - closeCount - 1);

                        }

                        sb.append(charArray[i]).append(charArray[++i]);

                        closeCount++;

                        lastNode = 1;

                        continue;

                    }

                    // 自闭合

                    if (charArray[i] == '>' && i > 0 && charArray[i - 1] == '/') {

                        sb.append(charArray[i]);

                        closeCount++;

                        continue;

                    }

                    //反着来了

                    sb.append(charArray[i]);

                }

            }

        } catch (Exception exception) {

 

        }

        sb.insert(0,XMLInfo);

        return sb.toString();

    }

 

    /**

     * 回车与制表符

     *

     * @param sb StringBuilder

     * @param tabCount 多少个制表符

     */

    private static void enterAndTabs(StringBuilder sb, int tabCount) {

            sb.append("\n");

        while (tabCount-- > 0) {

                sb.append(" ");

            }

        }

 

    }