mirror of
https://gitee.com/mybatis-flex/mybatis-flex.git
synced 2025-12-07 17:18:24 +08:00
feat: add sqlServer 2005 LimitOffsetProcessor
This commit is contained in:
开源海哥
parent
99729bd8ba
commit
d020fe26b4
@ -17,6 +17,7 @@ package com.mybatisflex.core.dialect;
|
||||
|
||||
import com.mybatisflex.core.query.CPI;
|
||||
import com.mybatisflex.core.query.QueryOrderBy;
|
||||
import com.mybatisflex.core.query.QueryTable;
|
||||
import com.mybatisflex.core.query.QueryWrapper;
|
||||
import com.mybatisflex.core.util.CollectionUtil;
|
||||
|
||||
@ -29,152 +30,172 @@ import static com.mybatisflex.core.constant.SqlConsts.*;
|
||||
*/
|
||||
public interface LimitOffsetProcessor {
|
||||
|
||||
/**
|
||||
* MySql 的处理器
|
||||
* 适合 {@link DbType#MYSQL,DbType#MARIADB,DbType#H2,DbType#CLICK_HOUSE,DbType#XCloud}
|
||||
*/
|
||||
LimitOffsetProcessor MYSQL = (sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
sql.append(LIMIT).append(limitOffset).append(DELIMITER).append(limitRows);
|
||||
} else if (limitRows != null) {
|
||||
sql.append(LIMIT).append(limitRows);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* Postgresql 的处理器
|
||||
* 适合 {@link DbType#POSTGRE_SQL,DbType#SQLITE,DbType#H2,DbType#HSQL,DbType#KINGBASE_ES,DbType#PHOENIX}
|
||||
* 适合 {@link DbType#SAP_HANA,DbType#IMPALA,DbType#HIGH_GO,DbType#VERTICA,DbType#REDSHIFT}
|
||||
* 适合 {@link DbType#OPENGAUSS,DbType#TDENGINE,DbType#UXDB}
|
||||
*/
|
||||
LimitOffsetProcessor POSTGRESQL = (sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
sql.append(LIMIT).append(limitRows).append(OFFSET).append(limitOffset);
|
||||
} else if (limitRows != null) {
|
||||
sql.append(LIMIT).append(limitRows);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* derby 的处理器
|
||||
* 适合 {@link DbType#DERBY,DbType#ORACLE_12C,DbType#SQLSERVER ,DbType#POSTGRE_SQL}
|
||||
*/
|
||||
LimitOffsetProcessor DERBY = (sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
// OFFSET ** ROWS FETCH NEXT ** ROWS ONLY")
|
||||
sql.append(OFFSET).append(limitOffset).append(ROWS_FETCH_NEXT).append(limitRows).append(ROWS_ONLY);
|
||||
} else if (limitRows != null) {
|
||||
sql.append(OFFSET).append(0).append(ROWS_FETCH_NEXT).append(limitRows).append(ROWS_ONLY);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* derby 的处理器
|
||||
* 适合 {@link DbType#DERBY,DbType#ORACLE_12C,DbType#SQLSERVER ,DbType#POSTGRE_SQL}
|
||||
*/
|
||||
LimitOffsetProcessor SQLSERVER = (sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
// OFFSET ** ROWS FETCH NEXT ** ROWS ONLY")
|
||||
sql.append(OFFSET).append(limitOffset).append(ROWS_FETCH_NEXT).append(limitRows).append(ROWS_ONLY);
|
||||
} else if (limitRows != null) {
|
||||
List<QueryOrderBy> orderBys = CPI.getOrderBys(queryWrapper);
|
||||
if (CollectionUtil.isNotEmpty(orderBys)) {
|
||||
sql.append(OFFSET).append(0).append(ROWS_FETCH_NEXT).append(limitRows).append(ROWS_ONLY);
|
||||
} else {
|
||||
sql.insert(6, TOP + limitRows);
|
||||
}
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
/**
|
||||
* 处理构建 limit 和 offset
|
||||
*
|
||||
* @param dialect 数据方言
|
||||
* @param sql 已经构建的 sql
|
||||
* @param queryWrapper 参数内容
|
||||
* @param limitRows 用户传入的 limit 参数 可能为 null
|
||||
* @param limitOffset 用户传入的 offset 参数，可能为 null
|
||||
*/
|
||||
StringBuilder process(IDialect dialect, StringBuilder sql, QueryWrapper queryWrapper, Integer limitRows, Integer limitOffset);
|
||||
|
||||
|
||||
/**
|
||||
* SqlServer 2005 limit 处理器
|
||||
*/
|
||||
LimitOffsetProcessor SQLSERVER_2005 = (sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null) {
|
||||
if (limitOffset == null) {
|
||||
limitOffset = 0;
|
||||
}
|
||||
StringBuilder newSql = new StringBuilder("SELECT * FROM (SELECT TEMP_DATAS.*, ROW_NUMBER() OVER() AS RN FROM (");
|
||||
newSql.append(sql).append(") AS TEMP_DATAS) WHERE RN > ").append(limitOffset).append(" AND RN <= ").append(limitRows + limitRows);
|
||||
return newSql;
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
/**
|
||||
* MySql 的处理器
|
||||
* 适合 {@link DbType#MYSQL,DbType#MARIADB,DbType#H2,DbType#CLICK_HOUSE,DbType#XCloud}
|
||||
*/
|
||||
LimitOffsetProcessor MYSQL = (dialect, sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
sql.append(LIMIT).append(limitOffset).append(DELIMITER).append(limitRows);
|
||||
} else if (limitRows != null) {
|
||||
sql.append(LIMIT).append(limitRows);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* Postgresql 的处理器
|
||||
* 适合 {@link DbType#POSTGRE_SQL,DbType#SQLITE,DbType#H2,DbType#HSQL,DbType#KINGBASE_ES,DbType#PHOENIX}
|
||||
* 适合 {@link DbType#SAP_HANA,DbType#IMPALA,DbType#HIGH_GO,DbType#VERTICA,DbType#REDSHIFT}
|
||||
* 适合 {@link DbType#OPENGAUSS,DbType#TDENGINE,DbType#UXDB}
|
||||
*/
|
||||
LimitOffsetProcessor POSTGRESQL = (dialect, sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
sql.append(LIMIT).append(limitRows).append(OFFSET).append(limitOffset);
|
||||
} else if (limitRows != null) {
|
||||
sql.append(LIMIT).append(limitRows);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* derby 的处理器
|
||||
* 适合 {@link DbType#DERBY,DbType#ORACLE_12C,DbType#SQLSERVER ,DbType#POSTGRE_SQL}
|
||||
*/
|
||||
LimitOffsetProcessor DERBY = (dialect, sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
// OFFSET ** ROWS FETCH NEXT ** ROWS ONLY")
|
||||
sql.append(OFFSET).append(limitOffset).append(ROWS_FETCH_NEXT).append(limitRows).append(ROWS_ONLY);
|
||||
} else if (limitRows != null) {
|
||||
sql.append(OFFSET).append(0).append(ROWS_FETCH_NEXT).append(limitRows).append(ROWS_ONLY);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* derby 的处理器
|
||||
* 适合 {@link DbType#DERBY,DbType#ORACLE_12C,DbType#SQLSERVER ,DbType#POSTGRE_SQL}
|
||||
*/
|
||||
LimitOffsetProcessor SQLSERVER = (dialect, sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
// OFFSET ** ROWS FETCH NEXT ** ROWS ONLY")
|
||||
sql.append(OFFSET).append(limitOffset).append(ROWS_FETCH_NEXT).append(limitRows).append(ROWS_ONLY);
|
||||
} else if (limitRows != null) {
|
||||
List<QueryOrderBy> orderBys = CPI.getOrderBys(queryWrapper);
|
||||
if (CollectionUtil.isNotEmpty(orderBys)) {
|
||||
sql.append(OFFSET).append(0).append(ROWS_FETCH_NEXT).append(limitRows).append(ROWS_ONLY);
|
||||
} else {
|
||||
sql.insert(6, TOP + limitRows);
|
||||
}
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
|
||||
/**
|
||||
* Informix 的处理器
|
||||
* 适合 {@link DbType#INFORMIX}
|
||||
* 文档 {@link <a href="https://www.ibm.com/docs/en/informix-servers/14.10?topic=clause-restricting-return-values-skip-limit-first-options">https://www.ibm.com/docs/en/informix-servers/14.10?topic=clause-restricting-return-values-skip-limit-first-options</a>}
|
||||
*/
|
||||
LimitOffsetProcessor INFORMIX = (sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
// SELECT SKIP 2 FIRST 1 * FROM
|
||||
sql.insert(6, SKIP + limitOffset + FIRST + limitRows);
|
||||
} else if (limitRows != null) {
|
||||
sql.insert(6, FIRST + limitRows);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* Firebird 的处理器
|
||||
* 适合 {@link DbType#FIREBIRD}
|
||||
*/
|
||||
LimitOffsetProcessor FIREBIRD = (sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
// ROWS 2 TO 3
|
||||
sql.append(ROWS).append(limitOffset).append(TO).append(limitOffset + limitRows);
|
||||
} else if (limitRows != null) {
|
||||
sql.insert(6, FIRST + limitRows);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* Oracle11g及以下数据库的处理器
|
||||
* 适合 {@link DbType#ORACLE,DbType#DM,DbType#GAUSS}
|
||||
*/
|
||||
LimitOffsetProcessor ORACLE = (sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null) {
|
||||
if (limitOffset == null) {
|
||||
limitOffset = 0;
|
||||
}
|
||||
StringBuilder newSql = new StringBuilder("SELECT * FROM (SELECT TEMP_DATAS.*, ROWNUM RN FROM (");
|
||||
newSql.append(sql);
|
||||
newSql.append(") TEMP_DATAS WHERE ROWNUM <= ").append(limitOffset + limitRows).append(") WHERE RN > ").append(limitOffset);
|
||||
return newSql;
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* Sybase 处理器
|
||||
* 适合 {@link DbType#SYBASE}
|
||||
*/
|
||||
LimitOffsetProcessor SYBASE = (sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
//SELECT TOP 1 START AT 3 * FROM
|
||||
sql.insert(6, TOP + limitRows + START_AT + (limitOffset + 1));
|
||||
} else if (limitRows != null) {
|
||||
sql.insert(6, TOP + limitRows);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
/**
|
||||
* SqlServer 2005 limit 处理器
|
||||
*/
|
||||
LimitOffsetProcessor SQLSERVER_2005 = (dialect, sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null) {
|
||||
if (limitOffset == null) {
|
||||
limitOffset = 0;
|
||||
}
|
||||
String orderByString;
|
||||
List<QueryOrderBy> orderBys = CPI.getOrderBys(queryWrapper);
|
||||
if (orderBys == null || orderBys.isEmpty()) {
|
||||
orderByString = "ORDER BY CURRENT_TIMESTAMP";
|
||||
} else {
|
||||
StringBuilder orderBySql = new StringBuilder(ORDER_BY);
|
||||
int index = 0;
|
||||
List<QueryTable> queryTables = CPI.getQueryTables(queryWrapper);
|
||||
for (QueryOrderBy orderBy : orderBys) {
|
||||
orderBySql.append(orderBy.toSql(queryTables, dialect));
|
||||
if (index != orderBys.size() - 1) {
|
||||
orderBySql.append(DELIMITER);
|
||||
}
|
||||
index++;
|
||||
}
|
||||
orderByString = orderBySql.toString();
|
||||
}
|
||||
StringBuilder newSql = new StringBuilder("SELECT * FROM (SELECT TEMP_DATAS.*, ROW_NUMBER() OVER(").append(orderByString).append(") AS RN FROM (");
|
||||
newSql.append(sql).append(") AS TEMP_DATAS) WHERE RN > ").append(limitOffset).append(" AND RN <= ").append(limitRows + limitRows);
|
||||
return newSql;
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
|
||||
/**
|
||||
* Informix 的处理器
|
||||
* 适合 {@link DbType#INFORMIX}
|
||||
* 文档 {@link <a href="https://www.ibm.com/docs/en/informix-servers/14.10?topic=clause-restricting-return-values-skip-limit-first-options">https://www.ibm.com/docs/en/informix-servers/14.10?topic=clause-restricting-return-values-skip-limit-first-options</a>}
|
||||
*/
|
||||
LimitOffsetProcessor INFORMIX = (dialect, sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
// SELECT SKIP 2 FIRST 1 * FROM
|
||||
sql.insert(6, SKIP + limitOffset + FIRST + limitRows);
|
||||
} else if (limitRows != null) {
|
||||
sql.insert(6, FIRST + limitRows);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* Firebird 的处理器
|
||||
* 适合 {@link DbType#FIREBIRD}
|
||||
*/
|
||||
LimitOffsetProcessor FIREBIRD = (dialect, sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
// ROWS 2 TO 3
|
||||
sql.append(ROWS).append(limitOffset).append(TO).append(limitOffset + limitRows);
|
||||
} else if (limitRows != null) {
|
||||
sql.insert(6, FIRST + limitRows);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* Oracle11g及以下数据库的处理器
|
||||
* 适合 {@link DbType#ORACLE,DbType#DM,DbType#GAUSS}
|
||||
*/
|
||||
LimitOffsetProcessor ORACLE = (dialect, sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null) {
|
||||
if (limitOffset == null) {
|
||||
limitOffset = 0;
|
||||
}
|
||||
StringBuilder newSql = new StringBuilder("SELECT * FROM (SELECT TEMP_DATAS.*, ROWNUM RN FROM (");
|
||||
newSql.append(sql);
|
||||
newSql.append(") TEMP_DATAS WHERE ROWNUM <= ").append(limitOffset + limitRows).append(") WHERE RN > ").append(limitOffset);
|
||||
return newSql;
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
/**
|
||||
* Sybase 处理器
|
||||
* 适合 {@link DbType#SYBASE}
|
||||
*/
|
||||
LimitOffsetProcessor SYBASE = (dialect, sql, queryWrapper, limitRows, limitOffset) -> {
|
||||
if (limitRows != null && limitOffset != null) {
|
||||
//SELECT TOP 1 START AT 3 * FROM
|
||||
sql.insert(6, TOP + limitRows + START_AT + (limitOffset + 1));
|
||||
} else if (limitRows != null) {
|
||||
sql.insert(6, TOP + limitRows);
|
||||
}
|
||||
return sql;
|
||||
};
|
||||
|
||||
|
||||
/**
|
||||
* 处理构建 limit 和 offset
|
||||
*
|
||||
* @param sql 已经构建的 sql
|
||||
* @param queryWrapper 参数内容
|
||||
* @param limitRows 用户传入的 limit 参数 可能为 null
|
||||
* @param limitOffset 用户传入的 offset 参数，可能为 null
|
||||
*/
|
||||
StringBuilder process(StringBuilder sql, QueryWrapper queryWrapper, Integer limitRows, Integer limitOffset);
|
||||
}
|
||||
|
||||
@ -128,8 +128,8 @@ public class CommonsDialectImpl implements IDialect {
|
||||
}
|
||||
sql.append(wrap(getRealTable(tableName)));
|
||||
sql.append(BLANK).append(BRACKET_LEFT)
|
||||
.append(fields)
|
||||
.append(BRACKET_RIGHT).append(BLANK);
|
||||
.append(fields)
|
||||
.append(BRACKET_RIGHT).append(BLANK);
|
||||
sql.append(VALUES).append(questions);
|
||||
return sql.toString();
|
||||
}
|
||||
@ -341,12 +341,12 @@ public class CommonsDialectImpl implements IDialect {
|
||||
|
||||
//用户未配置别名的情况下，自动未用户添加别名
|
||||
if (selectColumnTable != null
|
||||
&& selectColumnName != null
|
||||
&& !"*".equals(selectColumnName)
|
||||
&& StringUtil.isBlank(selectColumn.getAlias())
|
||||
&& !(selectColumnTable instanceof SelectQueryTable)
|
||||
&& !CPI.isSameTable(firstTable, selectColumnTable)
|
||||
&& ArrayUtil.contains(firstTableColumns, selectColumnName)
|
||||
&& selectColumnName != null
|
||||
&& !"*".equals(selectColumnName)
|
||||
&& StringUtil.isBlank(selectColumn.getAlias())
|
||||
&& !(selectColumnTable instanceof SelectQueryTable)
|
||||
&& !CPI.isSameTable(firstTable, selectColumnTable)
|
||||
&& ArrayUtil.contains(firstTableColumns, selectColumnName)
|
||||
) {
|
||||
QueryColumn newSelectColumn = selectColumn.as(selectColumnTable.getName() + "$" + selectColumnName);
|
||||
selectColumns.set(i, newSelectColumn);
|
||||
@ -514,9 +514,9 @@ public class CommonsDialectImpl implements IDialect {
|
||||
}
|
||||
|
||||
return sql.append(BRACKET_LEFT).append(sqlFields).append(BRACKET_RIGHT)
|
||||
.append(VALUES)
|
||||
.append(BRACKET_LEFT).append(sqlValues).append(BRACKET_RIGHT)
|
||||
.toString();
|
||||
.append(VALUES)
|
||||
.append(BRACKET_LEFT).append(sqlValues).append(BRACKET_RIGHT)
|
||||
.toString();
|
||||
}
|
||||
|
||||
|
||||
@ -542,9 +542,9 @@ public class CommonsDialectImpl implements IDialect {
|
||||
}
|
||||
|
||||
return sql.append(BRACKET_LEFT).append(sqlFields).append(BRACKET_RIGHT)
|
||||
.append(VALUES)
|
||||
.append(BRACKET_LEFT).append(sqlValues).append(BRACKET_RIGHT)
|
||||
.toString();
|
||||
.append(VALUES)
|
||||
.append(BRACKET_LEFT).append(sqlValues).append(BRACKET_RIGHT)
|
||||
.toString();
|
||||
}
|
||||
|
||||
|
||||
@ -558,8 +558,8 @@ public class CommonsDialectImpl implements IDialect {
|
||||
warpedInsertColumns[i] = wrap(insertColumns[i]);
|
||||
}
|
||||
sql.append(BRACKET_LEFT)
|
||||
.append(StringUtil.join(DELIMITER, warpedInsertColumns))
|
||||
.append(BRACKET_RIGHT);
|
||||
.append(StringUtil.join(DELIMITER, warpedInsertColumns))
|
||||
.append(BRACKET_RIGHT);
|
||||
sql.append(VALUES);
|
||||
|
||||
Map<String, String> onInsertColumns = tableInfo.getOnInsertColumns();
|
||||
@ -782,7 +782,7 @@ public class CommonsDialectImpl implements IDialect {
|
||||
Map<String, RawValue> rawValueMap = tableInfo.obtainUpdateRawValueMap(entity);
|
||||
|
||||
sql.append(UPDATE).append(forHint(CPI.getHint(queryWrapper)))
|
||||
.append(tableInfo.getWrapSchemaAndTableName(this)).append(SET);
|
||||
.append(tableInfo.getWrapSchemaAndTableName(this)).append(SET);
|
||||
|
||||
StringJoiner stringJoiner = new StringJoiner(DELIMITER);
|
||||
|
||||
@ -1051,7 +1051,7 @@ public class CommonsDialectImpl implements IDialect {
|
||||
* 构建 limit 和 offset 的参数
|
||||
*/
|
||||
protected StringBuilder buildLimitOffsetSql(StringBuilder sqlBuilder, QueryWrapper queryWrapper, Integer limitRows, Integer limitOffset) {
|
||||
return limitOffsetProcessor.process(sqlBuilder, queryWrapper, limitRows, limitOffset);
|
||||
return limitOffsetProcessor.process(this, sqlBuilder, queryWrapper, limitRows, limitOffset);
|
||||
}
|
||||
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user