graphql.nadel.engine.transform.hydration.NadelHydrationFieldsBuilder.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
Show all versions of nadel Show documentation
Nadel is a Java library that combines multiple GrahpQL services together into one API.
package graphql.nadel.engine.transform.hydration
import graphql.nadel.Service
import graphql.nadel.engine.blueprint.NadelBatchHydrationFieldInstruction
import graphql.nadel.engine.blueprint.NadelGenericHydrationInstruction
import graphql.nadel.engine.blueprint.NadelHydrationFieldInstruction
import graphql.nadel.engine.blueprint.NadelOverallExecutionBlueprint
import graphql.nadel.engine.blueprint.hydration.NadelHydrationActorInputDef
import graphql.nadel.engine.transform.GraphQLObjectTypeName
import graphql.nadel.engine.transform.artificial.NadelAliasHelper
import graphql.nadel.engine.transform.hydration.batch.NadelBatchHydrationInputBuilder
import graphql.nadel.engine.transform.hydration.batch.NadelBatchHydrationObjectIdFieldBuilder.makeObjectIdFields
import graphql.nadel.engine.transform.query.NFUtil
import graphql.nadel.engine.transform.result.json.JsonNode
import graphql.nadel.engine.util.deepClone
import graphql.nadel.engine.util.resolveObjectTypes
import graphql.nadel.engine.util.toBuilder
import graphql.nadel.engine.util.unwrapAll
import graphql.nadel.hooks.NadelExecutionHooks
import graphql.normalized.ExecutableNormalizedField
import graphql.normalized.NormalizedInputValue
internal object NadelHydrationFieldsBuilder {
fun makeActorQueries(
instruction: NadelHydrationFieldInstruction,
aliasHelper: NadelAliasHelper,
fieldToHydrate: ExecutableNormalizedField,
parentNode: JsonNode,
executionBlueprint: NadelOverallExecutionBlueprint,
): List {
return NadelHydrationInputBuilder.getInputValues(
instruction = instruction,
aliasHelper = aliasHelper,
fieldToHydrate = fieldToHydrate,
parentNode = parentNode,
).map { args ->
makeActorQueries(
instruction = instruction,
fieldArguments = args,
fieldChildren = deepClone(fields = fieldToHydrate.children),
executionBlueprint = executionBlueprint,
)
}
}
fun makeBatchActorQueries(
executionBlueprint: NadelOverallExecutionBlueprint,
instruction: NadelBatchHydrationFieldInstruction,
aliasHelper: NadelAliasHelper,
hydratedField: ExecutableNormalizedField,
parentNodes: List,
hooks: NadelExecutionHooks,
userContext: Any?,
): List {
val argBatches = NadelBatchHydrationInputBuilder.getInputValueBatches(
instruction = instruction,
aliasHelper = aliasHelper,
hydrationField = hydratedField,
parentNodes = parentNodes,
hooks = hooks,
userContext = userContext,
)
return makeBatchActorQueries(
executionBlueprint = executionBlueprint,
instruction = instruction,
aliasHelper = aliasHelper,
hydratedField = hydratedField,
argBatches = argBatches,
)
}
fun makeBatchActorQueries(
executionBlueprint: NadelOverallExecutionBlueprint,
instruction: NadelBatchHydrationFieldInstruction,
aliasHelper: NadelAliasHelper,
hydratedField: ExecutableNormalizedField,
argBatches: List>,
): List {
val actorFieldOverallObjectTypeNames = getActorFieldOverallObjectTypenames(instruction, executionBlueprint)
val fieldChildren = deepClone(fields = hydratedField.children)
.mapNotNull { childField ->
val objectTypesAreNotReturnedByActorField =
actorFieldOverallObjectTypeNames.none { it in childField.objectTypeNames }
if (objectTypesAreNotReturnedByActorField) {
null
} else {
childField.toBuilder()
.clearObjectTypesNames()
.objectTypeNames(childField.objectTypeNames.filter { it in actorFieldOverallObjectTypeNames })
.build()
}
}
.let { children ->
children + makeObjectIdFields(executionBlueprint, aliasHelper, instruction)
}
return argBatches.map { argBatch ->
makeActorQueries(
instruction = instruction,
fieldArguments = argBatch.mapKeys { (inputDef: NadelHydrationActorInputDef) -> inputDef.name },
fieldChildren = fieldChildren,
executionBlueprint = executionBlueprint,
)
}
}
private fun getActorFieldOverallObjectTypenames(
instruction: NadelBatchHydrationFieldInstruction,
executionBlueprint: NadelOverallExecutionBlueprint,
): Set {
val overallTypeName = instruction.actorFieldDef.type.unwrapAll().name
val overallType = executionBlueprint.engineSchema.getType(overallTypeName)
?: error("Unable to find overall type $overallTypeName")
val actorFieldOverallObjectTypes = resolveObjectTypes(executionBlueprint.engineSchema, overallType) { type ->
error("Unable to resolve to object type: $type")
}
return actorFieldOverallObjectTypes
.asSequence()
.map { it.name }
.toSet()
}
fun makeRequiredSourceFields(
service: Service,
executionBlueprint: NadelOverallExecutionBlueprint,
aliasHelper: NadelAliasHelper,
objectTypeName: GraphQLObjectTypeName,
instructions: List,
): List {
val underlyingTypeName = executionBlueprint.getUnderlyingTypeName(service, overallTypeName = objectTypeName)
val underlyingObjectType = service.underlyingSchema.getObjectType(underlyingTypeName)
?: error("No underlying object type")
return instructions
.asSequence()
.flatMap { it.sourceFields }
.map {
aliasHelper.toArtificial(
NFUtil.createField(
schema = service.underlyingSchema,
parentType = underlyingObjectType,
queryPathToField = it,
fieldArguments = emptyMap(),
fieldChildren = emptyList(), // This must be a leaf node
),
)
}
.toList()
}
private fun makeActorQueries(
instruction: NadelGenericHydrationInstruction,
fieldArguments: Map,
fieldChildren: List,
executionBlueprint: NadelOverallExecutionBlueprint,
): ExecutableNormalizedField {
return NFUtil.createField(
schema = executionBlueprint.engineSchema,
parentType = executionBlueprint.engineSchema.queryType,
queryPathToField = instruction.queryPathToActorField,
fieldArguments = fieldArguments,
fieldChildren = fieldChildren,
)
}
}