com.microsoft.azure.servicebus.primitives.AsyncUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of azure-servicebus Show documentation
Java library for Azure Service Bus
There is a newer version: 3.6.7
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package com.microsoft.azure.servicebus.primitives;

import java.util.concurrent.Callable;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;

// To complete futures using a different thread. Otherwise every future is completed on the single reactor thread
// which badly affects perf and a client can potentially kill the thread or lock the thread.
class AsyncUtil {

    public static  boolean completeFutureAndGetStatus(CompletableFuture future, T result) {
        try {
            return MessagingFactory.INTERNAL_THREAD_POOL.submit(new CompleteCallable<>(future, result)).get();
        } catch (InterruptedException | ExecutionException e) {
            e.printStackTrace();
            return false;
        }
    }

    public static  void completeFuture(CompletableFuture future, T result) {
        MessagingFactory.INTERNAL_THREAD_POOL.submit(new CompleteCallable<>(future, result));
    }

    public static  boolean completeFutureExceptionallyAndGetStatus(CompletableFuture future, Throwable exception) {
        try {
            return MessagingFactory.INTERNAL_THREAD_POOL.submit(new CompleteExceptionallyCallable<>(future, exception)).get();
        } catch (InterruptedException | ExecutionException e) {
            e.printStackTrace();
            return false;
        }
    }

    public static  void completeFutureExceptionally(CompletableFuture future, Throwable exception) {
        MessagingFactory.INTERNAL_THREAD_POOL.submit(new CompleteExceptionallyCallable<>(future, exception));
    }

    public static void run(Runnable runnable) {
        MessagingFactory.INTERNAL_THREAD_POOL.submit(runnable);
    }

    private static class CompleteCallable implements Callable {
        private CompletableFuture future;
        private T result;

        CompleteCallable(CompletableFuture future, T result) {
            this.future = future;
            this.result = result;
        }

        @Override
        public Boolean call() throws Exception {
            return this.future.complete(this.result);
        }
    }

    private static class CompleteExceptionallyCallable implements Callable {
        private CompletableFuture future;
        private Throwable exception;

        CompleteExceptionallyCallable(CompletableFuture future, Throwable exception) {
            this.future = future;
            this.exception = exception;
        }

        @Override
        public Boolean call() throws Exception {
            return this.future.completeExceptionally(this.exception);
        }
    }
}