com.alibaba.dashscope.protocol.RetryWithDelay Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dashscope-sdk-java Show documentation

The newest version!

// Copyright (c) Alibaba, Inc. and its affiliates.

package com.alibaba.dashscope.protocol;

import io.reactivex.Observable;
import io.reactivex.functions.Function;
import java.util.concurrent.TimeUnit;

public class RetryWithDelay implements Function, Observable> {
  private final int maxRetries;
  private final int retryDelayMillis;
  private int retryCount;

  public RetryWithDelay(final int maxRetries, final int retryDelayMillis) {
    this.maxRetries = maxRetries;
    this.retryDelayMillis = retryDelayMillis;
    this.retryCount = 0;
  }

  @Override
  public Observable apply(Observable attempts) {
    return attempts.flatMap(
        new Function>() {
          @Override
          public Observable apply(Throwable throwable) {
            if (++retryCount < maxRetries) {
              // When this Observable calls onNext, the original
              // Observable will be retried (i.e. re-subscribed).
              return Observable.timer(retryDelayMillis, TimeUnit.MILLISECONDS);
            }

            // Max retries hit. Just pass the error along.
            return Observable.error(throwable);
          }
        });
  }
}

// observable.retryWhen(new RetryWithDelay(3, 2000));