File tree Expand file tree Collapse file tree 2 files changed +5
-4
lines changed
main/java/software/amazon/awssdk/services/sqs/internal/batchmanager
test/java/software/amazon/awssdk/services/sqs/batchmanager Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Original file line number Diff line number Diff line change 24
24
import software .amazon .awssdk .services .sqs .SqsAsyncClient ;
25
25
import software .amazon .awssdk .services .sqs .model .ReceiveMessageRequest ;
26
26
import software .amazon .awssdk .services .sqs .model .ReceiveMessageResponse ;
27
+ import software .amazon .awssdk .utils .SdkAutoCloseable ;
27
28
28
29
@ SdkInternalApi
29
- public class ReceiveBatchManager {
30
+ public class ReceiveBatchManager implements SdkAutoCloseable {
30
31
31
32
private final SqsAsyncClient sqsClient ;
32
33
private final ScheduledExecutorService executor ;
@@ -62,8 +63,8 @@ public CompletableFuture<ReceiveMessageResponse> processRequest(ReceiveMessageRe
62
63
});
63
64
}
64
65
65
- public void shutdown () {
66
+ @ Override
67
+ public void close () {
66
68
receiveQueueBuffer .shutdown ();
67
- executor .shutdown ();
68
69
}
69
70
}
Original file line number Diff line number Diff line change @@ -165,7 +165,7 @@ public void testShutdown() throws Exception {
165
165
ReceiveMessageRequest request = ReceiveMessageRequest .builder ().maxNumberOfMessages (10 ).build ();
166
166
CompletableFuture <ReceiveMessageResponse > futureResponse = receiveBatchManager .processRequest (request );
167
167
168
- receiveBatchManager .shutdown ();
168
+ receiveBatchManager .close ();
169
169
170
170
assertThrows (IllegalStateException .class , () -> receiveBatchManager .processRequest (request ));
171
171
You can’t perform that action at this time.
0 commit comments