Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
apache
Shardingsphere
提交
662223f0
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,发现更多精彩内容 >>
提交
662223f0
编写于
5月 04, 2017
作者:
T
terrymanu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
refactor generateId
上级
e0a2692b
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
10 addition
and
5 deletion
+10
-5
sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/router/PreparedSQLRouter.java
...ngdang/ddframe/rdb/sharding/router/PreparedSQLRouter.java
+10
-5
未找到文件。
sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/router/PreparedSQLRouter.java
浏览文件 @
662223f0
...
...
@@ -25,6 +25,8 @@ import com.dangdang.ddframe.rdb.sharding.parser.sql.context.SQLContext;
import
com.google.common.base.Optional
;
import
lombok.RequiredArgsConstructor
;
import
java.util.ArrayList
;
import
java.util.Collections
;
import
java.util.List
;
/**
...
...
@@ -54,27 +56,30 @@ public class PreparedSQLRouter {
if
(
null
==
sqlContext
)
{
sqlContext
=
engine
.
parseSQL
(
logicSql
,
parameters
);
}
else
{
generateId
(
parameters
);
List
<
Object
>
generatedIds
=
generateId
();
parameters
.
addAll
(
generatedIds
);
sqlContext
.
getConditionContext
().
setNewConditionValue
(
parameters
);
}
return
engine
.
routeSQL
(
sqlContext
,
parameters
);
}
private
void
generateId
(
final
List
<
Object
>
parameters
)
{
private
List
<
Object
>
generateId
(
)
{
if
(!(
sqlContext
instanceof
InsertSQLContext
))
{
return
;
return
Collections
.
emptyList
()
;
}
Optional
<
TableRule
>
tableRuleOptional
=
shardingRule
.
tryFindTableRule
(
sqlContext
.
getTables
().
iterator
().
next
().
getName
());
if
(!
tableRuleOptional
.
isPresent
())
{
return
;
return
Collections
.
emptyList
()
;
}
TableRule
tableRule
=
tableRuleOptional
.
get
();
GeneratedKeyContext
generatedKeyContext
=
((
InsertSQLContext
)
sqlContext
).
getGeneratedKeyContext
();
List
<
Object
>
result
=
new
ArrayList
<>(
generatedKeyContext
.
getColumns
().
size
());
for
(
String
each
:
generatedKeyContext
.
getColumns
())
{
Object
id
=
tableRule
.
generateId
(
each
);
parameters
.
add
(
id
);
result
.
add
(
id
);
generatedKeyContext
.
putValue
(
each
,
id
);
}
return
result
;
}
}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录