Function GetQuerySpan

backend/plugin/parser/elasticsearch/query_span.go:18–49

GetQuerySpan returns the query span for an ElasticSearch REST API request.

(
	_ context.Context,
	_ base.GetQuerySpanContext,
	stmt base.Statement,
	_, _ string,
	_ bool,
)

// GetQuerySpan returns the query span for an ElasticSearch REST API request.
func GetQuerySpan(
	_ context.Context,
	_ base.GetQuerySpanContext,
	stmt base.Statement,
	_, _ string,
	_ bool,
) (*base.QuerySpan, error) {
	omniSpan, err := es.GetQuerySpan(stmt.Text)
	if err != nil {
		return nil, err
	}
	if omniSpan == nil {
		return &base.QuerySpan{Type: base.QueryTypeUnknown}, nil
	}

	span := &base.QuerySpan{
		Type: convertQueryType(omniSpan.Type),
	}

	if omniSpan.ElasticsearchAnalysis != nil {
		span.ElasticsearchAnalysis = convertAnalysis(omniSpan.ElasticsearchAnalysis)
	}

	if omniSpan.PredicatePaths != nil {
		span.PredicatePaths = make(map[string]*base.PathAST, len(omniSpan.PredicatePaths))
		for key, p := range omniSpan.PredicatePaths {
			span.PredicatePaths[key] = convertPathAST(p)
		}
	}

	return span, nil
}

// convertPathAST converts an omni PathAST to a bytebase base.PathAST.
func convertPathAST(p *es.PathAST) *base.PathAST {