diff --git a/sdk/core/azure-core-amqp/src/main/java/com/azure/core/amqp/implementation/AmqpReceiveLink.java b/sdk/core/azure-core-amqp/src/main/java/com/azure/core/amqp/implementation/AmqpReceiveLink.java index 5a769b866b4d1..aeed0e5a30454 100644 --- a/sdk/core/azure-core-amqp/src/main/java/com/azure/core/amqp/implementation/AmqpReceiveLink.java +++ b/sdk/core/azure-core-amqp/src/main/java/com/azure/core/amqp/implementation/AmqpReceiveLink.java @@ -8,6 +8,7 @@ import reactor.core.publisher.Flux; import reactor.core.publisher.Mono; +import java.io.UncheckedIOException; import java.util.function.Supplier; /** @@ -32,6 +33,9 @@ public interface AmqpReceiveLink extends AmqpLink { * allow the receiver to receive {@code credits} more deliveries. * * @param credits Number of credits to add to the receive link. + * + * @throws IllegalStateException if adding credits to a closed link. + * @throws UncheckedIOException if the work could not be scheduled on the receive link. */ Mono addCredits(int credits); diff --git a/sdk/core/azure-core-amqp/src/main/java/com/azure/core/amqp/implementation/ReactorReceiver.java b/sdk/core/azure-core-amqp/src/main/java/com/azure/core/amqp/implementation/ReactorReceiver.java index 69466cbd8adf4..69528da03de6c 100644 --- a/sdk/core/azure-core-amqp/src/main/java/com/azure/core/amqp/implementation/ReactorReceiver.java +++ b/sdk/core/azure-core-amqp/src/main/java/com/azure/core/amqp/implementation/ReactorReceiver.java @@ -24,6 +24,7 @@ import reactor.core.scheduler.Schedulers; import java.io.IOException; +import java.io.UncheckedIOException; import java.util.Objects; import java.util.concurrent.RejectedExecutionException; import java.util.concurrent.atomic.AtomicBoolean; @@ -181,7 +182,7 @@ public Mono addCredits(int credits) { sink.success(); }); } catch (IOException e) { - sink.error(new RuntimeException(String.format( + sink.error(new UncheckedIOException(String.format( "connectionId[%s] linkName[%s] Unable to schedule work to add more credits.", handler.getConnectionId(), getLinkName()), e)); }