ValueEscaper.java

package org.nutz.dao.impl.sql;

import org.nutz.lang.util.LinkedArray;
import org.nutz.lang.util.LinkedCharArray;

/**
 * 通过 add 函数，可以增加需要逃逸的字符，以及如何逃逸
 * <p>
 * 然后调用 escape 函数，执行逃逸
 * 
 * @author zozoh(zozohtnt@gmail.com)
 * 
 */
public class ValueEscaper {

    public ValueEscaper() {
        this.charStack = new LinkedCharArray(20);
        this.escapeStack = new LinkedArray<String>(20);
    }

    private LinkedCharArray charStack;
    private LinkedArray<String> escapeStack;
    private char[] chars;
    private String[] escapes;

    /**
     * 设定逃逸方式
     * 
     * @param c
     *            要逃逸的字符
     * @param s
     *            如何逃逸
     * @return 自身
     */
    public ValueEscaper add(char c, String s) {
        charStack.push(c);
        escapeStack.push(s);
        return this;
    }

    /**
     * 在执行 escape 前，这个函数一定要调用，它会把你增加的逃逸设置，初始化一下
     * 
     * @return 自身
     */
    public ValueEscaper ready() {
        chars = charStack.toArray();
        escapes = escapeStack.toArray();
        return this;
    }

    /**
     * 根据逃逸的设置，对传入的字符串进行逃逸
     * 
     * @param cs
     *            字符序列
     * @return 逃逸后的字符序列
     */
    public CharSequence escape(CharSequence cs) {
        StringBuilder sb = new StringBuilder();
        boolean find;
        for (int i = 0; i < cs.length(); i++) {
            char c = cs.charAt(i);
            find = false;
            for (int x = 0; x < chars.length; x++) {
                if (c == chars[x]) {
                    sb.append(escapes[x]);
                    find = true;
                    break;
                }
            }
            if (!find)
                sb.append(c);
        }
        return sb;
    }

}