Spring源码学习-PropertyPlaceholderHelper

今天在看Spring 3.0.0.RELEASE的源码，发现PropertyPlaceholderHelper的一个bug
当时觉得奇怪，上网一搜，果然是个bug，不过早就有人发现了，且已经修复：
详见：[url]http://forum.spring.io/forum/spring-projects/container/88107-propertyplaceholderhelper-bug[/url]

PropertyPlaceholderHelper的另一个Bug（ SPR-5369）就没那么好理解了，后来还是在测试类PropertyPlaceholderHelperTests.java中找到例子，
一步一步跟踪，总算明白是怎么回事了。详见：[url]https://github.com/spring-projects/spring-framework/pull/42/files[/url]

下面是简化版的PropertyPlaceholderHelper，原理跟Spring的一样，加入了一些我自己的理解

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import org.springframework.util.StringUtils;

/**
 * @author lijinnan
 * @date:2014-4-25
 */
public class PropertyPlaceholderHelper {


    public static final String PREFIX = "${";
    public static final String SUFFIX = "}";

    private static String parseStringValue(String strVal, Set<String> set) {
        StringBuilder buf = new StringBuilder(strVal);
        int startIndex = buf.indexOf(PREFIX);

        //递归的出口：没有变量需要替换了
        if (startIndex < 0) {
            return buf.toString();
        }

        while (true) {
            int endIndex = findPlaceholderEndIndex(buf, startIndex);
            if (endIndex == -1) {
                break;
            }

            //找到了最外层的变量，例如"${x${abc}x}"里面的"x${abc}x"--显然，变量里还有变量；这就是为什么要用递归
            String placeHolder = buf.substring(startIndex + PREFIX.length(), endIndex);

            //确保后面移出的变量是我们放进去的变量（而不是解析过程中产生的、新的变量，见求解"${top}"的例子）
            String originPlaceHolder = placeHolder;

            //防止循环定义：例如strVal="${a}"，而valueMap=[a="${b}", b="${a}"]
            if (!set.add(originPlaceHolder)) {
                throw new IllegalArgumentException("circular placeholder");
            }

            //递归对"变量里的变量"求解，直到最里面的变量求解得出结果为止，再一层一层的向外层返回
            placeHolder = parseStringValue(placeHolder, set);

            //可能value里面还有变量，需要递归求解
            String val = parseStringValue(valueMap.get(placeHolder), set);

            if (val != null) {
                buf.replace(startIndex, endIndex + SUFFIX.length(), val);

                //继续向后替换变量："${a}xx${b}"，替换好${a}之后，继续替换${b}
                startIndex = buf.indexOf(PREFIX, startIndex + val.length());
            }
            set.remove(originPlaceHolder);  //注意这里
            //set.remove(placeHolder);  //bug!
        }

        return buf.toString();

    }

    /**
     * 查找与PREFIX配对的SUFFIX
     * 注意处理嵌套的情况：用within变量来记录
     * 例如${ab${cd}}，startIndex指向a，从a开始找，当找到"${"时，within为1；
     * 找到第一个"}"时，within减1，抵消前面的"${"；while循环继续，直到找到最后的"}"
     * 这有点像"利用栈来判断括号是否配对"
     */
    private static int findPlaceholderEndIndex(StringBuilder buf, int startIndex) {
        int within = 0;
        int index = startIndex + PREFIX.length();
        while (index < buf.length()) {

            //发现了一个嵌套的PREFIX
            if (StringUtils.substringMatch(buf, index, PREFIX)) {
                within++;
                index = index + PREFIX.length();

            //发现了一个嵌套的SUFFIX，因此抵消一个PREFIX：within减1
            } else if (StringUtils.substringMatch(buf, index, SUFFIX)) {
                if (within > 0){
                    within--;
                    index = index + SUFFIX.length();
                } else if (within == 0){
                    return index;
                }
            } else {
                index++;
            }
        }
        return -1;
    }

    //for test
    private static Map<String, String> valueMap = new HashMap<String, String>();
    static {
        valueMap.put("inner", "ar");
        valueMap.put("bar", "ljn");

        //求解"${top}"的过程中，第一次递归返回时，${differentiator}被替换为"first"，然后产生
        //了一个新的placeHolder="first.grandchild"，这个placeHolder用"actualValue"替换
        //然后移除变量，如果set.remove(placeHolder)的话，那移除的就是"first.grandchild"
        //而实际上，它应该移除的是"${differentiator}.grandchild"。这个bug难以发现
        valueMap.put("top", "${child}+${child}");
        valueMap.put("child", "${${differentiator}.grandchild}");
        valueMap.put("differentiator", "first");
        valueMap.put("first.grandchild", "actualValue");
    }

    public static void main(String[] args) {
        String s ="${top}";
        System.out.println(parseStringValue(s, new HashSet<String>())); //actualValue+actualValue

        s = "foo=${b${inner}}-${bar}-end";
        System.out.println(parseStringValue(s, new HashSet<String>())); //foo=ljn-ljn-end

    }

}