java - 替换占位符

像模版引擎一样,替换字符串中的 ${} 占位符,好处就是使用原生的API,不需要额外导包了。

public class A{

  private static final String BRACE_LEFT = "{";
    private static final String BRACE_RIGHT = "}";
    private static final String $ = "$";

    /**
     * 替换字符,占位符限定 ${}
     *
     * @param tmp    代码模版
     * @param params 参数
     * @return PreparedSql
     */
    public static String processMap(@NotNull String tmp, Map<String, ?> params) {
        StringTokenizer tokenizer = new StringTokenizer(tmp, "${}", true);
        StringBuilder res = new StringBuilder(tmp.length());
        String token;
        boolean escaped = false;
        while (tokenizer.hasMoreTokens()) {
            token = tokenizer.nextToken();
            if (escaped) {
                if ($.equals(token) || BRACE_LEFT.equals(token)) {
                    //占位符中出现美元符或者花括号
                    throw new IllegalArgumentException("Contain invalid char: " + token);
                } else if (BRACE_RIGHT.equals(token)) {
                    //标志复位
                    escaped = false;
                } else {
                    //参数名占位符
                    res.append(Converter.toString(params.get(token)));
                }
            } else {
                if ($.equals(token)) {
                    token = tokenizer.nextToken();
                    if (BRACE_LEFT.equals(token)) {
                        //检索到 '${' 开始尝试获取参数名
                        escaped = true;
                    } else {
                        //'$'独立出现的场景
                        res.append($).append(token);
                    }
                } else {
                    //普通字符
                    res.append(token);
                }
            }
        }
        return res.toString();
    }
}

posted on 2023-06-08 15:52  疯狂的妞妞  阅读(141)  评论(0编辑  收藏  举报

导航