f8fdfebb6c
- TermSearcher has alternate constructor if term is []byte, this can avoid copying in some cases. TermScorer updated to accept []byte term. Also removed a few struct fields which were not being used. - New MultiTermSearcher searches for documents containing any of a list of terms. Current implementation simply uses DisjunctionSearcher. - Several other searcher constructors now simply build a list of terms and then delegate to the MultiTermSearcher - NewPrefixSearcher - NewRegexpSearcher - NewFuzzySearcher - NewNumericRangeSearcher - NewGeoBoundingBoxSearcher and NewGeoPointDistanceSearcher make use of the MultiTermSearcher internally, and follow the pattern of returning an existing search.Searcher, as opposed to their own wrapping struct. - Callback filter functions used in NewGeoBoundingBoxSearcher and NewGeoPointDistanceSearcher have been extracted into separate functions which makes the code much easier to read.
81 lines
2.4 KiB
Go
81 lines
2.4 KiB
Go
// Copyright (c) 2015 Couchbase, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package searcher
|
|
|
|
import (
|
|
"regexp"
|
|
|
|
"github.com/blevesearch/bleve/index"
|
|
"github.com/blevesearch/bleve/search"
|
|
)
|
|
|
|
// NewRegexpSearcher creates a searcher which will match documents that
|
|
// contain terms which match the pattern regexp. The match must be EXACT
|
|
// matching the entire term. The provided regexp SHOULD NOT start with ^
|
|
// or end with $ as this can intefere with the implementation. Separately,
|
|
// matches will be checked to ensure they match the entire term.
|
|
func NewRegexpSearcher(indexReader index.IndexReader, pattern *regexp.Regexp,
|
|
field string, boost float64, options search.SearcherOptions) (
|
|
search.Searcher, error) {
|
|
|
|
prefixTerm, complete := pattern.LiteralPrefix()
|
|
var candidateTerms []string
|
|
if complete {
|
|
// there is no pattern
|
|
candidateTerms = []string{prefixTerm}
|
|
} else {
|
|
var err error
|
|
candidateTerms, err = findRegexpCandidateTerms(indexReader, pattern, field,
|
|
prefixTerm)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return NewMultiTermSearcher(indexReader, candidateTerms, field, boost,
|
|
options)
|
|
}
|
|
|
|
func findRegexpCandidateTerms(indexReader index.IndexReader,
|
|
pattern *regexp.Regexp, field, prefixTerm string) (rv []string, err error) {
|
|
rv = make([]string, 0)
|
|
var fieldDict index.FieldDict
|
|
if len(prefixTerm) > 0 {
|
|
fieldDict, err = indexReader.FieldDictPrefix(field, []byte(prefixTerm))
|
|
} else {
|
|
fieldDict, err = indexReader.FieldDict(field)
|
|
}
|
|
defer func() {
|
|
if cerr := fieldDict.Close(); cerr != nil && err == nil {
|
|
err = cerr
|
|
}
|
|
}()
|
|
|
|
// enumerate the terms and check against regexp
|
|
tfd, err := fieldDict.Next()
|
|
for err == nil && tfd != nil {
|
|
matchPos := pattern.FindStringIndex(tfd.Term)
|
|
if matchPos != nil && matchPos[0] == 0 && matchPos[1] == len(tfd.Term) {
|
|
rv = append(rv, tfd.Term)
|
|
if tooManyClauses(len(rv)) {
|
|
return rv, tooManyClausesErr()
|
|
}
|
|
}
|
|
tfd, err = fieldDict.Next()
|
|
}
|
|
|
|
return rv, err
|
|
}
|