org.finos.tracdap.common.data.IExecutionContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tracdap-lib-data Show documentation

TRAC D.A.P. data library, interfaces and core functionality for working with primary data

The newest version!

/*
 * Licensed to the Fintech Open Source Foundation (FINOS) under one or
 * more contributor license agreements. See the NOTICE file distributed
 * with this work for additional information regarding copyright ownership.
 * FINOS licenses this file to you 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 org.finos.tracdap.common.data;

import io.netty.util.concurrent.OrderedEventExecutor;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionException;
import java.util.concurrent.CompletionStage;


public interface IExecutionContext {

    OrderedEventExecutor eventLoopExecutor();

    default  CompletableFuture toContext(CompletionStage promise) {

        var ctxPromise = new CompletableFuture();
        useContext(promise, ctxPromise);
        return ctxPromise;
    }

    default  CompletableFuture fromContext(CompletableFuture ctxPromise) {

        var promise = new CompletableFuture();
        useContext(promise, ctxPromise);
        return promise;
    }

    private  void useContext(
            CompletionStage promise,
            CompletableFuture ctxPromise) {

        promise.whenComplete((result, error) -> {

            var wrappedError = (error != null && !(error instanceof CompletionException))
                    ? new CompletionException(error)
                    : error;

            if (eventLoopExecutor().inEventLoop()) {

                if (error != null)
                    ctxPromise.completeExceptionally(wrappedError);
                else
                    ctxPromise.complete(result);
            }
            else {

                if (error != null)
                    eventLoopExecutor().execute(() -> ctxPromise.completeExceptionally(wrappedError));
                else
                    eventLoopExecutor().execute(() -> ctxPromise.complete(result));
            }
        });
    }
}