lsm/hutool
1
0
Fork 0
mirror of https://gitee.com/dromara/hutool.git synced 2025-05-06 05:38:00 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix db bug

Browse Source
This commit is contained in:
Looly 2021-04-13 15:04:19 +08:00
parent 56e32846cd
commit 97e15f8152
4 changed files with 101 additions and 68 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
CHANGELOG.md
View File

@ -3,12 +3,13 @@
-------------------------------------------------------------------------------------------------------------
# 5.6.3 (2021-04-11)
# 5.6.4 (2021-04-13)
### 新特性
* 【core 】 DatePattern补充DateTimeFormatter（pr#308@Gitee）
### Bug修复
* 【db 】 修复SQL分页时未使用别名导致的错误，同时count时取消order by子句（issue#I3IJ8X@Gitee）
-------------------------------------------------------------------------------------------------------------

15
hutool-db/src/main/java/cn/hutool/db/SqlConnRunner.java
View File

@ -1,7 +1,9 @@
package cn.hutool.db;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.lang.Assert;
import cn.hutool.core.map.MapUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.db.dialect.Dialect;
import cn.hutool.db.dialect.DialectFactory;
import cn.hutool.db.handler.EntityListHandler;
@ -279,7 +281,7 @@ public class SqlConnRunner extends DialectRunner {
}
/**
* 获取查询结果总数，生成类似于 SELECT count(1) from (sql)
* 获取查询结果总数，生成类似于 SELECT count(1) from (sql) as _count
*
* @param conn 数据库连接对象
* @param selectSql 查询语句
@ -287,7 +289,16 @@ public class SqlConnRunner extends DialectRunner {
* @throws SQLException SQL异常
*/
public long count(Connection conn, CharSequence selectSql) throws SQLException {
SqlBuilder sqlBuilder = SqlBuilder.of(selectSql).insertPreFragment("SELECT count(1) from(").append(")");
Assert.notBlank(selectSql, "Select SQL must be not blank!");
final int orderByIndex = StrUtil.indexOfIgnoreCase(selectSql, " order by");
if(orderByIndex > 0){
selectSql = StrUtil.subPre(selectSql, orderByIndex);
}
SqlBuilder sqlBuilder = SqlBuilder.of(selectSql)
.insertPreFragment("SELECT count(1) from(")
// issue#I3IJ8X@Gitee，在子查询时需设置单独别名，此处为了防止和用户的表名冲突，使用自定义的较长别名
.append(") as _hutool_alias_count_");
return page(conn, sqlBuilder, null, new NumberHandler()).intValue();
}

69
hutool-db/src/main/java/cn/hutool/db/sql/SqlBuilder.java
View File

