提交 0b252b26 编写于 作者: T tristaZero

do not use shardingRule.getShardingEncryptorEngine()

上级 bbe932cf
......@@ -104,7 +104,7 @@ public abstract class BaseShardingEngine {
private Collection<RouteUnit> rewriteAndConvert(final List<Object> parameters, final SQLRouteResult sqlRouteResult) {
SQLRewriteEngine rewriteEngine = new SQLRewriteEngine(shardingRule, sqlRouteResult.getSqlStatement(), parameters);
ShardingSQLRewriter shardingSQLRewriter = new ShardingSQLRewriter(shardingRule, databaseType, sqlRouteResult);
EncryptSQLRewriter encryptSQLRewriter = new EncryptSQLRewriter(shardingRule.getShardingEncryptorEngine(), sqlRouteResult.getSqlStatement(), sqlRouteResult.getOptimizeResult());
EncryptSQLRewriter encryptSQLRewriter = new EncryptSQLRewriter(shardingRule.getEncryptRule().getEncryptorEngine(), sqlRouteResult.getSqlStatement(), sqlRouteResult.getOptimizeResult());
rewriteEngine.init(shardingSQLRewriter, encryptSQLRewriter);
Collection<RouteUnit> result = new LinkedHashSet<>();
for (RoutingUnit each : sqlRouteResult.getRoutingResult().getRoutingUnits()) {
......
......@@ -114,7 +114,7 @@ public final class InsertOptimizeEngine implements OptimizeEngine {
result += 1;
}
if (isNeededToAppendQueryAssistedColumn()) {
result += shardingRule.getShardingEncryptorEngine().getAssistedQueryColumnCount(insertStatement.getTables().getSingleTableName());
result += shardingRule.getEncryptRule().getEncryptorEngine().getAssistedQueryColumnCount(insertStatement.getTables().getSingleTableName());
}
return result;
}
......@@ -154,14 +154,14 @@ public final class InsertOptimizeEngine implements OptimizeEngine {
}
private boolean isNeededToAppendQueryAssistedColumn() {
return shardingRule.getShardingEncryptorEngine().isHasShardingQueryAssistedEncryptor(insertStatement.getTables().getSingleTableName());
return shardingRule.getEncryptRule().getEncryptorEngine().isHasShardingQueryAssistedEncryptor(insertStatement.getTables().getSingleTableName());
}
private void fillWithQueryAssistedColumn(final InsertOptimizeResult insertOptimizeResult, final int insertOptimizeResultIndex) {
Collection<String> assistedColumnNames = new LinkedList<>();
for (String each : insertOptimizeResult.getColumnNames()) {
InsertOptimizeResultUnit unit = insertOptimizeResult.getUnits().get(insertOptimizeResultIndex);
Optional<String> assistedColumnName = shardingRule.getShardingEncryptorEngine().getAssistedQueryColumn(insertStatement.getTables().getSingleTableName(), each);
Optional<String> assistedColumnName = shardingRule.getEncryptRule().getEncryptorEngine().getAssistedQueryColumn(insertStatement.getTables().getSingleTableName(), each);
if (assistedColumnName.isPresent()) {
assistedColumnNames.add(assistedColumnName.get());
fillInsertOptimizeResultUnit(unit, (Comparable<?>) unit.getColumnValue(each));
......
......@@ -115,8 +115,8 @@ public final class PreparedStatementExecutor extends AbstractStatementExecutor {
ResultSet resultSet = preparedStatement.executeQuery();
ShardingRule shardingRule = getConnection().getShardingContext().getShardingRule();
getResultSets().add(resultSet);
return ConnectionMode.MEMORY_STRICTLY == connectionMode ? new StreamQueryResult(resultSet, shardingRule, shardingRule.getShardingEncryptorEngine())
: new MemoryQueryResult(resultSet, shardingRule, shardingRule.getShardingEncryptorEngine());
return ConnectionMode.MEMORY_STRICTLY == connectionMode ? new StreamQueryResult(resultSet, shardingRule, shardingRule.getEncryptRule().getEncryptorEngine())
: new MemoryQueryResult(resultSet, shardingRule, shardingRule.getEncryptRule().getEncryptorEngine());
}
/**
......
......@@ -103,8 +103,8 @@ public final class StatementExecutor extends AbstractStatementExecutor {
ResultSet resultSet = statement.executeQuery(routeUnit.getSqlUnit().getSql());
ShardingRule shardingRule = getConnection().getShardingContext().getShardingRule();
getResultSets().add(resultSet);
return ConnectionMode.MEMORY_STRICTLY == connectionMode ? new StreamQueryResult(resultSet, shardingRule, shardingRule.getShardingEncryptorEngine())
: new MemoryQueryResult(resultSet, shardingRule, shardingRule.getShardingEncryptorEngine());
return ConnectionMode.MEMORY_STRICTLY == connectionMode ? new StreamQueryResult(resultSet, shardingRule, shardingRule.getEncryptRule().getEncryptorEngine())
: new MemoryQueryResult(resultSet, shardingRule, shardingRule.getEncryptRule().getEncryptorEngine());
}
/**
......
......@@ -20,7 +20,6 @@ package org.apache.shardingsphere.shardingjdbc.jdbc.adapter;
import com.google.common.base.Preconditions;
import lombok.Getter;
import org.apache.shardingsphere.core.rule.ShardingRule;
import org.apache.shardingsphere.core.strategy.encrypt.ShardingEncryptorEngine;
import org.apache.shardingsphere.shardingjdbc.jdbc.adapter.executor.ForceExecuteCallback;
import org.apache.shardingsphere.shardingjdbc.jdbc.adapter.executor.ForceExecuteTemplate;
import org.apache.shardingsphere.shardingjdbc.jdbc.core.resultset.ShardingResultSetMetaData;
......@@ -69,10 +68,6 @@ public abstract class AbstractResultSetAdapter extends AbstractUnsupportedOperat
: ((ShardingStatement) statement).getConnection().getShardingContext().getShardingRule();
}
protected final ShardingEncryptorEngine getShardingEncryptorEngine() {
return getShardingRule().getShardingEncryptorEngine();
}
@Override
public final int findColumn(final String columnLabel) throws SQLException {
return resultSets.get(0).findColumn(columnLabel);
......
......@@ -133,7 +133,7 @@ public final class ShardingPreparedStatement extends AbstractShardingPreparedSta
ResultSet resultSet = each.getResultSet();
resultSets.add(resultSet);
ShardingRule shardingRule = connection.getShardingContext().getShardingRule();
queryResults.add(new StreamQueryResult(resultSet, shardingRule, shardingRule.getShardingEncryptorEngine()));
queryResults.add(new StreamQueryResult(resultSet, shardingRule, shardingRule.getEncryptRule().getEncryptorEngine()));
}
if (routeResult.getSqlStatement() instanceof SelectStatement || routeResult.getSqlStatement() instanceof DALStatement) {
MergeEngine mergeEngine = MergeEngineFactory.newInstance(connection.getShardingContext().getDatabaseType(),
......
......@@ -114,7 +114,7 @@ public final class ShardingStatement extends AbstractStatementAdapter {
ResultSet resultSet = each.getResultSet();
resultSets.add(resultSet);
ShardingRule shardingRule = connection.getShardingContext().getShardingRule();
queryResults.add(new StreamQueryResult(resultSet, shardingRule, shardingRule.getShardingEncryptorEngine()));
queryResults.add(new StreamQueryResult(resultSet, shardingRule, shardingRule.getEncryptRule().getEncryptorEngine()));
}
if (routeResult.getSqlStatement() instanceof SelectStatement || routeResult.getSqlStatement() instanceof DALStatement) {
MergeEngine mergeEngine = MergeEngineFactory.newInstance(connection.getShardingContext().getDatabaseType(),
......
......@@ -100,8 +100,8 @@ public final class ProxySQLExecuteCallback extends SQLExecuteCallback<ExecuteRes
LogicSchema logicSchema = backendConnection.getLogicSchema();
if (logicSchema instanceof ShardingSchema) {
ShardingRule shardingRule = logicSchema.getShardingRule();
return connectionMode == ConnectionMode.MEMORY_STRICTLY ? new StreamQueryResult(resultSet, shardingRule, shardingRule.getShardingEncryptorEngine())
: new MemoryQueryResult(resultSet, shardingRule, shardingRule.getShardingEncryptorEngine());
return connectionMode == ConnectionMode.MEMORY_STRICTLY ? new StreamQueryResult(resultSet, shardingRule, shardingRule.getEncryptRule().getEncryptorEngine())
: new MemoryQueryResult(resultSet, shardingRule, shardingRule.getEncryptRule().getEncryptorEngine());
}
if (logicSchema instanceof EncryptSchema) {
EncryptSchema encryptSchema = (EncryptSchema) logicSchema;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册