提交 cc406218 编写于 作者: T tristaZero

modify constructor of QueryResultMetaData

上级 a41a8dde
......@@ -53,13 +53,13 @@ public final class MemoryQueryResult implements QueryResult {
@SneakyThrows
public MemoryQueryResult(final ResultSet resultSet, final ShardingRule shardingRule) {
resultData = getResultData(resultSet);
metaData = new QueryResultMetaData(resultSet.getMetaData(), shardingRule, shardingRule.getEncryptRule().getEncryptorEngine());
metaData = new QueryResultMetaData(resultSet.getMetaData(), shardingRule);
}
@SneakyThrows
public MemoryQueryResult(final ResultSet resultSet, final EncryptRule encryptRule) {
resultData = getResultData(resultSet);
metaData = new QueryResultMetaData(resultSet.getMetaData(), null, encryptRule.getEncryptorEngine());
metaData = new QueryResultMetaData(resultSet.getMetaData(), encryptRule);
}
@SneakyThrows
......
......@@ -19,6 +19,7 @@ package org.apache.shardingsphere.core.execute.sql.execute.result;
import com.google.common.base.Optional;
import lombok.SneakyThrows;
import org.apache.shardingsphere.core.rule.EncryptRule;
import org.apache.shardingsphere.core.rule.ShardingRule;
import org.apache.shardingsphere.core.rule.TableRule;
import org.apache.shardingsphere.core.strategy.encrypt.ShardingEncryptorEngine;
......@@ -44,17 +45,28 @@ public final class QueryResultMetaData {
private final ShardingEncryptorEngine shardingEncryptorEngine;
@SneakyThrows
public QueryResultMetaData(final ResultSetMetaData resultSetMetaData, final ShardingRule shardingRule, final ShardingEncryptorEngine shardingEncryptorEngine) {
@SneakyThrows
public QueryResultMetaData(final ResultSetMetaData resultSetMetaData, final ShardingRule shardingRule) {
columnLabelAndIndexes = getColumnLabelAndIndexMap(resultSetMetaData);
this.resultSetMetaData = resultSetMetaData;
this.shardingRule = shardingRule;
this.shardingEncryptorEngine = shardingEncryptorEngine;
this.shardingEncryptorEngine = shardingRule.getEncryptRule().getEncryptorEngine();
}
@SneakyThrows
public QueryResultMetaData(final ResultSetMetaData resultSetMetaData, final EncryptRule encryptRule) {
columnLabelAndIndexes = getColumnLabelAndIndexMap(resultSetMetaData);
this.resultSetMetaData = resultSetMetaData;
this.shardingRule = null;
this.shardingEncryptorEngine = encryptRule.getEncryptorEngine();
}
@SneakyThrows
public QueryResultMetaData(final ResultSetMetaData resultSetMetaData) {
this(resultSetMetaData, null, new ShardingEncryptorEngine());
columnLabelAndIndexes = getColumnLabelAndIndexMap(resultSetMetaData);
this.resultSetMetaData = resultSetMetaData;
this.shardingRule = null;
this.shardingEncryptorEngine = new ShardingEncryptorEngine();
}
@SneakyThrows
......
......@@ -52,13 +52,13 @@ public final class StreamQueryResult implements QueryResult {
@SneakyThrows
public StreamQueryResult(final ResultSet resultSet, final ShardingRule shardingRule) {
this.resultSet = resultSet;
metaData = new QueryResultMetaData(resultSet.getMetaData(), shardingRule, shardingRule.getEncryptRule().getEncryptorEngine());
metaData = new QueryResultMetaData(resultSet.getMetaData(), shardingRule);
}
@SneakyThrows
public StreamQueryResult(final ResultSet resultSet, final EncryptRule encryptRule) {
this.resultSet = resultSet;
metaData = new QueryResultMetaData(resultSet.getMetaData(), null, encryptRule.getEncryptorEngine());
metaData = new QueryResultMetaData(resultSet.getMetaData(), encryptRule);
}
@SneakyThrows
......
......@@ -49,7 +49,7 @@ public class QueryResultMetaDataTest {
public void setUp() {
ResultSetMetaData resultSetMetaData = getResultMetaData();
ShardingRule shardingRule = getShardingRule();
queryResultMetaData = new QueryResultMetaData(resultSetMetaData, shardingRule, shardingRule.getEncryptRule().getEncryptorEngine());
queryResultMetaData = new QueryResultMetaData(resultSetMetaData, shardingRule);
}
@SuppressWarnings("unchecked")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册