Function convertQueryPlanToJSON

backend/plugin/db/spanner/spanner.go:647–702

convertQueryPlanToJSON converts a Spanner QueryPlan to JSON string.

(plan *sppb.QueryPlan)

// convertQueryPlanToJSON converts a Spanner QueryPlan to JSON string.
func convertQueryPlanToJSON(plan *sppb.QueryPlan) (string, error) {
	if plan == nil {
		return "{}", nil
	}

	queryPlan := QueryPlan{
		PlanNodes: make([]PlanNode, 0, len(plan.PlanNodes)),
	}

	for _, node := range plan.PlanNodes {
		planNode := PlanNode{
			Index:       node.Index,
			Kind:        node.Kind.String(),
			DisplayName: node.DisplayName,
		}

		// Convert child links
		if len(node.ChildLinks) > 0 {
			planNode.ChildLinks = make([]ChildLink, 0, len(node.ChildLinks))
			for _, link := range node.ChildLinks {
				planNode.ChildLinks = append(planNode.ChildLinks, ChildLink{
					ChildIndex: link.ChildIndex,
					Type:       link.Type,
					Variable:   link.Variable,
				})
			}
		}

		// Convert short representation
		if node.ShortRepresentation != nil {
			planNode.ShortRepresentation = &ShortRepresentation{
				Description: node.ShortRepresentation.Description,
				Subqueries:  node.ShortRepresentation.Subqueries,
			}
		}

		// Convert metadata
		if node.Metadata != nil {
			planNode.Metadata = node.Metadata.AsMap()
		}

		// Convert execution stats
		if node.ExecutionStats != nil {
			planNode.ExecutionStats = node.ExecutionStats.AsMap()
		}

		queryPlan.PlanNodes = append(queryPlan.PlanNodes, planNode)
	}

	jsonBytes, err := json.Marshal(queryPlan)
	if err != nil {
		return "", err
	}

	return string(jsonBytes), nil
}