Function collectCTENames

backend/plugin/parser/tidb/backup.go:410–416

collectCTENames returns the set of CTE (WITH clause) names that precede the statement, lower-cased for case-insensitive matching. omni does not attach UPDATE/DELETE CTEs to the statement node, so we parse the text before the statement Loc as a WITH clause.

(fullSQL string, stmtLoc ast.Loc)

// UPDATE/DELETE CTEs to the statement node, so we parse the text before the
// statement Loc as a WITH clause.
func collectCTENames(fullSQL string, stmtLoc ast.Loc) map[string]bool {
	result := make(map[string]bool)
	for _, cte := range parseCTEPrefix(fullSQL, stmtLoc) {
		result[strings.ToLower(cte.Name)] = true
	}
	return result
}

// isCTERef reports whether ref resolves to a CTE. A CTE reference is always
// unqualified, so a schema-qualified physical table (db.test) is never a CTE