@@ -470,7 +470,7 @@ class CodeGen(schema: Schema) {
470
470
471
471
s """ /** ${adjacentNode.customStepDoc.getOrElse(" " )}
472
472
| * Traverse to ${neighbor.name} via ${adjacentNode.viaEdge.name}.
473
- | */
473
+ | */ ${docAnnotationMaybe(adjacentNode.customStepDoc)}
474
474
|def $accessorName: ${fullScalaType(neighbor, cardinality)} =
475
475
| $edgeAccessorName.collectAll[ ${neighbor.className}] $appendix""" .stripMargin
476
476
}
@@ -896,11 +896,15 @@ class CodeGen(schema: Schema) {
896
896
val nodeDelegators = neighborInfo.nodeInfos.collect {
897
897
case neighborNodeInfo if ! neighborNodeInfo.isInherited =>
898
898
val accessorNameForNode = accessorName(neighborNodeInfo)
899
- s " def $accessorNameForNode: ${neighborNodeInfo.returnType} = get(). $accessorNameForNode"
899
+ s """ /** ${neighborNodeInfo.customStepDoc.getOrElse(" " )}
900
+ | * Traverse to ${neighborNodeInfo.neighborNode.name} via ${neighborNodeInfo.edge.name}.
901
+ | */ ${docAnnotationMaybe(neighborNodeInfo.customStepDoc)}
902
+ |def $accessorNameForNode: ${neighborNodeInfo.returnType} = get(). $accessorNameForNode""" .stripMargin
900
903
}.mkString(" \n " )
901
904
902
905
s """ def $edgeAccessorName = get(). $edgeAccessorName
903
906
|override def _ $edgeAccessorName = get()._ $edgeAccessorName
907
+ |
904
908
| $nodeDelegators
905
909
| """ .stripMargin
906
910
}.mkString(" \n " )
@@ -1162,7 +1166,7 @@ class CodeGen(schema: Schema) {
1162
1166
}
1163
1167
s """ /** ${customStepDoc.getOrElse(" " )}
1164
1168
| * Traverse to ${neighbor.name} via ${viaEdge.name} - this relationship was given a customStepName in the schema.
1165
- | */
1169
+ | */ ${docAnnotationMaybe(customStepDoc)}
1166
1170
|def $customStepName: Traversal[ ${neighbor.className}] =
1167
1171
| traversal. $mapOrFlatMap(_. $customStepName)
1168
1172
| """ .stripMargin
0 commit comments