com.expediagroup.graphql.generator.federation.extensions.CompletableFutureExtensions.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of graphql-kotlin-federation Show documentation

Federated GraphQL schema generator

The newest version!

/*
 * Copyright 2023 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
 *
 *     https://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.graphql.generator.federation.extensions

import java.util.concurrent.CompletableFuture

/**
 * Returns a [CompletableFuture] that completes when all the input futures have completed,
 * with a list of the resolved values obtained from the completed futures.
 * If any of the input futures complete exceptionally, then the returned [CompletableFuture] also completes exceptionally
 * with a [java.util.concurrent.CompletionException] holding the exception as its cause.
 *
 * @return a [CompletableFuture] that completes with a list of resolved values.
 */
internal fun  List>.joinAll(): CompletableFuture> =
    CompletableFuture.allOf(
        *toTypedArray()
    ).thenApply {
        map(CompletableFuture::join)
    }

/**
 * Returns a [CompletableFuture] that completes when all the input futures have completed,
 * with a list of [Result] objects that indicate whether each future completed successfully or with an error.
 * If a future completed with an error, the corresponding [Result] object will contain the exception that was thrown.
 *
 * @return a [CompletableFuture] that completes with a list of [Result] objects.
 */
@Suppress("TooGenericExceptionCaught")
internal fun  List>.allSettled(): CompletableFuture>> {
    val resultFutures = map { future ->
        future.thenApply { result -> Result.success(result) }
            .exceptionally { exception -> Result.failure(exception) }
    }

    return resultFutures.joinAll()
}