admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 00:58:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

perf: optimize pref of SqlUtil.replaceSqlParams()

Browse Source
This commit is contained in:
开源海哥 2023-08-02 11:21:56 +08:00
parent 65d5c6a480
commit 46651fe8ff
2 changed files with 61 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

91
mybatis-flex-core/src/main/java/com/mybatisflex/core/util/SqlUtil.java
View File

@ -108,41 +108,70 @@ public class SqlUtil {
* @return 完整的 sql
*/
public static String replaceSqlParams(String sql, Object[] params) {
if (params != null && params.length > 0) {
for (Object value : params) {
// null
if (value == null) {
sql = sql.replaceFirst("\\?", "null");
if (params == null || params.length == 0) {
return sql;
}
StringBuilder sqlBuilder = new StringBuilder();
char quote = 0;
int index = 0;
for (int i = 0; i < sql.length(); ++i) {
char ch = sql.charAt(i);
if (ch == '\'') {
if (quote == 0) {
quote = ch;
} else if (quote == '\'') {
quote = 0;
}
// number
else if (value instanceof Number || value instanceof Boolean) {
sql = sql.replaceFirst("\\?", value.toString());
}
// array
else if (ClassUtil.isArray(value.getClass())) {
StringJoiner joiner = new StringJoiner(",");
for (int i = 0; i < Array.getLength(value); i++) {
joiner.add(String.valueOf(Array.get(value, i)));
}
sql = sql.replaceFirst("\\?", "[" + joiner + "]");
}
// other
else {
StringBuilder sb = new StringBuilder();
sb.append("'");
if (value instanceof Date) {
sb.append(DateUtil.toDateTimeString((Date) value));
} else if (value instanceof LocalDateTime) {
sb.append(DateUtil.toDateTimeString(DateUtil.toDate((LocalDateTime) value)));
} else {
sb.append(value);
}
sb.append("'");
sql = sql.replaceFirst("\\?", Matcher.quoteReplacement(sb.toString()));
} else if (ch == '"') {
if (quote == 0) {
quote = ch;
} else if (quote == '"') {
quote = 0;
}
}
if (quote == 0 && ch == '?' && index < params.length) {
sqlBuilder.append(getParamString(params, index++));
} else {
sqlBuilder.append(ch);
}
}
return sqlBuilder.toString();
}
private static String getParamString(Object[] params, int index) {
Object value = params[index];
if (value == null) {
return "null";
}
// number
else if (value instanceof Number || value instanceof Boolean) {
return value.toString();
}
// array
else if (ClassUtil.isArray(value.getClass())) {
StringJoiner joiner = new StringJoiner(",", "[", "]");
for (int i = 0; i < Array.getLength(value); i++) {
joiner.add(String.valueOf(Array.get(value, i)));
}
return joiner.toString();
}
// other
else {
StringBuilder sb = new StringBuilder();
sb.append("'");
if (value instanceof Date) {
sb.append(DateUtil.toDateTimeString((Date) value));
} else if (value instanceof LocalDateTime) {
sb.append(DateUtil.toDateTimeString(DateUtil.toDate((LocalDateTime) value)));
} else {
sb.append(value);
}
sb.append("'");
return Matcher.quoteReplacement(sb.toString());
}
return sql;
}

4
mybatis-flex-core/src/test/java/com/mybatisflex/coretest/AccountSqlTester.java
View File

@ -179,9 +179,7 @@ public class AccountSqlTester {
.where(ACCOUNT.ID.ge(100))
.and(column("aaa").in("michael", "aaa"));
IDialect dialect = new CommonsDialectImpl();
String sql = dialect.forSelectByQuery(queryWrapper);
System.out.println(sql);
System.out.println(queryWrapper.toSQL());
}