com.azure.cosmos.implementation.ObservableHelper Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of azure-cosmos Show documentation

This Package contains Microsoft Azure Cosmos SDK (with Reactive Extension Reactor support) for Azure Cosmos DB SQL API

There is a newer version: 4.60.0

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
package com.azure.cosmos.implementation;

import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.util.concurrent.Callable;

/**
 * While this class is public, but it is not part of our published public APIs.
 * This is meant to be internally used only by our sdk.
 *
 **/
public class ObservableHelper {

    static public  Mono inlineIfPossible(Callable> function, IRetryPolicy retryPolicy) {

        if (retryPolicy == null) {
            // shortcut
            try {
                return function.call();
            } catch (Exception e) {
                return Mono.error(e);
            }
        } else {
            return BackoffRetryUtility.executeRetry(function, retryPolicy);
        }
    }

    static public  Mono inlineIfPossibleAsObs(Callable> function, IRetryPolicy retryPolicy) {

        if (retryPolicy == null) {
            // shortcut
            return Mono.defer(() -> {
                try {
                    return function.call();
                } catch (Exception e) {
                    return Mono.error(e);
                }
            });

        } else {
            return BackoffRetryUtility.executeRetry(() -> function.call(), retryPolicy);
        }
    }

    static public  Flux fluxInlineIfPossibleAsObs(Callable> function, IRetryPolicy retryPolicy) {

        if (retryPolicy == null) {
            // shortcut
            return Flux.defer(() -> {
                try {
                    return function.call();
                } catch (Exception e) {
                    return Flux.error(e);
                }
            });

        } else {
            return BackoffRetryUtility.fluxExecuteRetry(() -> function.call(), retryPolicy);
        }
    }
}