参照mybatis替换方法,替换@$
/**
* token解析处理函数
* @ClassName: TokenHandler
*/
@FunctionalInterface
public interface TokenHandler {
/**
* 解析参水返回替换后的参数
* @param params 用来替换的表达式
* @return 替换后的值
*/
String handleToken(String content);
}
/**
* @ClassName: TokenParserUtil
* @Description: 通用参数替换
*/
public class TokenParserUtil {
private final static String OPEN_STR = "@";
private final static String END_STR = "$";
/**
* map参数解析形式@{param}类型(如果参数中不存在抛出运行时异常)
* @param str 需要解析的字符串
* @param params 替换的参数
* @return
*/
public static String parserNoNull(String text, ParamBean paramBean) {
MapTokenHandler MapTokenHandler = new MapTokenHandler(paramBean, true);
return parse(text, OPEN_STR, END_STR, MapTokenHandler);
}
/**
* map参数解析形式@{param}类型(如果不存在则不替换)
* @param str 需要解析的字符串
* @param params 替换的参数
* @return
*/
public static String parser(String text, ParamBean paramBean) {
MapTokenHandler MapTokenHandler = new MapTokenHandler(paramBean);
return parse(text, OPEN_STR, END_STR, MapTokenHandler);
}
/**
* map参数解析形式@{param}类型
* @param str 需要解析的字符串
* @param params 替换的参数
* @return
*/
public static String parser(String text, Map<String, Object> params) {
MapTokenHandler MapTokenHandler = new MapTokenHandler(params);
return parse(text, OPEN_STR, END_STR, MapTokenHandler);
}
/**
* @Title: parse
* @Description: 公用解析方法 例 我是#{name} 替换为 我是小喵
* @param text 需要解析的文本
* @param openToken 开始参数 #{
* @param closeToken 结束参数 }
* @param handler 处理函数
* @return
*/
public static String parse(String text, String openToken, String closeToken, TokenHandler handler) {
if (text == null || text.isEmpty()) {
return "";
}
char[] src = text.toCharArray();
int offset = 0;
// search open token
int start = text.indexOf(openToken, offset);
if (start == -1) {
return text;
}
final StringBuilder builder = new StringBuilder();
StringBuilder expression = null;
while (start > -1) {
if (start > 0 && src[start - 1] == '\\') {
// this open token is escaped. remove the backslash and continue.
builder.append(src, offset, start - offset - 1).append(openToken);
offset = start + openToken.length();
} else {
// found open token. let's search close token.
if (expression == null) {
expression = new StringBuilder();
} else {
expression.setLength(0);
}
builder.append(src, offset, start - offset);
offset = start + openToken.length();
int end = text.indexOf(closeToken, offset);
while (end > -1) {
if (end > offset && src[end - 1] == '\\') {
// this close token is escaped. remove the backslash and continue.
expression.append(src, offset, end - offset - 1).append(closeToken);
offset = end + closeToken.length();
end = text.indexOf(closeToken, offset);
} else {
expression.append(src, offset, end - offset);
offset = end + closeToken.length();
break;
}
}
if (end == -1) {
// close token was not found.
builder.append(src, start, src.length - start);
offset = src.length;
} else {
builder.append(handler.handleToken(expression.toString()));
offset = end + closeToken.length();
}
}
start = text.indexOf(openToken, offset);
}
if (offset < src.length) {
builder.append(src, offset, src.length - offset);
}
return builder.toString();
}
/**
* @ClassName: MapTokenHandler
* @Description: map参数处理
*/
static class MapTokenHandler implements TokenHandler {
private Map<String, Object> params;
/**是否必须存在*/
private boolean exists = false;
public MapTokenHandler(Map<String, Object> params) {
this.params = params;
}
public MapTokenHandler(Map<String, Object> params, boolean exists) {
this.params = params;
this.exists = exists;
}
@Override
public String handleToken(String content) {
Object obj = this.params.get(content);
if (obj == null) {
if (exists) {
throw new RuntimeException("无法在参数中获得表达式 @{" + OPEN_STR + content + END_STR + "所代表的变量");
}
return OPEN_STR + content + END_STR;
}
return obj.toString();
}
}
}