本站源代码
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.

65 lines
1.7KB

  1. // Copyright (c) 2014 Couchbase, Inc.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package query
  15. import (
  16. "github.com/blevesearch/bleve/index"
  17. "github.com/blevesearch/bleve/mapping"
  18. "github.com/blevesearch/bleve/search"
  19. "github.com/blevesearch/bleve/search/searcher"
  20. )
  21. type BoolFieldQuery struct {
  22. Bool bool `json:"bool"`
  23. FieldVal string `json:"field,omitempty"`
  24. BoostVal *Boost `json:"boost,omitempty"`
  25. }
  26. // NewBoolFieldQuery creates a new Query for boolean fields
  27. func NewBoolFieldQuery(val bool) *BoolFieldQuery {
  28. return &BoolFieldQuery{
  29. Bool: val,
  30. }
  31. }
  32. func (q *BoolFieldQuery) SetBoost(b float64) {
  33. boost := Boost(b)
  34. q.BoostVal = &boost
  35. }
  36. func (q *BoolFieldQuery) Boost() float64 {
  37. return q.BoostVal.Value()
  38. }
  39. func (q *BoolFieldQuery) SetField(f string) {
  40. q.FieldVal = f
  41. }
  42. func (q *BoolFieldQuery) Field() string {
  43. return q.FieldVal
  44. }
  45. func (q *BoolFieldQuery) Searcher(i index.IndexReader, m mapping.IndexMapping, options search.SearcherOptions) (search.Searcher, error) {
  46. field := q.FieldVal
  47. if q.FieldVal == "" {
  48. field = m.DefaultSearchField()
  49. }
  50. term := "F"
  51. if q.Bool {
  52. term = "T"
  53. }
  54. return searcher.NewTermSearcher(i, term, field, q.BoostVal.Value(), options)
  55. }
上海开阖软件有限公司 沪ICP备12045867号-1