Examples with QueryRequest software.amazon.awssdk.services.query.model.QueryRequest used on opensource projects

Example 1 with QueryRequest

use of software.amazon.awssdk.services.query.model.QueryRequest in project aws-sdk-java-v2 by aws.

the class DefaultQueryWaiter method applyWaitersUserAgent.

private <T extends QueryRequest> T applyWaitersUserAgent(T request) {
    Consumer<AwsRequestOverrideConfiguration.Builder> userAgentApplier = b -> b.addApiName(ApiName.builder().version("waiter").name("hll").build());
    AwsRequestOverrideConfiguration overrideConfiguration = request.overrideConfiguration().map(c -> c.toBuilder().applyMutation(userAgentApplier).build()).orElse((AwsRequestOverrideConfiguration.builder().applyMutation(userAgentApplier).build()));
    return (T) request.toBuilder().overrideConfiguration(overrideConfiguration).build();
}

Example 2 with QueryRequest

use of software.amazon.awssdk.services.query.model.QueryRequest in project aws-sdk-java-v2 by aws.

the class DefaultQueryAsyncWaiter method applyWaitersUserAgent.

private <T extends QueryRequest> T applyWaitersUserAgent(T request) {
    Consumer<AwsRequestOverrideConfiguration.Builder> userAgentApplier = b -> b.addApiName(ApiName.builder().version("waiter").name("hll").build());
    AwsRequestOverrideConfiguration overrideConfiguration = request.overrideConfiguration().map(c -> c.toBuilder().applyMutation(userAgentApplier).build()).orElse((AwsRequestOverrideConfiguration.builder().applyMutation(userAgentApplier).build()));
    return (T) request.toBuilder().overrideConfiguration(overrideConfiguration).build();
}

Example 3 with QueryRequest

use of software.amazon.awssdk.services.query.model.QueryRequest in project aws-sdk-java-v2 by aws.

the class DefaultQueryAsyncClient method applySignerOverride.

private <T extends QueryRequest> T applySignerOverride(T request, Signer signer) {
    if (request.overrideConfiguration().flatMap(c -> c.signer()).isPresent()) {
        return request;
    }
    Consumer<AwsRequestOverrideConfiguration.Builder> signerOverride = b -> b.signer(signer).build();
    AwsRequestOverrideConfiguration overrideConfiguration = request.overrideConfiguration().map(c -> c.toBuilder().applyMutation(signerOverride).build()).orElse((AwsRequestOverrideConfiguration.builder().applyMutation(signerOverride).build()));
    return (T) request.toBuilder().overrideConfiguration(overrideConfiguration).build();
}