mirror of https://github.com/go-gitea/gitea
Git with a cup of tea, painless self-hosted git service
Mirror for internal git.with.parts use
https://git.with.parts
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.7 KiB
62 lines
1.7 KiB
// 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
|
|
}
|
|
|