1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162 |
- // Copyright 2023 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package bleve
-
- import (
- "code.gitea.io/gitea/modules/optional"
-
- "github.com/blevesearch/bleve/v2"
- "github.com/blevesearch/bleve/v2/search/query"
- )
-
- // NumericEqualityQuery generates a numeric equality query for the given value and field
- func NumericEqualityQuery(value int64, field string) *query.NumericRangeQuery {
- f := float64(value)
- tru := true
- q := bleve.NewNumericRangeInclusiveQuery(&f, &f, &tru, &tru)
- q.SetField(field)
- return q
- }
-
- // MatchPhraseQuery generates a match phrase query for the given phrase, field and analyzer
- func MatchPhraseQuery(matchPhrase, field, analyzer string) *query.MatchPhraseQuery {
- q := bleve.NewMatchPhraseQuery(matchPhrase)
- q.FieldVal = field
- q.Analyzer = analyzer
- return q
- }
-
- // PrefixQuery generates a match prefix query for the given prefix and field
- func PrefixQuery(matchPrefix, field string) *query.PrefixQuery {
- q := bleve.NewPrefixQuery(matchPrefix)
- q.FieldVal = field
- return q
- }
-
- // BoolFieldQuery generates a bool field query for the given value and field
- func BoolFieldQuery(value bool, field string) *query.BoolFieldQuery {
- q := bleve.NewBoolFieldQuery(value)
- q.SetField(field)
- return q
- }
-
- func NumericRangeInclusiveQuery(min, max optional.Option[int64], field string) *query.NumericRangeQuery {
- var minF, maxF *float64
- var minI, maxI *bool
- if min.Has() {
- minF = new(float64)
- *minF = float64(min.Value())
- minI = new(bool)
- *minI = true
- }
- if max.Has() {
- maxF = new(float64)
- *maxF = float64(max.Value())
- maxI = new(bool)
- *maxI = true
- }
- q := bleve.NewNumericRangeInclusiveQuery(minF, maxF, minI, maxI)
- q.SetField(field)
- return q
- }
|