Stuck on java.util.concurrent.Future#get(long, java.util.concurrent.TimeUnit) when using java.util.concurrent.Executors#newVirtualThreadPerTaskExecutor

[ashr123]

Describe the bug

When using java.util.concurrent.Executors#newVirtualThreadPerTaskExecutor & java.util.concurrent.ExecutorService#invokeAll(java.util.Collection<? extends java.util.concurrent.Callable<T>>, long, java.util.concurrent.TimeUnit) & inside the Callable I call software.amazon.awssdk.enhanced.dynamodb.DynamoDbTable#getItem(software.amazon.awssdk.enhanced.dynamodb.Key), the code doesn't return from the getItem method, in addition, the Future#get doesn't return either from cancelation or timeout

Expected Behavior

Return normally or return exceptionally with CancelationException or TimeoutException

Current Behavior

See bug desctiption

Reproduction Steps

final DynamoDbTable<EnhancedDocument> table = DynamoDbEnhancedClient.create()
        .table(
                "<table-name>",
                TableSchema.documentSchemaBuilder()
                        .addIndexPartitionKey(TableMetadata.primaryIndexName(), "<primary partition key>", AttributeValueType.N)
                        .addIndexSortKey(TableMetadata.primaryIndexName(), "<primary sorting key>", AttributeValueType.S)
                        .build()
        );

try (ExecutorService executorService = Executors.newVirtualThreadPerTaskExecutor()) {
    final List<Future<EnhancedDocument>> futures = executorService.invokeAll(
            List.of(() -> table.getItem(Key.builder()
                    .partitionValue(123)
                    .sortValue("456")
                    .build())),
            5,
            TimeUnit.SECONDS
    );
    System.out.println(futures.getFirst().get());
} catch (ExecutionException | InterruptedException e) {
    throw new RuntimeException(e);
}

But, when using Executors.newThreadPerTaskExecutor(Thread.ofPlatform().factory()), this doesn't happen, in addition, when deploying our code to production (using virtual threads), on Ubuntu Server 22.04 EC2s, this doesn't happen either.

Possible Solution

No response

Additional Information/Context

No response

AWS Java SDK version used

2.20.157

JDK version used

openjdk 21.0.3 2024-04-16 LTS
OpenJDK Runtime Environment Corretto-21.0.3.9.1 (build 21.0.3+9-LTS)
OpenJDK 64-Bit Server VM Corretto-21.0.3.9.1 (build 21.0.3+9-LTS, mixed mode, sharing)

Operating System and version

MacOS 14.4.1

[ashr123]

Never mind, I've havn't been able to recreate this error in an independent project, but I've discovered that since SDK v2.22 there is new HTTP client called AwsCrtHttpClient that uses non-blocking I/O (in contrast of the default ApacheHttpClient) that is perfect for working with virtual threads, you should update your documentation and reference the methodology of working with virtual threads

[github-actions]

This issue is now closed. Comments on closed issues are hard for our team to see.
If you need more assistance, please open a new issue that references this one.