Skip to content

Commit 2735d42

Browse files
committed
fix(codegen): pass correct env variable to buildEdge
1 parent a703988 commit 2735d42

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

graphql-kotlin-toolkit-codegen/src/main/kotlin/com/auritylab/graphql/kotlin/toolkit/codegen/generator/fieldResolver/PaginationFieldResolverGenerator.kt

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -89,12 +89,13 @@ internal class PaginationFieldResolverGenerator(
8989

9090
getFunSpec.addStatement("val map = env.arguments")
9191
getFunSpec.addStatement(
92-
"val result = resolve(${resolveArgs}env = %T(env, %M(map)))",
92+
"val internalEnv = %T(env, %M(map))",
9393
generatedMapper.getFieldResolverEnvironment(container, field),
9494
generatedMapper.getPaginationInfoBuilderMemberName()
9595
)
96+
getFunSpec.addStatement("val result = resolve(${resolveArgs}env = internalEnv)")
9697

97-
getFunSpec.addStatement("val edges = result.data.map { _buildEdge(it, env) }")
98+
getFunSpec.addStatement("val edges = result.data.map { _buildEdge(it, internalEnv) }")
9899
getFunSpec.addStatement("val pageInfo = _buildPageInfo(result.hasNextPage, result.hasPreviousPage, edges.first().cursor, edges.last().cursor)")
99100
getFunSpec.addStatement("return _buildConnection(edges, pageInfo)")
100101
}
@@ -179,7 +180,12 @@ internal class PaginationFieldResolverGenerator(
179180
.addParameter("data", unwrappedReturnKotlinType)
180181
.addParameter("env", generatedMapper.getFieldResolverEnvironment(container, field))
181182
.returns(generatedMapper.getPaginationEdgeClassName().parameterizedBy(unwrappedReturnKotlinType))
182-
.addCode(CodeBlock.of("return %T(data, resolveCursor(data, env))", generatedMapper.getPaginationEdgeClassName()))
183+
.addCode(
184+
CodeBlock.of(
185+
"return %T(data, resolveCursor(data, env))",
186+
generatedMapper.getPaginationEdgeClassName()
187+
)
188+
)
183189
.build()
184190

185191
private val buildConnectionFunSpec = FunSpec.builder("_buildConnection")

0 commit comments

Comments
 (0)