Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
apache
Shardingsphere
提交
c33a09b7
Shardingsphere
项目概览
apache
/
Shardingsphere
通知
56
Star
3
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Shardingsphere
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
c33a09b7
编写于
5月 12, 2017
作者:
T
terrymanu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add limit to rewrite module
上级
fb77124c
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
58 addition
and
19 deletion
+58
-19
sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/rewrite/LimitUtils.java
...com/dangdang/ddframe/rdb/sharding/rewrite/LimitUtils.java
+49
-0
sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/rewrite/SQLRewriteEngine.java
...ngdang/ddframe/rdb/sharding/rewrite/SQLRewriteEngine.java
+5
-1
sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/router/PreparedSQLRouter.java
...ngdang/ddframe/rdb/sharding/router/PreparedSQLRouter.java
+0
-18
sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/router/SQLRouteEngine.java
.../dangdang/ddframe/rdb/sharding/router/SQLRouteEngine.java
+4
-0
未找到文件。
sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/rewrite/LimitUtils.java
0 → 100644
浏览文件 @
c33a09b7
/*
* Copyright 1999-2015 dangdang.com.
* <p>
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* </p>
*/
package
com.dangdang.ddframe.rdb.sharding.rewrite
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.context.LimitContext
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.context.SQLContext
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.exception.SQLParsingException
;
import
java.util.List
;
/**
* 补列工具类.
*
* @author zhangliang
*/
public
final
class
LimitUtils
{
/**
* 追加分页.
*
* @param sqlContext SQL上下文
* @param parameters 参数
*/
public
static
void
appendLimit
(
final
SQLContext
sqlContext
,
final
List
<
Object
>
parameters
)
{
int
offset
=
-
1
==
sqlContext
.
getLimitContext
().
getOffsetParameterIndex
()
?
sqlContext
.
getLimitContext
().
getOffset
()
:
(
int
)
parameters
.
get
(
sqlContext
.
getLimitContext
().
getOffsetParameterIndex
());
int
rowCount
=
-
1
==
sqlContext
.
getLimitContext
().
getRowCountParameterIndex
()
?
sqlContext
.
getLimitContext
().
getRowCount
()
:
(
int
)
parameters
.
get
(
sqlContext
.
getLimitContext
().
getRowCountParameterIndex
());
sqlContext
.
setLimitContext
(
new
LimitContext
(
offset
,
rowCount
,
sqlContext
.
getLimitContext
().
getOffsetParameterIndex
(),
sqlContext
.
getLimitContext
().
getRowCountParameterIndex
()));
if
(
offset
<
0
||
rowCount
<
0
)
{
throw
new
SQLParsingException
(
"LIMIT offset and row count can not be a negative value."
);
}
}
}
sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/rewrite/SQLRewriteEngine.java
浏览文件 @
c33a09b7
...
...
@@ -18,6 +18,7 @@
package
com.dangdang.ddframe.rdb.sharding.rewrite
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.context.LimitContext
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.context.SQLContext
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.context.TableContext
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.token.ItemsToken
;
...
...
@@ -46,6 +47,8 @@ public final class SQLRewriteEngine {
private
final
Collection
<
String
>
tableNames
=
new
TreeSet
<>(
String
.
CASE_INSENSITIVE_ORDER
);
private
final
LimitContext
limit
;
private
SQLBuilder
sqlBuilder
;
public
SQLRewriteEngine
(
final
String
originalSQL
,
final
SQLContext
sqlContext
)
{
...
...
@@ -54,10 +57,11 @@ public final class SQLRewriteEngine {
for
(
TableContext
each
:
sqlContext
.
getTables
())
{
tableNames
.
add
(
each
.
getName
());
}
limit
=
sqlContext
.
getLimitContext
();
}
/**
* SQL
重
写.
* SQL
改
写.
*
* @return SQL构建器
*/
...
...
sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/router/PreparedSQLRouter.java
浏览文件 @
c33a09b7
...
...
@@ -21,9 +21,7 @@ import com.dangdang.ddframe.rdb.sharding.api.rule.ShardingRule;
import
com.dangdang.ddframe.rdb.sharding.api.rule.TableRule
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.context.GeneratedKeyContext
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.context.InsertSQLContext
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.context.LimitContext
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.context.SQLContext
;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.exception.SQLParsingException
;
import
com.google.common.base.Optional
;
import
lombok.RequiredArgsConstructor
;
...
...
@@ -61,25 +59,9 @@ public class PreparedSQLRouter {
List
<
Number
>
generatedIds
=
generateId
();
parameters
.
addAll
(
generatedIds
);
}
// TODO 提炼至rewrite模块
setLimit
(
parameters
);
return
engine
.
routeSQL
(
logicSql
,
sqlContext
,
parameters
);
}
private
void
setLimit
(
final
List
<
Object
>
parameters
)
{
if
(
null
==
sqlContext
.
getLimitContext
())
{
return
;
}
int
offset
=
-
1
==
sqlContext
.
getLimitContext
().
getOffsetParameterIndex
()
?
sqlContext
.
getLimitContext
().
getOffset
()
:
(
int
)
parameters
.
get
(
sqlContext
.
getLimitContext
().
getOffsetParameterIndex
());
int
rowCount
=
-
1
==
sqlContext
.
getLimitContext
().
getRowCountParameterIndex
()
?
sqlContext
.
getLimitContext
().
getRowCount
()
:
(
int
)
parameters
.
get
(
sqlContext
.
getLimitContext
().
getRowCountParameterIndex
());
sqlContext
.
setLimitContext
(
new
LimitContext
(
offset
,
rowCount
,
sqlContext
.
getLimitContext
().
getOffsetParameterIndex
(),
sqlContext
.
getLimitContext
().
getRowCountParameterIndex
()));
if
(
offset
<
0
||
rowCount
<
0
)
{
throw
new
SQLParsingException
(
"LIMIT offset and row count can not be a negative value."
);
}
}
private
List
<
Number
>
generateId
()
{
if
(!(
sqlContext
instanceof
InsertSQLContext
))
{
return
Collections
.
emptyList
();
...
...
sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/router/SQLRouteEngine.java
浏览文件 @
c33a09b7
...
...
@@ -36,6 +36,7 @@ import com.dangdang.ddframe.rdb.sharding.parsing.parser.token.OffsetLimitToken;
import
com.dangdang.ddframe.rdb.sharding.parsing.parser.token.RowCountLimitToken
;
import
com.dangdang.ddframe.rdb.sharding.rewrite.DerivedColumnUtils
;
import
com.dangdang.ddframe.rdb.sharding.rewrite.GenerateKeysUtils
;
import
com.dangdang.ddframe.rdb.sharding.rewrite.LimitUtils
;
import
com.dangdang.ddframe.rdb.sharding.rewrite.SQLBuilder
;
import
com.dangdang.ddframe.rdb.sharding.rewrite.SQLRewriteEngine
;
import
com.dangdang.ddframe.rdb.sharding.router.binding.BindingTablesRouter
;
...
...
@@ -102,6 +103,9 @@ public final class SQLRouteEngine {
if
(
result
instanceof
SelectSQLContext
)
{
DerivedColumnUtils
.
appendDerivedColumns
((
SelectSQLContext
)
result
);
}
if
(
null
!=
result
.
getLimitContext
())
{
LimitUtils
.
appendLimit
(
result
,
parameters
);
}
return
result
;
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录