Function GetQuerySpan

backend/plugin/parser/base/interface.go:209–243

GetQuerySpan gets the span of a query from pre-split statements. The interface will return the query spans with non-critical errors, or return an error if the query is invalid. Callers should split the SQL first using SplitMultiSQL and pass the resulting []Statement.

(ctx context.Context, gCtx GetQuerySpanContext, engine storepb.Engine, statements []Statement, database, schema string, ignoreCaseSensitive bool)

// The interface will return the query spans with non-critical errors, or return an error if the query is invalid.
// Callers should split the SQL first using SplitMultiSQL and pass the resulting []Statement.
func GetQuerySpan(ctx context.Context, gCtx GetQuerySpanContext, engine storepb.Engine, statements []Statement, database, schema string, ignoreCaseSensitive bool) ([]*QuerySpan, error) {
	f, ok := spans[engine]
	if !ok {
		return nil, nil
	}
	gCtx.Engine = engine
	gCtx.TempTables = make(map[string]*PhysicalTable)
	var results []*QuerySpan
	var nonEmptyStatements []Statement
	for _, stmt := range statements {
		if stmt.Empty {
			continue
		}
		nonEmptyStatements = append(nonEmptyStatements, stmt)
		result, err := f(ctx, gCtx, stmt, database, schema, ignoreCaseSensitive)
		if err != nil {
			// Try to unwrap the error to see if it's a ResourceNotFoundError to decrease the error noise.
			// TODO(d): remove resource not found error checks.
			var resourceNotFound *ResourceNotFoundError
			if errors.As(err, &resourceNotFound) {
				return nil, resourceNotFound
			}
			var typeNotSupported *TypeNotSupportedError
			if errors.As(err, &typeNotSupported) {
				return nil, typeNotSupported
			}
			return nil, err
		}
		results = append(results, result)
	}
	if engine == storepb.Engine_MSSQL {
		tsqlRecognizeExplainType(results, nonEmptyStatements)
	}
	return results, nil
}

// RegisterTransformDMLToSelect registers the transformDMLToSelect function for the engine.
func RegisterTransformDMLToSelect(engine storepb.Engine, f TransformDMLToSelectFunc) {