Курнат Андрей
36 lines
803 B
Go
36 lines
803 B
Go
package convert
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/qsfera/server/pkg/kql"
|
|
"github.com/qsfera/server/services/search/pkg/opensearch/internal/osu"
|
|
)
|
|
|
|
var (
|
|
ErrUnsupportedNodeType = fmt.Errorf("unsupported node type")
|
|
)
|
|
|
|
func KQLToOpenSearchBoolQuery(kqlQuery string) (*osu.BoolQuery, error) {
|
|
kqlAst, err := kql.Builder{}.Build(kqlQuery)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to build query: %w", err)
|
|
}
|
|
|
|
kqlNodes, err := ExpandKQL(kqlAst.Nodes)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to expand KQL AST nodes: %w", err)
|
|
}
|
|
|
|
builder, err := TranspileKQLToOpenSearch(kqlNodes)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to compile query: %w", err)
|
|
}
|
|
|
|
if q, ok := builder.(*osu.BoolQuery); !ok {
|
|
return osu.NewBoolQuery().Must(builder), nil
|
|
} else {
|
|
return q, nil
|
|
}
|
|
}
|