Fix https://gitee.com/mybatis-flex/mybatis-flex/issues/IBIJT3 [Bug]: SqlServer2005方言List查询SQL语句BUG

This commit is contained in:
F3235157 2025-01-20 17:34:01 +08:00
parent ab03a41aa4
commit 8b9585ba3c
2 changed files with 35 additions and 49 deletions

View File

@ -129,13 +129,19 @@ public interface LimitOffsetProcessor {
originalSQL = originalSQL.substring(0, sql.lastIndexOf(ORDER_BY)); originalSQL = originalSQL.substring(0, sql.lastIndexOf(ORDER_BY));
orderByString = orderBySql.toString(); orderByString = orderBySql.toString();
} }
//Fix https://gitee.com/mybatis-flex/mybatis-flex/issues/IBIJT3 [Bug]: SqlServer2005方言List查询SQL语句BUG
String columeSQL = originalSQL.substring(0, sql.indexOf(FROM));
String contitionSQL = originalSQL.substring(sql.indexOf(FROM));
StringBuilder newSql = new StringBuilder(); StringBuilder newSql = new StringBuilder();
//fix SqlServer 多表关联查询主表去重执行SQL异常 https://gitee.com/mybatis-flex/mybatis-flex/issues/IABEJG //fix SqlServer 多表关联查询主表去重执行SQL异常 https://gitee.com/mybatis-flex/mybatis-flex/issues/IABEJG
newSql.append("WITH temp_datas AS(SELECT __Tab.*, ROW_NUMBER() OVER ( ").append(orderByString) newSql.append("WITH temp_datas AS(").append(columeSQL).append(", ROW_NUMBER() OVER (")
.append(") as __rn ").append(" FROM ( ").append(originalSQL).append(" ) as __Tab ) "); .append(orderByString)
newSql.append(" SELECT * FROM temp_datas WHERE __rn BETWEEN ") .append(") as __rn ");
.append(limitOffset + 1).append(" AND ").append(limitOffset + limitRows); newSql.append(contitionSQL);
newSql.append(")");
newSql.append(" SELECT * FROM temp_datas WHERE __rn BETWEEN ").append(limitOffset + 1).append(" AND ")
.append(limitOffset + limitRows);
newSql.append(" ORDER BY __rn"); newSql.append(" ORDER BY __rn");
return newSql; return newSql;
} }
@ -193,9 +199,7 @@ public interface LimitOffsetProcessor {
} }
StringBuilder newSql = new StringBuilder("SELECT * FROM (SELECT TEMP_DATAS.*, ROWNUM RN FROM ("); StringBuilder newSql = new StringBuilder("SELECT * FROM (SELECT TEMP_DATAS.*, ROWNUM RN FROM (");
newSql.append(sql); newSql.append(sql);
newSql.append(") TEMP_DATAS WHERE ROWNUM <= ") newSql.append(") TEMP_DATAS WHERE ROWNUM <= ").append(limitOffset + limitRows).append(") WHERE RN > ")
.append(limitOffset + limitRows)
.append(") WHERE RN > ")
.append(limitOffset); .append(limitOffset);
return newSql; return newSql;
} }

View File

