Skip to content

Commit 93d2b7e

Browse files
committed
SdkClosable implemented
1 parent f510b82 commit 93d2b7e

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

services/sqs/src/main/java/software/amazon/awssdk/services/sqs/internal/batchmanager/ReceiveBatchManager.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,10 @@
2424
import software.amazon.awssdk.services.sqs.SqsAsyncClient;
2525
import software.amazon.awssdk.services.sqs.model.ReceiveMessageRequest;
2626
import software.amazon.awssdk.services.sqs.model.ReceiveMessageResponse;
27+
import software.amazon.awssdk.utils.SdkAutoCloseable;
2728

2829
@SdkInternalApi
29-
public class ReceiveBatchManager {
30+
public class ReceiveBatchManager implements SdkAutoCloseable {
3031

3132
private final SqsAsyncClient sqsClient;
3233
private final ScheduledExecutorService executor;
@@ -62,8 +63,8 @@ public CompletableFuture<ReceiveMessageResponse> processRequest(ReceiveMessageRe
6263
});
6364
}
6465

65-
public void shutdown() {
66+
@Override
67+
public void close() {
6668
receiveQueueBuffer.shutdown();
67-
executor.shutdown();
6869
}
6970
}

services/sqs/src/test/java/software/amazon/awssdk/services/sqs/batchmanager/ReceiveBatchManagerTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ public void testShutdown() throws Exception {
165165
ReceiveMessageRequest request = ReceiveMessageRequest.builder().maxNumberOfMessages(10).build();
166166
CompletableFuture<ReceiveMessageResponse> futureResponse = receiveBatchManager.processRequest(request);
167167

168-
receiveBatchManager.shutdown();
168+
receiveBatchManager.close();
169169

170170
assertThrows(IllegalStateException.class, () -> receiveBatchManager.processRequest(request));
171171

0 commit comments

Comments
 (0)