graphql.nadel.engine.transform.query.DynamicServiceResolution.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nadel Show documentation

Nadel is a Java library that combines multiple GrahpQL services together into one API.

There is a newer version: 2024-12-10T04-34-06-f2ee9344

package graphql.nadel.engine.transform.query

import graphql.GraphqlErrorException
import graphql.nadel.Service
import graphql.nadel.engine.util.queryPath
import graphql.nadel.engine.util.toGraphQLErrorException
import graphql.nadel.engine.util.unwrapNonNull
import graphql.nadel.hooks.NadelExecutionHooks
import graphql.nadel.schema.NadelDirectives.dynamicServiceDirectiveDefinition
import graphql.normalized.ExecutableNormalizedField
import graphql.schema.GraphQLInterfaceType
import graphql.schema.GraphQLSchema

internal class DynamicServiceResolution(
    private val engineSchema: GraphQLSchema,
    private val executionHooks: NadelExecutionHooks,
    private val services: List,
) {

    /**
     * Checks if the field needs to have its service dynamically resolved
     */
    fun needsDynamicServiceResolution(
        topLevelField: ExecutableNormalizedField,
    ): Boolean =
        topLevelField.getFieldDefinitions(engineSchema)
            .asSequence()
            .filter {
                it.getAppliedDirective(dynamicServiceDirectiveDefinition.name) != null
            }
            .onEach {
                require(it.type.unwrapNonNull() is GraphQLInterfaceType) {
                    "field annotated with ${dynamicServiceDirectiveDefinition.name} directive is expected to be of GraphQLInterfaceType"
                }
            }
            .any { it != null }

    /**
     * Resolves the service for a field
     */
    fun resolveServiceForField(field: ExecutableNormalizedField): Service {
        val serviceOrError = executionHooks.resolveServiceForField(services, field)
            ?: throw GraphqlErrorException.newErrorException()
                .message("Could not resolve service for field '${field.name}'")
                .path(field.queryPath.segments)
                .build()

        // Either or… but needs to be better
        val error = serviceOrError.error
        if (error != null) {
            throw error.toGraphQLErrorException()
        } else {
            return requireNotNull(serviceOrError.service)
        }
    }
}