@ -16,6 +16,9 @@
package com.mybatisflex.coretest; package com.mybatisflex.coretest;
import static com.mybatisflex.coretest.table.AccountTableDef.ACCOUNT;
import static com.mybatisflex.coretest.table.ArticleTableDef.ARTICLE;
import com.mybatisflex.core.dialect.IDialect; import com.mybatisflex.core.dialect.IDialect;
import com.mybatisflex.core.dialect.KeywordWrap; import com.mybatisflex.core.dialect.KeywordWrap;
import com.mybatisflex.core.dialect.LimitOffsetProcessor; import com.mybatisflex.core.dialect.LimitOffsetProcessor;
@ -26,78 +29,62 @@ import com.mybatisflex.core.query.QueryWrapper;
import org.junit.Assert; import org.junit.Assert;
import org.junit.Test; import org.junit.Test;
import static com.mybatisflex.coretest.table.AccountTableDef.ACCOUNT;
public class SqlServer2005DialectTester { public class SqlServer2005DialectTester {
@Test @Test
public void testSelectSql() { public void testSelectSql() {
QueryWrapper query = new QueryWrapper() QueryWrapper query = new QueryWrapper().select().from(ACCOUNT).where(ACCOUNT.ID.in("100", "200"))
.select() .and(ACCOUNT.SEX.eq(1)).orderBy(ACCOUNT.ID.desc()).limit(10, 10);
.from(ACCOUNT)
.where(ACCOUNT.ID.in("100", "200"))
.and(ACCOUNT.SEX.eq(1))
.orderBy(ACCOUNT.ID.desc())
.limit(10, 10);
IDialect dialect = new CommonsDialectImpl(KeywordWrap.SQUARE_BRACKETS, LimitOffsetProcessor.SQLSERVER_2005); IDialect dialect = new CommonsDialectImpl(KeywordWrap.SQUARE_BRACKETS, LimitOffsetProcessor.SQLSERVER_2005);
String sql = dialect.forSelectByQuery(query); String sql = dialect.forSelectByQuery(query);
System.out.println(sql); System.out.println(sql);
Assert.assertEquals("WITH temp_datas AS(" Assert.assertEquals("WITH temp_datas AS("
+ "SELECT ROW_NUMBER() OVER ( ORDER BY [id] DESC) as __rn, * FROM [tb_account] WHERE [id] IN (?, ?) AND [sex] = ?" + "SELECT *, ROW_NUMBER() OVER ( ORDER BY [id] DESC) as __rn FROM [tb_account] WHERE [id] IN (?, ?) AND [sex] = ?"
+ + ") " + "SELECT * FROM temp_datas WHERE __rn BETWEEN 11 AND 20 ORDER BY __rn", sql);
") " +
"SELECT * FROM temp_datas WHERE __rn BETWEEN 11 AND 20 ORDER BY __rn", sql);
} }
@Test @Test
public void testSelectSqlSqlserver2005() { public void testSelectSqlSqlserver2005() {
QueryWrapper query = new QueryWrapper().select() QueryWrapper query = new QueryWrapper().select(ACCOUNT.ID.as("ID"), ACCOUNT.AGE.as("AGE"), ARTICLE.ID)
.from("TEST.dbo.tb_account") .from(ACCOUNT.as("account"))
.where(ACCOUNT.ID.in("100", "200")) .innerJoin(ARTICLE).on(ARTICLE.ACCOUNT_ID.eq(ACCOUNT.ID))
.and(ACCOUNT.SEX.eq(1)) .where(ACCOUNT.ID.in("100", "200")).and(ACCOUNT.SEX.eq(1))
.orderBy(ACCOUNT.ID.desc()) .orderBy(ACCOUNT.ID.as("ID").desc(), ACCOUNT.AGE.asc(), ARTICLE.ID.desc()).limit(10);
.limit(0, 10);
IDialect dialect = new Sqlserver2005DialectImpl(KeywordWrap.SQUARE_BRACKETS, IDialect dialect = new Sqlserver2005DialectImpl(KeywordWrap.SQUARE_BRACKETS,
LimitOffsetProcessor.SQLSERVER_2005); LimitOffsetProcessor.SQLSERVER_2005);
String sql = dialect.forSelectByQuery(query); String sql = dialect.forSelectByQuery(query);
System.out.println(sql); System.out.println(sql);
Assert.assertEquals( Assert.assertEquals(
"WITH temp_datas AS(SELECT ROW_NUMBER() OVER ( ORDER BY [tb_account].[id] DESC) as __rn, * FROM [TEST].[dbo].[tb_account] WHERE [tb_account].[id] IN (?, ?) AND [tb_account].[sex] = ?) SELECT * FROM temp_datas WHERE __rn BETWEEN 1 AND 10 ORDER BY __rn", "WITH temp_datas AS(SELECT [account].[id] AS [ID], [account].[age] AS [AGE], [tb_article].[id], ROW_NUMBER() OVER ( ORDER BY [account].[id] DESC, [account].[age] ASC, [tb_article].[id] DESC) as __rn FROM [tb_account] AS [account] INNER JOIN [tb_article] ON [tb_article].[account_id] = [account].[id] WHERE [account].[id] IN (?, ?) AND [account].[sex] = ?) SELECT * FROM temp_datas WHERE __rn BETWEEN 1 AND 10 ORDER BY __rn",
sql); sql);
} }
@Test @Test
public void testSelectSqlSqlserver2005With() { public void testSelectSqlSqlserver2005With() {
QueryWrapper query = new QueryWrapper().select(ACCOUNT.ID.as("user_id"), ACCOUNT.AGE) QueryWrapper query = new QueryWrapper()
.select(ACCOUNT.ID.as("user_id"), ACCOUNT.AGE)
.from("TEST.dbo.tb_account") .from("TEST.dbo.tb_account")
.where(ACCOUNT.ID.in("100", "200")) .innerJoin(ARTICLE).on(ARTICLE.ACCOUNT_ID.eq(ACCOUNT.ID)).where(ACCOUNT.ID.in("100", "200"))
.and(ACCOUNT.SEX.eq(1)) .and(ACCOUNT.SEX.eq(1))
.orderBy(ACCOUNT.ID.desc()) .orderBy(ACCOUNT.ID.desc()).limit(0, 10);
.limit(0, 10);
IDialect dialect = new Sqlserver2005DialectImpl(KeywordWrap.SQUARE_BRACKETS, IDialect dialect = new Sqlserver2005DialectImpl(KeywordWrap.SQUARE_BRACKETS,
LimitOffsetProcessor.SQLSERVER_2005); LimitOffsetProcessor.SQLSERVER_2005);
String sql = dialect.forSelectByQuery(query); String sql = dialect.forSelectByQuery(query);
System.out.println(sql); System.out.println(sql);
Assert.assertEquals( Assert.assertEquals(
"WITH temp_datas AS(SELECT ROW_NUMBER() OVER ( ORDER BY [tb_account].[id] DESC) as __rn, [tb_account].[id] AS [user_id], [tb_account].[age] FROM [TEST].[dbo].[tb_account] WHERE [tb_account].[id] IN (?, ?) AND [tb_account].[sex] = ?) SELECT user_id, age FROM temp_datas WHERE __rn BETWEEN 1 AND 10 ORDER BY __rn", "WITH temp_datas AS(SELECT [tb_account].[id] AS [user_id], [tb_account].[age], ROW_NUMBER() OVER ( ORDER BY [tb_account].[id] DESC) as __rn FROM [TEST].[dbo].[tb_account] INNER JOIN [tb_article] ON [tb_article].[account_id] = [tb_account].[id] WHERE [tb_account].[id] IN (?, ?) AND [tb_account].[sex] = ?) SELECT * FROM temp_datas WHERE __rn BETWEEN 1 AND 10 ORDER BY __rn",
sql); sql);
} }
@Test @Test
public void testSelectSqlSqlserver() { public void testSelectSqlSqlserver() {
QueryWrapper query = new QueryWrapper().select() QueryWrapper query = new QueryWrapper().select().from("TEST.dbo.tb_account").where(ACCOUNT.ID.in("100", "200"))
.from("TEST.dbo.tb_account") .and(ACCOUNT.SEX.eq(1)).orderBy(ACCOUNT.ID.desc()).limit(0, 10);
.where(ACCOUNT.ID.in("100", "200"))
.and(ACCOUNT.SEX.eq(1))
.orderBy(ACCOUNT.ID.desc())
.limit(0, 10);
IDialect dialect = new SqlserverDialectImpl(KeywordWrap.SQUARE_BRACKETS, IDialect dialect = new SqlserverDialectImpl(KeywordWrap.SQUARE_BRACKETS, LimitOffsetProcessor.SQLSERVER);
LimitOffsetProcessor.SQLSERVER);
String sql = dialect.forSelectByQuery(query); String sql = dialect.forSelectByQuery(query);
System.out.println(sql); System.out.println(sql);
Assert.assertEquals( Assert.assertEquals(
@ -107,15 +94,10 @@ public class SqlServer2005DialectTester {
@Test @Test
public void testSelectSqlSqlserver1() { public void testSelectSqlSqlserver1() {
QueryWrapper query = new QueryWrapper().select() QueryWrapper query = new QueryWrapper().select().from("tb_account").where(ACCOUNT.ID.in("100", "200"))
.from("tb_account") .and(ACCOUNT.SEX.eq(1)).orderBy(ACCOUNT.ID.desc()).limit(0, 10);
.where(ACCOUNT.ID.in("100", "200"))
.and(ACCOUNT.SEX.eq(1))
.orderBy(ACCOUNT.ID.desc())
.limit(0, 10);
IDialect dialect = new SqlserverDialectImpl(KeywordWrap.SQUARE_BRACKETS, IDialect dialect = new SqlserverDialectImpl(KeywordWrap.SQUARE_BRACKETS, LimitOffsetProcessor.SQLSERVER);
LimitOffsetProcessor.SQLSERVER);
String sql = dialect.forSelectByQuery(query); String sql = dialect.forSelectByQuery(query);
System.out.println(sql); System.out.println(sql);
Assert.assertEquals( Assert.assertEquals(