com.microsoft.azure.eventhubs.impl.MessageReceiver Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of azure-eventhubs Show documentation
Show all versions of azure-eventhubs Show documentation
Please note, a newer package azure-messaging-eventhubs for Azure Event Hubs is available at https://search.maven.org/artifact/com.azure/azure-messaging-eventhubs as of February 2020. While this package will continue to receive critical bug fixes, we strongly encourage you to upgrade. Read the migration guide at https://aka.ms/azsdk/java/migrate/eh for more details.
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
package com.microsoft.azure.eventhubs.impl;
import com.microsoft.azure.eventhubs.ErrorContext;
import com.microsoft.azure.eventhubs.EventHubException;
import com.microsoft.azure.eventhubs.TimeoutException;
import org.apache.qpid.proton.Proton;
import org.apache.qpid.proton.amqp.Symbol;
import org.apache.qpid.proton.amqp.UnknownDescribedType;
import org.apache.qpid.proton.amqp.messaging.Source;
import org.apache.qpid.proton.amqp.messaging.Target;
import org.apache.qpid.proton.amqp.transport.ErrorCondition;
import org.apache.qpid.proton.amqp.transport.ReceiverSettleMode;
import org.apache.qpid.proton.amqp.transport.SenderSettleMode;
import org.apache.qpid.proton.engine.BaseHandler;
import org.apache.qpid.proton.engine.Delivery;
import org.apache.qpid.proton.engine.EndpointState;
import org.apache.qpid.proton.engine.Receiver;
import org.apache.qpid.proton.engine.Session;
import org.apache.qpid.proton.message.Message;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.IOException;
import java.time.Duration;
import java.time.Instant;
import java.time.ZonedDateTime;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.concurrent.CancellationException;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.RejectedExecutionException;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.BiConsumer;
import java.util.function.Consumer;
/**
* Common Receiver that abstracts all amqp related details
* translates event-driven reactor model into async receive Api
*/
public final class MessageReceiver extends ClientEntity implements AmqpReceiver, ErrorContextProvider {
private static final Logger TRACE_LOGGER = LoggerFactory.getLogger(MessageReceiver.class);
private static final int MIN_TIMEOUT_DURATION_MILLIS = 20;
private static final int MAX_OPERATION_TIMEOUT_SCHEDULED = 2;
// TestHooks for code injection
private static volatile Consumer onOpenRetry = null;
private final AtomicInteger operationTimeoutScheduled = new AtomicInteger(0);
private final ConcurrentLinkedQueue pendingReceives;
private final MessagingFactory underlyingFactory;
private final String receivePath;
private final Runnable onOperationTimedout;
private final Duration operationTimeout;
private final CompletableFuture linkClose;
private final ReceiverSettingsProvider settingsProvider;
private final String tokenAudience;
private final ActiveClientTokenManager activeClientTokenManager;
private final WorkItem linkOpen;
private final ConcurrentLinkedQueue prefetchedMessages;
private final ReceiveWork receiveWork;
private final CreateAndReceive createAndReceive;
private final Object errorConditionLock;
private final Timer timer;
private volatile int nextCreditToFlow;
private volatile Receiver receiveLink;
private volatile Duration receiveTimeout;
private volatile Message lastReceivedMessage;
private volatile boolean creatingLink;
private volatile CompletableFuture openTimer;
private volatile CompletableFuture closeTimer;
private int prefetchCount;
private Exception lastKnownLinkError;
private String linkCreationTime; // Used when looking at Java dumps, do not remove.
private MessageReceiver(final MessagingFactory factory,
final String name,
final String recvPath,
final int prefetchCount,
final ReceiverSettingsProvider settingsProvider) {
super(name, factory, factory.executor);
this.underlyingFactory = factory;
this.operationTimeout = factory.getOperationTimeout();
this.receivePath = recvPath;
this.prefetchCount = prefetchCount;
this.prefetchedMessages = new ConcurrentLinkedQueue<>();
this.linkClose = new CompletableFuture<>();
this.lastKnownLinkError = null;
this.receiveTimeout = factory.getOperationTimeout();
this.settingsProvider = settingsProvider;
this.linkOpen = new WorkItem<>(new CompletableFuture<>(), factory.getOperationTimeout());
this.timer = new Timer(factory);
this.pendingReceives = new ConcurrentLinkedQueue<>();
this.errorConditionLock = new Object();
// onOperationTimeout delegate - per receive call
this.onOperationTimedout = new Runnable() {
public void run() {
MessageReceiver.this.operationTimeoutTimerFired();
WorkItem> topWorkItem = null;
while ((topWorkItem = MessageReceiver.this.pendingReceives.peek()) != null) {
if (topWorkItem.getTimeoutTracker().remaining().toMillis() <= MessageReceiver.MIN_TIMEOUT_DURATION_MILLIS) {
WorkItem> dequedWorkItem = MessageReceiver.this.pendingReceives.poll();
if (dequedWorkItem != null && dequedWorkItem.getWork() != null && !dequedWorkItem.getWork().isDone()) {
dequedWorkItem.getWork().complete(null);
} else {
break;
}
} else {
if (MessageReceiver.this.shouldScheduleOperationTimeoutTimer()) {
TimeoutTracker timeoutTracker = topWorkItem.getTimeoutTracker();
if (TRACE_LOGGER.isDebugEnabled()) {
TRACE_LOGGER.debug(
String.format(Locale.US,
"clientId[%s], path[%s], linkName[%s] - Reschedule operation timer, current: [%s], remaining: [%s] secs",
getClientId(),
receivePath,
getReceiveLinkName(),
Instant.now(),
timeoutTracker.remaining().getSeconds()));
}
MessageReceiver.this.scheduleOperationTimer(timeoutTracker);
}
break;
}
}
}
};
this.receiveWork = new ReceiveWork();
this.createAndReceive = new CreateAndReceive();
this.tokenAudience = String.format(ClientConstants.TOKEN_AUDIENCE_FORMAT, underlyingFactory.getHostName(), receivePath);
this.activeClientTokenManager = new ActiveClientTokenManager(
this,
new Runnable() {
@Override
public void run() {
underlyingFactory.getCBSChannel().sendToken(
underlyingFactory.getReactorDispatcher(),
underlyingFactory.getTokenProvider().getToken(tokenAudience, ClientConstants.TOKEN_VALIDITY),
tokenAudience,
new OperationResult() {
@Override
public void onComplete(Void result) {
if (TRACE_LOGGER.isDebugEnabled()) {
TRACE_LOGGER.debug(
String.format(Locale.US,
"clientId[%s], path[%s], linkName[%s] - token renewed",
getClientId(), receivePath, getReceiveLinkName()));
}
}
@Override
public void onError(Exception error) {
if (TRACE_LOGGER.isInfoEnabled()) {
TRACE_LOGGER.info(
String.format(Locale.US,
"clientId[%s], path[%s], linkName[%s], tokenRenewalFailure[%s]",
getClientId(), receivePath, getReceiveLinkName(), error.getMessage()));
}
}
},
(exception) -> {
if (TRACE_LOGGER.isInfoEnabled()) {
TRACE_LOGGER.info(
String.format(Locale.US,
"clientId[%s], path[%s], linkName[%s], tokenRenewalScheduleFailure[%s]",
getClientId(), receivePath, getReceiveLinkName(), exception.getMessage()));
}
});
}
},
ClientConstants.TOKEN_REFRESH_INTERVAL,
this.underlyingFactory);
}
// @param connection Connection on which the MessageReceiver's receive AMQP link need to be created on.
// Connection has to be associated with Reactor before Creating a receiver on it.
public static CompletableFuture create(
final MessagingFactory factory,
final String name,
final String recvPath,
final int prefetchCount,
final ReceiverSettingsProvider settingsProvider) {
MessageReceiver msgReceiver = new MessageReceiver(
factory,
name,
recvPath,
prefetchCount,
settingsProvider);
return msgReceiver.createLink();
}
public String getReceivePath() {
return this.receivePath;
}
private CompletableFuture createLink() {
try {
this.underlyingFactory.scheduleOnReactorThread(new DispatchHandler() {
@Override
public void onEvent() {
MessageReceiver.this.createReceiveLink();
}
});
} catch (IOException | RejectedExecutionException schedulerException) {
this.linkOpen.getWork().completeExceptionally(schedulerException);
}
return this.linkOpen.getWork();
}
private List receiveCore(final int messageCount) {
List returnMessages = null;
Message currentMessage;
while ((currentMessage = this.pollPrefetchQueue()) != null) {
if (returnMessages == null) {
returnMessages = new LinkedList<>();
}
returnMessages.add(currentMessage);
if (returnMessages.size() >= messageCount) {
break;
}
}
return returnMessages;
}
private String getReceiveLinkName() {
return this.receiveLink == null ? "null" : this.receiveLink.getName();
}
public Duration getReceiveTimeout() {
return this.receiveTimeout;
}
public void setReceiveTimeout(final Duration value) {
this.receiveTimeout = value;
}
public CompletableFuture> receive(final int maxMessageCount) {
this.throwIfClosed();
final CompletableFuture> onReceive = new CompletableFuture<>();
if (maxMessageCount <= 0 || maxMessageCount > this.prefetchCount) {
onReceive.completeExceptionally(new IllegalArgumentException(String.format(
Locale.US,
"Entity(%s): maxEventCount(%s) should be a positive number and should be less than prefetchCount(%s)",
this.receivePath, maxMessageCount, this.prefetchCount)));
return onReceive;
}
if (this.shouldScheduleOperationTimeoutTimer()) {
if (TRACE_LOGGER.isDebugEnabled()) {
TRACE_LOGGER.debug(
String.format(Locale.US,
"clientId[%s], path[%s], linkName[%s] - schedule operation timer, current: [%s], remaining: [%s] secs",
this.getClientId(),
this.receivePath,
this.getReceiveLinkName(),
Instant.now(),
this.receiveTimeout.getSeconds()));
}
timer.schedule(this.onOperationTimedout, this.receiveTimeout);
}
pendingReceives.offer(new ReceiveWorkItem(onReceive, receiveTimeout, maxMessageCount));
try {
this.underlyingFactory.scheduleOnReactorThread(this.createAndReceive);
} catch (IOException | RejectedExecutionException schedulerException) {
onReceive.completeExceptionally(schedulerException);
}
return onReceive;
}
@Override
public void onOpenComplete(Exception exception) {
this.creatingLink = false;
this.cancelOpenTimer();
if (exception == null) {
if (this.linkOpen != null && !this.linkOpen.getWork().isDone()) {
this.linkOpen.getWork().complete(this);
}
if (this.getIsClosingOrClosed()) {
return;
}
synchronized (this.errorConditionLock) {
this.lastKnownLinkError = null;
}
this.underlyingFactory.getRetryPolicy().resetRetryCount(this.underlyingFactory.getClientId());
this.nextCreditToFlow = 0;
this.sendFlow(this.prefetchCount - this.prefetchedMessages.size());
if (TRACE_LOGGER.isInfoEnabled()) {
TRACE_LOGGER.info(String.format(Locale.US, "onOpenComplete - clientId[%s], receiverPath[%s], linkName[%s], updated-link-credit[%s], sentCredits[%s]",
this.getClientId(), this.receivePath, this.getReceiveLinkName(), this.receiveLink.getCredit(), this.prefetchCount));
}
} else {
synchronized (this.errorConditionLock) {
this.lastKnownLinkError = exception;
}
if (this.linkOpen != null && !this.linkOpen.getWork().isDone()) {
if ((exception instanceof EventHubException) && ((EventHubException)exception).getIsTransient()) {
final Duration nextRetryInterval = this.underlyingFactory.getRetryPolicy().getNextRetryInterval(
this.getClientId(), exception, this.linkOpen.getTimeoutTracker().remaining());
if (nextRetryInterval != null) {
if (onOpenRetry != null) {
onOpenRetry.accept(this);
}
try {
this.underlyingFactory.scheduleOnReactorThread((int) nextRetryInterval.toMillis(), new DispatchHandler() {
@Override
public void onEvent() {
if (!MessageReceiver.this.getIsClosingOrClosed()
&& (receiveLink == null || receiveLink.getLocalState() == EndpointState.CLOSED || receiveLink.getRemoteState() == EndpointState.CLOSED)) {
createReceiveLink();
underlyingFactory.getRetryPolicy().incrementRetryCount(getClientId());
}
}
});
} catch (IOException | RejectedExecutionException schedulerException) {
if (TRACE_LOGGER.isWarnEnabled()) {
TRACE_LOGGER.warn(
String.format(Locale.US, "clientId[%s], receiverPath[%s], scheduling createLink encountered error: %s",
this.getClientId(), this.receivePath, schedulerException.getLocalizedMessage()));
}
this.cancelOpen(schedulerException);
}
} else {
// no retries left
this.cancelOpen(exception);
}
} else {
// not transient exception
this.cancelOpen(exception);
}
} else {
if (TRACE_LOGGER.isInfoEnabled()) {
TRACE_LOGGER.info(
String.format(Locale.US, "onOpenComplete - clientId[%s], receiverPath[%s], called with error when linkOpen already completed, error [%s]",
this.getClientId(), this.receivePath, this.lastKnownLinkError.toString()));
}
}
}
}
private void cancelOpen(final Exception completionException) {
this.setClosed();
ExceptionUtil.completeExceptionally(this.linkOpen.getWork(), completionException, this);
this.cancelOpenTimer();
}
private void cancelOpenTimer() {
if (this.openTimer != null && !this.openTimer.isCancelled()) {
this.openTimer.cancel(false);
}
}
@Override
public void onReceiveComplete(Delivery delivery) {
int msgSize = delivery.pending();
byte[] buffer = new byte[msgSize];
int read = receiveLink.recv(buffer, 0, msgSize);
Message message = Proton.message();
message.decode(buffer, 0, read);
delivery.settle();
this.prefetchedMessages.add(message);
this.underlyingFactory.getRetryPolicy().resetRetryCount(this.getClientId());
this.receiveWork.onEvent();
}
@Override
public void onError(final Exception exception, final String failingLinkName) {
if ((failingLinkName != null) && (this.receiveLink.getName().compareTo(failingLinkName) != 0)) {
if (TRACE_LOGGER.isWarnEnabled()) {
TRACE_LOGGER.warn(
String.format(Locale.US, "Link error on unknown linkName[%s], onError: %s", failingLinkName, exception.toString()));
}
return;
}
this.prefetchedMessages.clear();
if (this.getIsClosingOrClosed()) {
if (this.closeTimer != null) {
this.closeTimer.cancel(false);
}
this.drainPendingReceives(exception);
this.linkClose.complete(null);
} else {
synchronized (this.errorConditionLock) {
this.lastKnownLinkError = exception == null ? this.lastKnownLinkError : exception;
}
final Exception completionException = exception == null
? new EventHubException(true, String.format(Locale.US,
"Entity(%s): client encountered transient error for unknown reasons, please retry the operation.", this.receivePath))
: exception;
if (TRACE_LOGGER.isWarnEnabled()) {
TRACE_LOGGER.warn(
String.format(Locale.US, "clientId[%s], receiverPath[%s], linkName[%s], onError: %s",
this.getClientId(),
this.receivePath,
this.getReceiveLinkName(),
completionException));
}
this.onOpenComplete(completionException);
final WorkItem> workItem = this.pendingReceives.peek();
final Duration nextRetryInterval = workItem != null && workItem.getTimeoutTracker() != null
? this.underlyingFactory.getRetryPolicy().getNextRetryInterval(this.getClientId(), completionException, workItem.getTimeoutTracker().remaining())
: null;
boolean recreateScheduled = true;
if (nextRetryInterval != null) {
try {
this.underlyingFactory.scheduleOnReactorThread((int) nextRetryInterval.toMillis(), new DispatchHandler() {
@Override
public void onEvent() {
if (!MessageReceiver.this.getIsClosingOrClosed()
&& (receiveLink == null || receiveLink.getLocalState() == EndpointState.CLOSED || receiveLink.getRemoteState() == EndpointState.CLOSED)) {
createReceiveLink();
underlyingFactory.getRetryPolicy().incrementRetryCount(getClientId());
}
}
});
} catch (IOException | RejectedExecutionException ignore) {
recreateScheduled = false;
if (TRACE_LOGGER.isWarnEnabled()) {
TRACE_LOGGER.warn(
String.format(Locale.US, "clientId[%s], receiverPath[%s], linkName[%s], scheduling createLink encountered error: %s",
this.getClientId(),
this.receivePath,
this.getReceiveLinkName(), ignore.getLocalizedMessage()));
}
}
}
if (nextRetryInterval == null || !recreateScheduled) {
this.drainPendingReceives(completionException);
}
}
}
private void drainPendingReceives(final Exception exception) {
WorkItem> workItem;
final boolean shouldReturnNull = (exception == null
|| (exception instanceof EventHubException && ((EventHubException) exception).getIsTransient()));
while ((workItem = this.pendingReceives.poll()) != null) {
final CompletableFuture> future = workItem.getWork();
if (shouldReturnNull) {
future.complete(null);
} else {
ExceptionUtil.completeExceptionally(future, exception, this);
}
}
}
private void scheduleOperationTimer(final TimeoutTracker tracker) {
if (tracker != null) {
timer.schedule(this.onOperationTimedout, tracker.remaining());
}
}
private void createReceiveLink() {
synchronized (this.errorConditionLock) {
if (this.creatingLink) {
if (TRACE_LOGGER.isInfoEnabled()) {
TRACE_LOGGER.info(
String.format(Locale.US,
"clientId[%s], path[%s], operationTimeout[%s], creating a receive link is already in progress",
this.getClientId(), this.receivePath, this.operationTimeout));
}
return;
}
this.creatingLink = true;
}
if (TRACE_LOGGER.isInfoEnabled()) {
TRACE_LOGGER.info(
String.format(Locale.US,
"clientId[%s], path[%s], operationTimeout[%s], creating a receive link",
this.getClientId(), this.receivePath, this.operationTimeout));
}
this.linkCreationTime = Instant.now().toString();
this.scheduleLinkOpenTimeout(TimeoutTracker.create(this.operationTimeout));
final Consumer onSessionOpen = new Consumer() {
@Override
public void accept(Session session) {
// if the MessageReceiver is closed - we no-longer need to create the link
if (MessageReceiver.this.getIsClosingOrClosed()) {
if (TRACE_LOGGER.isInfoEnabled()) {
TRACE_LOGGER.info(
String.format(Locale.US,
"clientId[%s], path[%s], canceling the job of creating a receive link because the receiver was closed",
getClientId(), receivePath));
}
session.close();
return;
}
final Source source = new Source();
source.setAddress(receivePath);
final Map filterMap = MessageReceiver.this.settingsProvider.getFilter(MessageReceiver.this.lastReceivedMessage);
if (filterMap != null) {
source.setFilter(filterMap);
}
final Receiver receiver = session.receiver(TrackingUtil.getLinkName(session));
receiver.setSource(source);
final Target target = new Target();
receiver.setTarget(target);
// use explicit settlement via dispositions (not pre-settled)
receiver.setSenderSettleMode(SenderSettleMode.UNSETTLED);
receiver.setReceiverSettleMode(ReceiverSettleMode.SECOND);
final Map linkProperties = MessageReceiver.this.settingsProvider.getProperties();
if (linkProperties != null) {
receiver.setProperties(linkProperties);
}
final Symbol[] desiredCapabilities = MessageReceiver.this.settingsProvider.getDesiredCapabilities();
if (desiredCapabilities != null) {
receiver.setDesiredCapabilities(desiredCapabilities);
}
final ReceiveLinkHandler handler = new ReceiveLinkHandler(MessageReceiver.this, MessageReceiver.this.getClientId(),
MessageReceiver.this.underlyingFactory.executor);
BaseHandler.setHandler(receiver, handler);
if (MessageReceiver.this.receiveLink != null) {
MessageReceiver.this.underlyingFactory.deregisterForConnectionError(MessageReceiver.this.receiveLink);
}
MessageReceiver.this.underlyingFactory.registerForConnectionError(receiver);
receiver.open();
synchronized (MessageReceiver.this.errorConditionLock) {
MessageReceiver.this.receiveLink = receiver;
}
}
};
final BiConsumer onSessionOpenFailed = new BiConsumer() {
@Override
public void accept(ErrorCondition t, Exception u) {
if (t != null) {
onError((t.getCondition() != null) ? ExceptionUtil.toException(t) : null, null);
} else if (u != null) {
onError(u, null);
}
}
};
this.underlyingFactory.getCBSChannel().sendToken(
this.underlyingFactory.getReactorDispatcher(),
this.underlyingFactory.getTokenProvider().getToken(tokenAudience, ClientConstants.TOKEN_VALIDITY),
tokenAudience,
new OperationResult() {
@Override
public void onComplete(Void result) {
if (MessageReceiver.this.getIsClosingOrClosed()) {
return;
}
underlyingFactory.getSession(
receivePath,
onSessionOpen,
onSessionOpenFailed);
}
@Override
public void onError(Exception error) {
final Exception completionException;
if (error != null && error instanceof AmqpException) {
completionException = ExceptionUtil.toException(((AmqpException) error).getError());
if (completionException != error && completionException.getCause() == null) {
completionException.initCause(error);
}
} else {
completionException = error;
}
MessageReceiver.this.onError(completionException, null);
}
},
(exception) -> {
MessageReceiver.this.onError(exception, null);
});
}
// CONTRACT: message should be delivered to the caller of MessageReceiver.receive() only via Poll on prefetchqueue
private Message pollPrefetchQueue() {
final Message message = this.prefetchedMessages.poll();
if (message != null) {
// message lastReceivedOffset should be up-to-date upon each poll - as recreateLink will depend on this
this.lastReceivedMessage = message;
this.sendFlow(1);
}
return message;
}
private void sendFlow(final int credits) {
// slow down sending the flow - to make the protocol less-chat'y
this.nextCreditToFlow += credits;
if (this.shouldSendFlow()) {
final int tempFlow = this.nextCreditToFlow;
this.receiveLink.flow(tempFlow);
this.nextCreditToFlow = 0;
if (TRACE_LOGGER.isDebugEnabled()) {
TRACE_LOGGER.debug(String.format(Locale.US, "clientId[%s], receiverPath[%s], linkName[%s], updated-link-credit[%s], sentCredits[%s], ThreadId[%s]",
this.getClientId(), this.receivePath, this.getReceiveLinkName(), this.receiveLink.getCredit(), tempFlow, Thread.currentThread().getId()));
}
}
}
private boolean shouldSendFlow() {
return (this.nextCreditToFlow > 0 && this.nextCreditToFlow >= (this.prefetchCount / 2))
|| (this.nextCreditToFlow >= 100);
}
private void scheduleLinkOpenTimeout(final TimeoutTracker timeout) {
// timer to signal a timeout if exceeds the operationTimeout on MessagingFactory
this.openTimer = timer.schedule(
new Runnable() {
public void run() {
creatingLink = false;
if (!linkOpen.getWork().isDone()) {
final Exception lastReportedLinkError;
synchronized (errorConditionLock) {
lastReportedLinkError = MessageReceiver.this.lastKnownLinkError;
}
final Exception operationTimedout = new TimeoutException(
String.format(Locale.US, "Open operation on entity(%s) timed out at %s.",
MessageReceiver.this.receivePath, ZonedDateTime.now()),
lastReportedLinkError);
if (TRACE_LOGGER.isWarnEnabled()) {
TRACE_LOGGER.warn(
String.format(Locale.US, "clientId[%s], receiverPath[%s], Open call timed out",
MessageReceiver.this.getClientId(), MessageReceiver.this.receivePath), operationTimedout);
}
ExceptionUtil.completeExceptionally(linkOpen.getWork(), operationTimedout, MessageReceiver.this);
setClosed();
}
}
}, timeout.remaining());
this.openTimer.handleAsync(
(unUsed, exception) -> {
if (exception != null
&& exception instanceof Exception
&& !(exception instanceof CancellationException)) {
ExceptionUtil.completeExceptionally(linkOpen.getWork(), (Exception) exception, MessageReceiver.this);
}
return null;
}, this.executor);
}
private void scheduleLinkCloseTimeout(final TimeoutTracker timeout) {
// timer to signal a timeout if exceeds the operationTimeout on MessagingFactory
this.closeTimer = timer.schedule(
new Runnable() {
public void run() {
if (!linkClose.isDone()) {
final Receiver link;
synchronized (errorConditionLock) {
link = MessageReceiver.this.receiveLink;
}
final String originalLinkName = timeout.getContext();
if ((originalLinkName != null) && (originalLinkName.compareTo(link.getName()) != 0)) {
if (TRACE_LOGGER.isWarnEnabled()) {
TRACE_LOGGER.warn(
String.format(Locale.US, "Timeout for old receive link %s, current link is %s, ignoring", originalLinkName, link.getName())
);
}
return;
}
final Exception operationTimedout = new TimeoutException(String.format(Locale.US, "Close operation on Receive Link(%s) timed out at %s",
link.getName(), ZonedDateTime.now()));
if (TRACE_LOGGER.isInfoEnabled()) {
TRACE_LOGGER.info(
String.format(Locale.US, "clientId[%s], receiverPath[%s], linkName[%s], Close call timed out",
MessageReceiver.this.getClientId(), MessageReceiver.this.receivePath, link.getName()),
operationTimedout);
}
ExceptionUtil.completeExceptionally(linkClose, operationTimedout, MessageReceiver.this);
MessageReceiver.this.onError((Exception) null, link.getName());
}
}
}, timeout.remaining());
this.closeTimer.handleAsync(
(unUsed, exception) -> {
if (exception != null && exception instanceof Exception && !(exception instanceof CancellationException)) {
ExceptionUtil.completeExceptionally(linkClose, (Exception) exception, MessageReceiver.this);
}
return null;
}, this.executor);
}
private boolean shouldScheduleOperationTimeoutTimer() {
boolean scheduleTimer = this.operationTimeoutScheduled.getAndIncrement() < MAX_OPERATION_TIMEOUT_SCHEDULED;
if (!scheduleTimer) {
this.operationTimeoutScheduled.decrementAndGet();
}
return scheduleTimer;
}
private void operationTimeoutTimerFired() {
MessageReceiver.this.operationTimeoutScheduled.decrementAndGet();
}
@Override
public void onClose(final ErrorCondition condition, final String errorContext) {
if (this.receiveLink != null) {
this.underlyingFactory.deregisterForConnectionError(MessageReceiver.this.receiveLink);
}
final Exception completionException = (condition != null && condition.getCondition() != null) ? ExceptionUtil.toException(condition) : null;
this.onError(completionException, errorContext);
}
@Override
public ErrorContext getContext() {
final Receiver link;
synchronized (this.errorConditionLock) {
link = this.receiveLink;
}
final boolean isLinkOpened = this.linkOpen != null && this.linkOpen.getWork().isDone();
final String referenceId = link != null && link.getRemoteProperties() != null && link.getRemoteProperties().containsKey(ClientConstants.TRACKING_ID_PROPERTY)
? link.getRemoteProperties().get(ClientConstants.TRACKING_ID_PROPERTY).toString()
: ((link != null) ? link.getName() : null);
final ReceiverContext errorContext = new ReceiverContext(this.underlyingFactory != null ? this.underlyingFactory.getHostName() : null,
this.receivePath,
referenceId,
isLinkOpened ? this.prefetchCount : null,
isLinkOpened && link != null ? link.getCredit() : null,
isLinkOpened && this.prefetchedMessages != null ? this.prefetchedMessages.size() : null);
return errorContext;
}
@Override
protected CompletableFuture onClose() {
if (!this.getIsClosed()) {
try {
this.activeClientTokenManager.cancel();
scheduleLinkCloseTimeout(TimeoutTracker.create(operationTimeout, this.receiveLink.getName()));
this.underlyingFactory.scheduleOnReactorThread(new DispatchHandler() {
@Override
public void onEvent() {
if (receiveLink != null && receiveLink.getLocalState() != EndpointState.CLOSED) {
receiveLink.close();
} else if (receiveLink == null || receiveLink.getRemoteState() == EndpointState.CLOSED) {
if (closeTimer != null && !closeTimer.isCancelled()) {
closeTimer.cancel(false);
}
linkClose.complete(null);
}
}
});
} catch (IOException | RejectedExecutionException schedulerException) {
this.linkClose.completeExceptionally(schedulerException);
}
}
return this.linkClose;
}
@Override
protected Exception getLastKnownError() {
synchronized (this.errorConditionLock) {
return this.lastKnownLinkError;
}
}
private static class ReceiveWorkItem extends WorkItem> {
private final int maxMessageCount;
ReceiveWorkItem(CompletableFuture> completableFuture, Duration timeout, final int maxMessageCount) {
super(completableFuture, timeout);
this.maxMessageCount = maxMessageCount;
}
}
private final class ReceiveWork extends DispatchHandler {
@Override
public void onEvent() {
// If there are prefetched messages, then we check to see if there are any pendingReceives before pulling it
// from the top of the pendingReceives queue.
while (!prefetchedMessages.isEmpty() && !pendingReceives.isEmpty()) {
ReceiveWorkItem pendingReceive = pendingReceives.poll();
CompletableFuture> work = pendingReceive.getWork();
if (work != null && !work.isDone()) {
Collection receivedMessages = receiveCore(pendingReceive.maxMessageCount);
work.complete(receivedMessages);
}
}
}
}
private final class CreateAndReceive extends DispatchHandler {
@Override
public void onEvent() {
receiveWork.onEvent();
if (!MessageReceiver.this.getIsClosingOrClosed()
&& (receiveLink == null || receiveLink.getLocalState() == EndpointState.CLOSED || receiveLink.getRemoteState() == EndpointState.CLOSED)) {
createReceiveLink();
}
}
}
}