com.expediagroup.beekeeper.scheduler.apiary.messaging.RetryingMessageReader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of beekeeper-scheduler-apiary Show documentation

There is a newer version: 3.5.700

Show newest version

/**
 * Copyright (C) ${license.git.copyrightYears} Expedia, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.expediagroup.beekeeper.scheduler.apiary.messaging;

import static java.lang.String.format;

import java.io.IOException;
import java.util.Optional;

import org.springframework.retry.annotation.Backoff;
import org.springframework.retry.annotation.Recover;
import org.springframework.retry.annotation.Retryable;

import com.amazonaws.AmazonClientException;

import com.expedia.apiary.extensions.receiver.common.messaging.MessageEvent;
import com.expedia.apiary.extensions.receiver.common.messaging.MessageReader;
import com.expedia.apiary.extensions.receiver.sqs.messaging.SqsMessageProperty;

import com.expediagroup.beekeeper.core.error.BeekeeperException;

public class RetryingMessageReader implements MessageReader {

  private static final long INITIAL_DELAY_MS = 1000;
  private static final int BACKOFF_MULTIPLIER = 2;
  private static final int MAX_ATTEMPTS = 4;
  private static final String UNKNOWN_HANDLE = "unknown";

  private final MessageReader delegate;

  public RetryingMessageReader(MessageReader delegate) {
    this.delegate = delegate;
  }

  @Override
  @Retryable(value = { AmazonClientException.class },
      maxAttempts = MAX_ATTEMPTS,
      backoff = @Backoff(delay = INITIAL_DELAY_MS, multiplier = BACKOFF_MULTIPLIER))
  public Optional read() {
    return delegate.read();
  }

  @Override
  @Retryable(include = { AmazonClientException.class },
      maxAttempts = MAX_ATTEMPTS,
      backoff = @Backoff(delay = INITIAL_DELAY_MS, multiplier = BACKOFF_MULTIPLIER))
  public void delete(MessageEvent messageEvent) {
    delegate.delete(messageEvent);
  }

  @Override
  public void close() throws IOException {
    delegate.close();
  }

  @Recover
  public Optional recoverRead(AmazonClientException e) {
    throw new BeekeeperException(format("Error reading from queue after %s attempts.", MAX_ATTEMPTS), e);
  }

  @Recover
  public void recoverDelete(AmazonClientException e, MessageEvent messageEvent) {
    throw new BeekeeperException(format("Error deleting message %s from queue after %s attempts.",
        messageEvent.getMessageProperties()
            .getOrDefault(SqsMessageProperty.SQS_MESSAGE_RECEIPT_HANDLE, UNKNOWN_HANDLE), MAX_ATTEMPTS), e);
  }
}