Function ExtractFunctions

pkg/gosqlx/extract.go:369–383

ExtractFunctions extracts all function calls from an AST. This function traverses the AST and collects all function names, including: - Aggregate functions (COUNT, SUM, AVG, etc.) - Window functions (ROW_NUMBER, RANK, etc.) - Scalar functions (UPPER, LOWER, NOW, etc.) Returns a deduplicated slice

(astNode *ast.AST)

//	functions := gosqlx.ExtractFunctions(ast)
//	// functions = ["COUNT", "UPPER"]
func ExtractFunctions(astNode *ast.AST) []string {
	if astNode == nil {
		return nil
	}

	collector := &functionCollector{
		functions: make(map[string]bool),
	}

	for _, stmt := range astNode.Statements {
		collector.collectFromNode(stmt)
	}

	return collector.toSlice()
}

// tableCollector collects table names from AST nodes
type tableCollector struct {