Function extractColumns

pkg/gosqlx/testing/testing.go:378–404

extractColumns extracts column names from SELECT statements

(astNode *ast.AST)

// extractColumns extracts column names from SELECT statements
func extractColumns(astNode *ast.AST) []string {
	if astNode == nil {
		return nil
	}

	columns := make(map[string]bool)

	for _, stmt := range astNode.Statements {
		if selectStmt, ok := stmt.(*ast.SelectStatement); ok {
			for _, col := range selectStmt.Columns {
				extractColumnNames(col, columns)
			}
		}
	}

	// Convert map to slice
	result := make([]string, 0, len(columns))
	for col := range columns {
		// Skip wildcard
		if col == "*" {
			continue
		}
		result = append(result, col)
	}

	return result
}

// extractColumnNames extracts column names from expressions
func extractColumnNames(expr ast.Expression, columns map[string]bool) {