refactor(StringUtils.java):重构 SQL 替换方法不在重复调用 String.replace ， 改为一次扫描

mybatis-plus-core/src/main/java/com/baomidou/mybatisplus/core/toolkit/StringUtils.java

@@ -15,7 +15,8 @@
  */
 package com.baomidou.mybatisplus.core.toolkit;
 
-import static java.util.stream.Collectors.joining;
+import com.baomidou.mybatisplus.core.toolkit.sql.StringEscape;
+import com.baomidou.mybatisplus.core.toolkit.support.BiIntFunction;
 
 import java.nio.charset.StandardCharsets;
 import java.sql.Blob;
@@ -23,9 +24,10 @@ import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.List;
+import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
-import com.baomidou.mybatisplus.core.toolkit.sql.StringEscape;
+import static java.util.stream.Collectors.joining;
 
 /**
  * <p>
@@ -255,29 +257,55 @@ public class StringUtils {
     }
 
     /**
+     * MP 内定义的 SQL 占位符表达式，匹配诸如 {0},{1},{2} ... 的形式
+     */
+    public static Pattern MP_SQL_PLACE_HOLDER = Pattern.compile("[{](?<idx>\\d)}");
+
+    /**
+     * 替换 SQL 语句中的占位符，例如输入 SELECT * FROM test WHERE id = {0} AND name = {1} 会被替换为
+     * SELECT * FROM test WHERE id = 1 AND name = 'MP'
      * <p>
-     * SQL 参数填充
-     * </p>
+     * 当数组中参数不足时，该方法会抛出错误：数组下标越界{@link ArrayIndexOutOfBoundsException}
      *
      * @param content 填充内容
      * @param args    填充参数
-     * @return
      */
     public static String sqlArgsFill(String content, Object... args) {
-        if (StringUtils.isEmpty(content)) {
-            return null;
-        }
-        if (args != null) {
-            int length = args.length;
-            if (length != 0) {
-                for (int i = 0; i < length; i++) {
-                    content = content.replace(String.format(PLACE_HOLDER, i), sqlParam(args[i]));
-                }
-            }
+        if (StringUtils.isNotEmpty(content) && ArrayUtils.isNotEmpty(args)) {
+            // 索引不能使用，因为 SQL 中的占位符数字与索引不相同
+            return replace(content, MP_SQL_PLACE_HOLDER, (m, i) -> sqlParam(args[Integer.parseInt(m.group("idx"))])).toString();
         }
         return content;
     }
 
+    /**
+     * 根据指定的表达式替换字符串中指定格式的部分
+     * <p>
+     * BiIntFunction 中的 第二个 参数将传递 参数在字符串中的索引
+     *
+     * @param src      源字符串
+     * @param ptn      需要替换部分的正则表达式
+     * @param replacer 替换处理器
+     * @return 返回字符串构建起
+     */
+    public static StringBuilder replace(CharSequence src, Pattern ptn, BiIntFunction<Matcher, CharSequence> replacer) {
+        int idx = 0, last = 0, len = src.length();
+        Matcher m = ptn.matcher(src);
+        StringBuilder sb = new StringBuilder();
+
+        // 扫描一次字符串
+        while (m.find()) {
+            sb.append(src, last, m.start()).append(replacer.apply(m, idx++));
+            last = m.end();
+        }
+        // 如果表达式没有匹配或者匹配未到末尾，该判断保证字符串完整性
+        if (last < len) {
+            sb.append(src, last, len);
+        }
+
+        return sb;
+    }
+
     /**
      * <p>
      * 获取SQL PARAMS字符串

mybatis-plus-core/src/main/java/com/baomidou/mybatisplus/core/toolkit/support/BiIntFunction.java

@@ -0,0 +1,21 @@
+package com.baomidou.mybatisplus.core.toolkit.support;
+
+/**
+ * 接受 Int 小类型的处理函数，使用小类型来避免 Java 自动装箱
+ *
+ * @author HCL
+ * Create at 2018/11/19
+ */
+@FunctionalInterface
+public interface BiIntFunction<T, R> {
+
+    /**
+     * 函数主接口
+     *
+     * @param t 被执行类型 T
+     * @param i 参数
+     * @return 返回
+     */
+    R apply(T t, int i);
+
+}