Function GetTiDBAST

backend/plugin/parser/tidb/ast.go:39–50

GetTiDBAST extracts the TiDB AST from a base.AST. Returns the AST and true if it is a TiDB AST, nil and false otherwise. Also checks for PingCapASTProvider implementations (e.g. *OmniAST during the advisor migration), which lazily build a native pingcap AST on demand.

(a base.AST)

// Also checks for PingCapASTProvider implementations (e.g. *OmniAST during
// the advisor migration), which lazily build a native pingcap AST on demand.
func GetTiDBAST(a base.AST) (*AST, bool) {
	if a == nil {
		return nil, false
	}
	if tidbAST, ok := a.(*AST); ok {
		return tidbAST, true
	}
	if provider, ok := a.(PingCapASTProvider); ok {
		return provider.AsPingCapAST()
	}
	return nil, false
}