Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
apache
Shardingsphere
提交
e7abd780
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,发现更多精彩内容 >>
提交
e7abd780
编写于
6月 04, 2019
作者:
T
tristaZero
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
delete fillWithInsertSetAddItemsToken()
上级
bb71746f
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
1 addition
and
64 deletion
+1
-64
sharding-core/sharding-core-parse/sharding-core-parse-common/src/main/java/org/apache/shardingsphere/core/parse/filler/encrypt/dml/EncryptSetAssignmentsFiller.java
...parse/filler/encrypt/dml/EncryptSetAssignmentsFiller.java
+1
-30
sharding-core/sharding-core-parse/sharding-core-parse-common/src/main/java/org/apache/shardingsphere/core/parse/filler/sharding/dml/update/ShardingSetAssignmentsFiller.java
...ler/sharding/dml/update/ShardingSetAssignmentsFiller.java
+0
-34
未找到文件。
sharding-core/sharding-core-parse/sharding-core-parse-common/src/main/java/org/apache/shardingsphere/core/parse/filler/encrypt/dml/EncryptSetAssignmentsFiller.java
浏览文件 @
e7abd780
...
...
@@ -17,9 +17,7 @@
package
org.apache.shardingsphere.core.parse.filler.encrypt.dml
;
import
com.google.common.base.Optional
;
import
lombok.Setter
;
import
org.apache.shardingsphere.core.parse.filler.api.EncryptRuleAwareFiller
;
import
org.apache.shardingsphere.core.parse.filler.api.SQLSegmentFiller
;
import
org.apache.shardingsphere.core.parse.sql.context.condition.Column
;
import
org.apache.shardingsphere.core.parse.sql.context.insertvalue.InsertValue
;
...
...
@@ -29,11 +27,7 @@ import org.apache.shardingsphere.core.parse.sql.segment.dml.expr.ExpressionSegme
import
org.apache.shardingsphere.core.parse.sql.statement.SQLStatement
;
import
org.apache.shardingsphere.core.parse.sql.statement.dml.InsertStatement
;
import
org.apache.shardingsphere.core.parse.sql.statement.dml.UpdateStatement
;
import
org.apache.shardingsphere.core.parse.sql.token.impl.InsertSetAddItemsToken
;
import
org.apache.shardingsphere.core.rule.EncryptRule
;
import
java.util.ArrayList
;
import
java.util.Collection
;
import
java.util.LinkedList
;
import
java.util.List
;
...
...
@@ -43,9 +37,7 @@ import java.util.List;
* @author zhangliang
*/
@Setter
public
final
class
EncryptSetAssignmentsFiller
implements
SQLSegmentFiller
<
SetAssignmentsSegment
>,
EncryptRuleAwareFiller
{
private
EncryptRule
encryptRule
;
public
final
class
EncryptSetAssignmentsFiller
implements
SQLSegmentFiller
<
SetAssignmentsSegment
>
{
@Override
public
void
fill
(
final
SetAssignmentsSegment
sqlSegment
,
final
SQLStatement
sqlStatement
)
{
...
...
@@ -63,7 +55,6 @@ public final class EncryptSetAssignmentsFiller implements SQLSegmentFiller<SetAs
InsertValue
insertValue
=
getInsertValue
(
sqlSegment
,
insertStatement
);
insertStatement
.
getValues
().
add
(
insertValue
);
insertStatement
.
setParametersIndex
(
insertValue
.
getParametersCount
());
fillWithInsertSetAddItemsToken
(
insertStatement
,
sqlSegment
);
}
private
InsertValue
getInsertValue
(
final
SetAssignmentsSegment
sqlSegment
,
final
InsertStatement
insertStatement
)
{
...
...
@@ -74,26 +65,6 @@ public final class EncryptSetAssignmentsFiller implements SQLSegmentFiller<SetAs
return
new
InsertValue
(
columnValues
);
}
private
void
fillWithInsertSetAddItemsToken
(
final
InsertStatement
insertStatement
,
final
SetAssignmentsSegment
sqlSegment
)
{
Collection
<
String
>
columnNames
=
getQueryAssistedColumn
(
insertStatement
);
if
(
columnNames
.
isEmpty
())
{
return
;
}
List
<
AssignmentSegment
>
assignments
=
new
ArrayList
<>(
sqlSegment
.
getAssignments
());
insertStatement
.
getSQLTokens
().
add
(
new
InsertSetAddItemsToken
(
assignments
.
get
(
assignments
.
size
()
-
1
).
getStopIndex
()
+
1
,
columnNames
));
}
private
Collection
<
String
>
getQueryAssistedColumn
(
final
InsertStatement
insertStatement
)
{
Collection
<
String
>
result
=
new
LinkedList
<>();
for
(
String
each
:
insertStatement
.
getColumnNames
())
{
Optional
<
String
>
assistedColumnName
=
encryptRule
.
getEncryptorEngine
().
getAssistedQueryColumn
(
insertStatement
.
getTables
().
getSingleTableName
(),
each
);
if
(
assistedColumnName
.
isPresent
())
{
result
.
add
(
assistedColumnName
.
get
());
}
}
return
result
;
}
private
void
fillUpdate
(
final
SetAssignmentsSegment
sqlSegment
,
final
UpdateStatement
updateStatement
)
{
String
tableName
=
updateStatement
.
getTables
().
getSingleTableName
();
for
(
AssignmentSegment
each
:
sqlSegment
.
getAssignments
())
{
...
...
sharding-core/sharding-core-parse/sharding-core-parse-common/src/main/java/org/apache/shardingsphere/core/parse/filler/sharding/dml/update/ShardingSetAssignmentsFiller.java
浏览文件 @
e7abd780
...
...
@@ -17,7 +17,6 @@
package
org.apache.shardingsphere.core.parse.filler.sharding.dml.update
;
import
com.google.common.base.Optional
;
import
lombok.Setter
;
import
org.apache.shardingsphere.core.metadata.table.ShardingTableMetaData
;
import
org.apache.shardingsphere.core.parse.exception.SQLParsingException
;
...
...
@@ -36,11 +35,8 @@ import org.apache.shardingsphere.core.parse.sql.segment.dml.predicate.PredicateS
import
org.apache.shardingsphere.core.parse.sql.statement.SQLStatement
;
import
org.apache.shardingsphere.core.parse.sql.statement.dml.InsertStatement
;
import
org.apache.shardingsphere.core.parse.sql.statement.dml.UpdateStatement
;
import
org.apache.shardingsphere.core.parse.sql.token.impl.InsertSetAddItemsToken
;
import
org.apache.shardingsphere.core.rule.ShardingRule
;
import
java.util.ArrayList
;
import
java.util.Collection
;
import
java.util.Iterator
;
import
java.util.LinkedList
;
import
java.util.List
;
...
...
@@ -88,36 +84,6 @@ public final class ShardingSetAssignmentsFiller implements SQLSegmentFiller<SetA
insertStatement
.
getValues
().
add
(
insertValue
);
insertStatement
.
getRouteCondition
().
getOrConditions
().
add
(
andCondition
);
insertStatement
.
setParametersIndex
(
insertValue
.
getParametersCount
());
fillWithInsertSetAddItemsToken
(
insertStatement
,
sqlSegment
);
}
private
void
fillWithInsertSetAddItemsToken
(
final
InsertStatement
insertStatement
,
final
SetAssignmentsSegment
sqlSegment
)
{
Collection
<
String
>
columnNames
=
getQueryAssistedColumn
(
insertStatement
);
if
(
getGeneratedKeyColumn
(
insertStatement
).
isPresent
())
{
columnNames
.
add
(
getGeneratedKeyColumn
(
insertStatement
).
get
());
}
if
(
columnNames
.
isEmpty
())
{
return
;
}
List
<
AssignmentSegment
>
assignments
=
new
ArrayList
<>(
sqlSegment
.
getAssignments
());
insertStatement
.
getSQLTokens
().
add
(
new
InsertSetAddItemsToken
(
assignments
.
get
(
assignments
.
size
()
-
1
).
getStopIndex
()
+
1
,
columnNames
));
}
private
Optional
<
String
>
getGeneratedKeyColumn
(
final
InsertStatement
insertStatement
)
{
String
tableName
=
insertStatement
.
getTables
().
getSingleTableName
();
Optional
<
String
>
generateKeyColumn
=
shardingRule
.
findGenerateKeyColumnName
(
tableName
);
return
generateKeyColumn
.
isPresent
()
&&
!
insertStatement
.
getColumnNames
().
contains
(
generateKeyColumn
.
get
())
?
generateKeyColumn
:
Optional
.<
String
>
absent
();
}
private
Collection
<
String
>
getQueryAssistedColumn
(
final
InsertStatement
insertStatement
)
{
Collection
<
String
>
result
=
new
LinkedList
<>();
for
(
String
each
:
insertStatement
.
getColumnNames
())
{
Optional
<
String
>
assistedColumnName
=
shardingRule
.
getEncryptRule
().
getEncryptorEngine
().
getAssistedQueryColumn
(
insertStatement
.
getTables
().
getSingleTableName
(),
each
);
if
(
assistedColumnName
.
isPresent
())
{
result
.
add
(
assistedColumnName
.
get
());
}
}
return
result
;
}
private
int
getColumnCountExcludeAssistedQueryColumns
(
final
InsertStatement
insertStatement
)
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录