@ -21,12 +21,12 @@ import java.util.Map.Entry;
* 调用getParamValues()方法获得占位符对应的值
*
* @author Looly
*
*/
public class SqlBuilder implements Builder<String>{
public class SqlBuilder implements Builder<String> {
private static final long serialVersionUID = 1L;
// --------------------------------------------------------------- Static methods start
/**
* 创建SQL构建器
*
@ -48,41 +48,56 @@ public class SqlBuilder implements Builder<String>{
/**
* 从已有的SQL中构建一个SqlBuilder
*
* @param sql SQL语句
* @return SqlBuilder
* @since 5.5.3
*/
public static SqlBuilder of(CharSequence sql){
public static SqlBuilder of(CharSequence sql) {
return create().append(sql);
}
// --------------------------------------------------------------- Static methods end
// --------------------------------------------------------------- Enums start
/**
* SQL中多表关联用的关键字
*
* @author Looly
*
*/
public enum Join {
/** 如果表中有至少一个匹配，则返回行 */
/**
* 如果表中有至少一个匹配，则返回行
*/
INNER,
/** 即使右表中没有匹配，也从左表返回所有的行 */
/**
* 即使右表中没有匹配，也从左表返回所有的行
*/
LEFT,
/** 即使左表中没有匹配，也从右表返回所有的行 */
/**
* 即使左表中没有匹配，也从右表返回所有的行
*/
RIGHT,
/** 只要其中一个表中存在匹配，就返回行 */
/**
* 只要其中一个表中存在匹配，就返回行
*/
FULL
}
// --------------------------------------------------------------- Enums end
private final StringBuilder sql = new StringBuilder();
/** 字段列表（仅用于插入和更新） */
/**
* 字段列表（仅用于插入和更新）
*/
private final List<String> fields = new ArrayList<>();
/** 占位符对应的值列表 */
/**
* 占位符对应的值列表
*/
private final List<Object> paramValues = new ArrayList<>();
/** 包装器 */
/**
* 包装器
*/
private Wrapper wrapper;
// --------------------------------------------------------------- Constructor start
@ -110,7 +125,7 @@ public class SqlBuilder implements Builder<String>{
* 插入<br>
* 插入会忽略空的字段名及其对应值，但是对于有字段名对应值为{@code null}的情况不忽略
*
* @param entity 实体
* @param entity 实体
* @param dialectName 方言名，用于对特殊数据库特殊处理
* @return 自己
*/
@ -122,7 +137,7 @@ public class SqlBuilder implements Builder<String>{
* 插入<br>
* 插入会忽略空的字段名及其对应值，但是对于有字段名对应值为{@code null}的情况不忽略
*
* @param entity 实体
* @param entity 实体
* @param dialectName 方言名，用于对特殊数据库特殊处理
* @return 自己
* @since 5.5.3
@ -132,8 +147,7 @@ public class SqlBuilder implements Builder<String>{
validateEntity(entity);
if (null != wrapper) {
// 包装表名
// entity = wrapper.wrap(entity);
// 包装表名 entity = wrapper.wrap(entity);
entity.setTableName(wrapper.wrap(entity.getTableName()));
}
@ -169,7 +183,7 @@ public class SqlBuilder implements Builder<String>{
}
sql.append("INSERT INTO ")//
.append(entity.getTableName()).append(" (").append(fieldsPart).append(") VALUES (")//
.append(placeHolder.toString()).append(")");
.append(placeHolder).append(")");
return this;
}
@ -232,7 +246,7 @@ public class SqlBuilder implements Builder<String>{
* 查询
*
* @param isDistinct 是否添加DISTINCT关键字（查询唯一结果）
* @param fields 查询的字段
* @param fields 查询的字段
* @return 自己
*/
public SqlBuilder select(boolean isDistinct, String... fields) {
@ -243,7 +257,7 @@ public class SqlBuilder implements Builder<String>{
* 查询
*
* @param isDistinct 是否添加DISTINCT关键字（查询唯一结果）
* @param fields 查询的字段
* @param fields 查询的字段
* @return 自己
*/
public SqlBuilder select(boolean isDistinct, Collection<String> fields) {
@ -326,7 +340,7 @@ public class SqlBuilder implements Builder<String>{
* 只支持单一的逻辑运算符（例如多个条件之间）
*
* @param logicalOperator 逻辑运算符
* @param conditions 条件，当条件为空时，只添加WHERE关键字
* @param conditions 条件，当条件为空时，只添加WHERE关键字
* @return 自己
* @deprecated logicalOperator放在Condition中了，因此请使用 {@link #where(Condition...)}
*/
@ -351,8 +365,8 @@ public class SqlBuilder implements Builder<String>{
/**
* 多值选择
*
* @param <T> 值类型
* @param field 字段名
* @param <T> 值类型
* @param field 字段名
* @param values 值列表
* @return 自身
*/
@ -385,7 +399,7 @@ public class SqlBuilder implements Builder<String>{
* 添加Having语句
*
* @param logicalOperator 逻辑运算符
* @param conditions 条件
* @param conditions 条件
* @return 自己
* @deprecated logicalOperator放在Condition中了，因此请使用 {@link #having(Condition...)}
*/
@ -465,7 +479,7 @@ public class SqlBuilder implements Builder<String>{
* 多表关联
*
* @param tableName 被关联的表名
* @param join 内联方式
* @param join 内联方式
* @return 自己
*/
public SqlBuilder join(String tableName, Join join) {
@ -489,7 +503,7 @@ public class SqlBuilder implements Builder<String>{
* 只支持单一的逻辑运算符（例如多个条件之间）
*
* @param logicalOperator 逻辑运算符
* @param conditions 条件
* @param conditions 条件
* @return 自己
* @deprecated logicalOperator放在Condition中了，因此请使用 {@link #on(Condition...)}
*/
@ -548,7 +562,7 @@ public class SqlBuilder implements Builder<String>{
* SqlBuilder builder = SqlBuilder.of("select *");
* builder.append(" from ").append("user");
* </pre>
*
* <p>
* 如果需要追加带占位符的片段，需调用{@link #addParams(Object...)} 方法加入对应参数值。
*
* @param sqlFragment SQL其它部分片段
@ -574,8 +588,8 @@ public class SqlBuilder implements Builder<String>{
* @return this
* @since 5.5.3
*/
public SqlBuilder addParams(Object... params){
if(ArrayUtil.isNotEmpty(params)){
public SqlBuilder addParams(Object... params) {
if (ArrayUtil.isNotEmpty(params)) {
Collections.addAll(this.paramValues, params);
}
return this;
@ -644,6 +658,7 @@ public class SqlBuilder implements Builder<String>{
}
// --------------------------------------------------------------- private method start
/**
* 构建组合条件<br>
* 例如：name = ? AND type IN (?, ?) AND other LIKE ?

8
hutool-db/src/test/java/cn/hutool/db/DbTest.java
View File

@ -39,7 +39,7 @@ public class DbTest {
@Test
public void pageTest2() throws SQLException {
String sql = "select * from user";
String sql = "select * from user order by name";
// 测试数据库中一共4条数据，第0页有3条，第1页有1条
List<Entity> page0 = Db.use().page(
sql, Page.of(0, 3));
@ -56,6 +56,12 @@ public class DbTest {
Assert.assertEquals(4, count);
}
@Test
public void countTest2() throws SQLException {
final long count = Db.use().count("select * from user order by name DESC");
Assert.assertEquals(4, count);
}
@Test
public void findLikeTest() throws SQLException {
// 方式1