graphql.nadel.engine.plan.NadelExecutionPlan.kt Maven / Gradle / Ivy

Go to download
package graphql.nadel.engine.plan

import graphql.nadel.Service
import graphql.nadel.engine.transform.NadelTransform
import graphql.normalized.ExecutableNormalizedField

internal typealias AnyNadelExecutionPlanStep = NadelExecutionPlan.Step

data class NadelExecutionPlan(
    // this is a map for overall Fields
    val transformationSteps: Map>,
) {
    data class Step(
        val service: Service,
        val field: ExecutableNormalizedField,
        val transform: NadelTransform,
        val state: T,
    )

    /**
     * Creates and returns a new [NadelExecutionPlan] that is a merging of `this` plan
     * and the [other] plan.
     */
    fun merge(other: NadelExecutionPlan): NadelExecutionPlan {
        val newSteps = transformationSteps.toMutableMap()
        other.transformationSteps.forEach { (field, steps) ->
            newSteps.compute(field) { _, oldSteps ->
                oldSteps?.let { it + steps } ?: steps
            }
        }

        return copy(transformationSteps = newSteps)
    }
}