(shardingsphere) branch master updated: Refactor EncryptSQLRewriteContextDecorator (#32253)

[zhangliang]

This is an automated email from the ASF dual-hosted git repository.

zhangliang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new 5dcccf1c425 Refactor EncryptSQLRewriteContextDecorator (#32253)
5dcccf1c425 is described below

commit 5dcccf1c4252be1123097bf9799c5e999b2904c5
Author: Liang Zhang <zhangli...@apache.org>
AuthorDate: Wed Jul 24 21:54:12 2024 +0800

    Refactor EncryptSQLRewriteContextDecorator (#32253)
---
 .../encrypt/rewrite/context/EncryptSQLRewriteContextDecorator.java  | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git 
a/features/encrypt/core/src/main/java/org/apache/shardingsphere/encrypt/rewrite/context/EncryptSQLRewriteContextDecorator.java
 
b/features/encrypt/core/src/main/java/org/apache/shardingsphere/encrypt/rewrite/context/EncryptSQLRewriteContextDecorator.java
index 38ee7e74010..fc57662af6b 100644
--- 
a/features/encrypt/core/src/main/java/org/apache/shardingsphere/encrypt/rewrite/context/EncryptSQLRewriteContextDecorator.java
+++ 
b/features/encrypt/core/src/main/java/org/apache/shardingsphere/encrypt/rewrite/context/EncryptSQLRewriteContextDecorator.java
@@ -31,7 +31,7 @@ import 
org.apache.shardingsphere.infra.config.props.ConfigurationProperties;
 import org.apache.shardingsphere.infra.rewrite.context.SQLRewriteContext;
 import 
org.apache.shardingsphere.infra.rewrite.context.SQLRewriteContextDecorator;
 import 
org.apache.shardingsphere.infra.rewrite.parameter.rewriter.ParameterRewriter;
-import 
org.apache.shardingsphere.infra.rewrite.sql.token.generator.SQLTokenGenerator;
+import 
org.apache.shardingsphere.infra.rewrite.sql.token.generator.builder.SQLTokenGeneratorBuilder;
 import org.apache.shardingsphere.infra.route.context.RouteContext;
 import 
org.apache.shardingsphere.sql.parser.statement.core.segment.dml.column.ColumnSegment;
 import 
org.apache.shardingsphere.sql.parser.statement.core.segment.dml.predicate.WhereSegment;
@@ -58,8 +58,8 @@ public final class EncryptSQLRewriteContextDecorator 
implements SQLRewriteContex
                     new EncryptParameterRewriterBuilder(encryptRule, 
databaseName, sqlRewriteContext.getDatabase().getSchemas(), 
sqlStatementContext, encryptConditions).getParameterRewriters();
             rewriteParameters(sqlRewriteContext, parameterRewriters);
         }
-        Collection<SQLTokenGenerator> sqlTokenGenerators = new 
EncryptTokenGenerateBuilder(encryptRule, sqlStatementContext, 
encryptConditions, databaseName).getSQLTokenGenerators();
-        sqlRewriteContext.addSQLTokenGenerators(sqlTokenGenerators);
+        SQLTokenGeneratorBuilder sqlTokenGeneratorBuilder = new 
EncryptTokenGenerateBuilder(encryptRule, sqlStatementContext, 
encryptConditions, databaseName);
+        
sqlRewriteContext.addSQLTokenGenerators(sqlTokenGeneratorBuilder.getSQLTokenGenerators());
     }
     
     private boolean containsEncryptTable(final EncryptRule encryptRule, final 
SQLStatementContext sqlStatementContext) {