Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -106,27 +106,8 @@ public abstract class BaseJdbcClient
protected final Set<String> jdbcTypesMappedToVarchar;
protected final RemoteQueryModifier queryModifier;
private final IdentifierMapping identifierMapping;

private final boolean supportsRetries;

public BaseJdbcClient(
BaseJdbcConfig config,
String identifierQuote,
ConnectionFactory connectionFactory,
QueryBuilder queryBuilder,
IdentifierMapping identifierMapping,
RemoteQueryModifier remoteQueryModifier)
{
this(
identifierQuote,
connectionFactory,
queryBuilder,
config.getJdbcTypesMappedToVarchar(),
identifierMapping,
remoteQueryModifier,
false);
}

public BaseJdbcClient(
String identifierQuote,
ConnectionFactory connectionFactory,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ public TestingH2JdbcClient(BaseJdbcConfig config, ConnectionFactory connectionFa

public TestingH2JdbcClient(BaseJdbcConfig config, ConnectionFactory connectionFactory, IdentifierMapping identifierMapping)
{
super(config, "\"", connectionFactory, new DefaultQueryBuilder(RemoteQueryModifier.NONE), identifierMapping, RemoteQueryModifier.NONE);
super("\"", connectionFactory, new DefaultQueryBuilder(RemoteQueryModifier.NONE), config.getJdbcTypesMappedToVarchar(), identifierMapping, RemoteQueryModifier.NONE, false);
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -209,7 +209,7 @@ public ClickHouseClient(
IdentifierMapping identifierMapping,
RemoteQueryModifier queryModifier)
{
super(config, "\"", connectionFactory, queryBuilder, identifierMapping, queryModifier);
super("\"", connectionFactory, queryBuilder, config.getJdbcTypesMappedToVarchar(), identifierMapping, queryModifier, false);
this.uuidType = typeManager.getType(new TypeSignature(StandardTypes.UUID));
this.ipAddressType = typeManager.getType(new TypeSignature(StandardTypes.IPADDRESS));
JdbcTypeHandle bigintTypeHandle = new JdbcTypeHandle(Types.BIGINT, Optional.of("bigint"), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty());
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,7 @@ public class DruidJdbcClient
@Inject
public DruidJdbcClient(BaseJdbcConfig config, ConnectionFactory connectionFactory, QueryBuilder queryBuilder, IdentifierMapping identifierMapping, RemoteQueryModifier queryModifier)
{
super(config, "\"", connectionFactory, queryBuilder, identifierMapping, queryModifier);
super("\"", connectionFactory, queryBuilder, config.getJdbcTypesMappedToVarchar(), identifierMapping, queryModifier, false);
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ public IgniteClient(
IdentifierMapping identifierMapping,
RemoteQueryModifier queryModifier)
{
super(config, "`", connectionFactory, queryBuilder, identifierMapping, queryModifier);
super("`", connectionFactory, queryBuilder, config.getJdbcTypesMappedToVarchar(), identifierMapping, queryModifier, false);

JdbcTypeHandle bigintTypeHandle = new JdbcTypeHandle(Types.BIGINT, Optional.of("bigint"), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty());
this.connectorExpressionRewriter = JdbcConnectorExpressionRewriterBuilder.newBuilder()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,7 @@ public class MariaDbClient
@Inject
public MariaDbClient(BaseJdbcConfig config, ConnectionFactory connectionFactory, QueryBuilder queryBuilder, IdentifierMapping identifierMapping, RemoteQueryModifier queryModifier)
{
super(config, "`", connectionFactory, queryBuilder, identifierMapping, queryModifier);
super("`", connectionFactory, queryBuilder, config.getJdbcTypesMappedToVarchar(), identifierMapping, queryModifier, false);

JdbcTypeHandle bigintTypeHandle = new JdbcTypeHandle(Types.BIGINT, Optional.of("bigint"), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty());
ConnectorExpressionRewriter<String> connectorExpressionRewriter = JdbcConnectorExpressionRewriterBuilder.newBuilder()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -203,7 +203,7 @@ public OracleClient(
IdentifierMapping identifierMapping,
RemoteQueryModifier queryModifier)
{
super(config, "\"", connectionFactory, queryBuilder, identifierMapping, queryModifier);
super("\"", connectionFactory, queryBuilder, config.getJdbcTypesMappedToVarchar(), identifierMapping, queryModifier, false);

this.synonymsEnabled = oracleConfig.isSynonymsEnabled();

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -241,7 +241,7 @@ public RedshiftClient(
RemoteQueryModifier queryModifier,
RedshiftConfig redshiftConfig)
{
super(config, "\"", connectionFactory, queryBuilder, identifierMapping, queryModifier);
super("\"", connectionFactory, queryBuilder, config.getJdbcTypesMappedToVarchar(), identifierMapping, queryModifier, false);
this.legacyTypeMapping = redshiftConfig.isLegacyTypeMapping();
ConnectorExpressionRewriter<String> connectorExpressionRewriter = JdbcConnectorExpressionRewriterBuilder.newBuilder()
.addStandardRules(this::quoted)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -153,9 +153,34 @@ public class SingleStoreClient
private final Type jsonType;

@Inject
public SingleStoreClient(BaseJdbcConfig config, ConnectionFactory connectionFactory, QueryBuilder queryBuilder, TypeManager typeManager, IdentifierMapping identifierMapping, RemoteQueryModifier queryModifier)
public SingleStoreClient(
BaseJdbcConfig config,
ConnectionFactory connectionFactory,
QueryBuilder queryBuilder,
TypeManager typeManager,
IdentifierMapping identifierMapping,
RemoteQueryModifier queryModifier)
{
super(config, "`", connectionFactory, queryBuilder, identifierMapping, queryModifier);
this(
config,
connectionFactory,
queryBuilder,
typeManager,
identifierMapping,
queryModifier,
false);
}

protected SingleStoreClient(
BaseJdbcConfig config,
ConnectionFactory connectionFactory,
QueryBuilder queryBuilder,
TypeManager typeManager,
IdentifierMapping identifierMapping,
RemoteQueryModifier queryModifier,
boolean supportsRetries)
{
super("`", connectionFactory, queryBuilder, config.getJdbcTypesMappedToVarchar(), identifierMapping, queryModifier, supportsRetries);
requireNonNull(typeManager, "typeManager is null");
this.jsonType = typeManager.getType(new TypeSignature(StandardTypes.JSON));
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -81,14 +81,20 @@ public abstract class BaseFailureRecoveryTest
protected static final int INVOCATION_COUNT = 1;
private static final Duration MAX_ERROR_DURATION = new Duration(5, SECONDS);
private static final Duration REQUEST_TIMEOUT = new Duration(5, SECONDS);
private static final int MAX_PARALLEL_TEST_CONCURRENCY = 4;
private static final int DEFAULT_MAX_PARALLEL_TEST_CONCURRENCY = 4;

private final RetryPolicy retryPolicy;
private final Semaphore parallelTestsSemaphore = new Semaphore(MAX_PARALLEL_TEST_CONCURRENCY);
private final Semaphore parallelTestsSemaphore;

protected BaseFailureRecoveryTest(RetryPolicy retryPolicy)
{
this(retryPolicy, DEFAULT_MAX_PARALLEL_TEST_CONCURRENCY);
}

protected BaseFailureRecoveryTest(RetryPolicy retryPolicy, int maxParallelTestConcurrency)
{
this.retryPolicy = requireNonNull(retryPolicy, "retryPolicy is null");
this.parallelTestsSemaphore = new Semaphore(maxParallelTestConcurrency);
}

protected RetryPolicy getRetryPolicy()
Expand Down