Improve issue search (#2387)
* Improve issue indexer * Fix new issue sqlite bug * Different test indexer paths for each db * Add integration indexer paths to make cleantokarchuk/v1.17
parent
52e11b24bf
commit
b0f7457d9e
@ -0,0 +1,143 @@ |
|||||||
|
// Copyright 2017 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package indexer |
||||||
|
|
||||||
|
import ( |
||||||
|
"os" |
||||||
|
|
||||||
|
"code.gitea.io/gitea/modules/log" |
||||||
|
"code.gitea.io/gitea/modules/setting" |
||||||
|
|
||||||
|
"github.com/blevesearch/bleve" |
||||||
|
"github.com/blevesearch/bleve/analysis/analyzer/custom" |
||||||
|
"github.com/blevesearch/bleve/analysis/token/lowercase" |
||||||
|
"github.com/blevesearch/bleve/analysis/token/unicodenorm" |
||||||
|
"github.com/blevesearch/bleve/analysis/tokenizer/unicode" |
||||||
|
) |
||||||
|
|
||||||
|
// issueIndexer (thread-safe) index for searching issues
|
||||||
|
var issueIndexer bleve.Index |
||||||
|
|
||||||
|
// IssueIndexerData data stored in the issue indexer
|
||||||
|
type IssueIndexerData struct { |
||||||
|
RepoID int64 |
||||||
|
Title string |
||||||
|
Content string |
||||||
|
Comments []string |
||||||
|
} |
||||||
|
|
||||||
|
// IssueIndexerUpdate an update to the issue indexer
|
||||||
|
type IssueIndexerUpdate struct { |
||||||
|
IssueID int64 |
||||||
|
Data *IssueIndexerData |
||||||
|
} |
||||||
|
|
||||||
|
const issueIndexerAnalyzer = "issueIndexer" |
||||||
|
|
||||||
|
// InitIssueIndexer initialize issue indexer
|
||||||
|
func InitIssueIndexer(populateIndexer func() error) { |
||||||
|
_, err := os.Stat(setting.Indexer.IssuePath) |
||||||
|
if err != nil { |
||||||
|
if os.IsNotExist(err) { |
||||||
|
if err = createIssueIndexer(); err != nil { |
||||||
|
log.Fatal(4, "CreateIssuesIndexer: %v", err) |
||||||
|
} |
||||||
|
if err = populateIndexer(); err != nil { |
||||||
|
log.Fatal(4, "PopulateIssuesIndex: %v", err) |
||||||
|
} |
||||||
|
} else { |
||||||
|
log.Fatal(4, "InitIssuesIndexer: %v", err) |
||||||
|
} |
||||||
|
} else { |
||||||
|
issueIndexer, err = bleve.Open(setting.Indexer.IssuePath) |
||||||
|
if err != nil { |
||||||
|
log.Error(4, "Unable to open issues indexer (%s)."+ |
||||||
|
" If the error is due to incompatible versions, try deleting the indexer files;"+ |
||||||
|
" gitea will recreate them with the appropriate version the next time it runs."+ |
||||||
|
" Deleting the indexer files will not result in loss of data.", |
||||||
|
setting.Indexer.IssuePath) |
||||||
|
log.Fatal(4, "InitIssuesIndexer, open index: %v", err) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// createIssueIndexer create an issue indexer if one does not already exist
|
||||||
|
func createIssueIndexer() error { |
||||||
|
mapping := bleve.NewIndexMapping() |
||||||
|
docMapping := bleve.NewDocumentMapping() |
||||||
|
|
||||||
|
docMapping.AddFieldMappingsAt("RepoID", bleve.NewNumericFieldMapping()) |
||||||
|
|
||||||
|
textFieldMapping := bleve.NewTextFieldMapping() |
||||||
|
docMapping.AddFieldMappingsAt("Title", textFieldMapping) |
||||||
|
docMapping.AddFieldMappingsAt("Content", textFieldMapping) |
||||||
|
docMapping.AddFieldMappingsAt("Comments", textFieldMapping) |
||||||
|
|
||||||
|
const unicodeNormNFC = "unicodeNormNFC" |
||||||
|
if err := mapping.AddCustomTokenFilter(unicodeNormNFC, map[string]interface{}{ |
||||||
|
"type": unicodenorm.Name, |
||||||
|
"form": unicodenorm.NFC, |
||||||
|
}); err != nil { |
||||||
|
return err |
||||||
|
} else if err = mapping.AddCustomAnalyzer(issueIndexerAnalyzer, map[string]interface{}{ |
||||||
|
"type": custom.Name, |
||||||
|
"char_filters": []string{}, |
||||||
|
"tokenizer": unicode.Name, |
||||||
|
"token_filters": []string{unicodeNormNFC, lowercase.Name}, |
||||||
|
}); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
mapping.DefaultAnalyzer = issueIndexerAnalyzer |
||||||
|
mapping.AddDocumentMapping("issues", docMapping) |
||||||
|
|
||||||
|
var err error |
||||||
|
issueIndexer, err = bleve.New(setting.Indexer.IssuePath, mapping) |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
// UpdateIssue update the issue indexer
|
||||||
|
func UpdateIssue(update IssueIndexerUpdate) error { |
||||||
|
return issueIndexer.Index(indexerID(update.IssueID), update.Data) |
||||||
|
} |
||||||
|
|
||||||
|
// BatchUpdateIssues perform a batch update of the issue indexer
|
||||||
|
func BatchUpdateIssues(updates ...IssueIndexerUpdate) error { |
||||||
|
batch := issueIndexer.NewBatch() |
||||||
|
for _, update := range updates { |
||||||
|
err := batch.Index(indexerID(update.IssueID), update.Data) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
return issueIndexer.Batch(batch) |
||||||
|
} |
||||||
|
|
||||||
|
// SearchIssuesByKeyword searches for issues by given conditions.
|
||||||
|
// Returns the matching issue IDs
|
||||||
|
func SearchIssuesByKeyword(repoID int64, keyword string) ([]int64, error) { |
||||||
|
indexerQuery := bleve.NewConjunctionQuery( |
||||||
|
numericEqualityQuery(repoID, "RepoID"), |
||||||
|
bleve.NewDisjunctionQuery( |
||||||
|
newMatchPhraseQuery(keyword, "Title", issueIndexerAnalyzer), |
||||||
|
newMatchPhraseQuery(keyword, "Content", issueIndexerAnalyzer), |
||||||
|
newMatchPhraseQuery(keyword, "Comments", issueIndexerAnalyzer), |
||||||
|
)) |
||||||
|
search := bleve.NewSearchRequestOptions(indexerQuery, 2147483647, 0, false) |
||||||
|
|
||||||
|
result, err := issueIndexer.Search(search) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
issueIDs := make([]int64, len(result.Hits)) |
||||||
|
for i, hit := range result.Hits { |
||||||
|
issueIDs[i], err = idOfIndexerID(hit.ID) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
} |
||||||
|
return issueIDs, nil |
||||||
|
} |
@ -0,0 +1,145 @@ |
|||||||
|
// Copyright (c) 2014 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 custom |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
|
||||||
|
"github.com/blevesearch/bleve/analysis" |
||||||
|
"github.com/blevesearch/bleve/registry" |
||||||
|
) |
||||||
|
|
||||||
|
const Name = "custom" |
||||||
|
|
||||||
|
func AnalyzerConstructor(config map[string]interface{}, cache *registry.Cache) (*analysis.Analyzer, error) { |
||||||
|
|
||||||
|
var err error |
||||||
|
var charFilters []analysis.CharFilter |
||||||
|
charFiltersValue, ok := config["char_filters"] |
||||||
|
if ok { |
||||||
|
switch charFiltersValue := charFiltersValue.(type) { |
||||||
|
case []string: |
||||||
|
charFilters, err = getCharFilters(charFiltersValue, cache) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
case []interface{}: |
||||||
|
charFiltersNames, err := convertInterfaceSliceToStringSlice(charFiltersValue, "char filter") |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
charFilters, err = getCharFilters(charFiltersNames, cache) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
default: |
||||||
|
return nil, fmt.Errorf("unsupported type for char_filters, must be slice") |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
var tokenizerName string |
||||||
|
tokenizerValue, ok := config["tokenizer"] |
||||||
|
if ok { |
||||||
|
tokenizerName, ok = tokenizerValue.(string) |
||||||
|
if !ok { |
||||||
|
return nil, fmt.Errorf("must specify tokenizer as string") |
||||||
|
} |
||||||
|
} else { |
||||||
|
return nil, fmt.Errorf("must specify tokenizer") |
||||||
|
} |
||||||
|
|
||||||
|
tokenizer, err := cache.TokenizerNamed(tokenizerName) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
var tokenFilters []analysis.TokenFilter |
||||||
|
tokenFiltersValue, ok := config["token_filters"] |
||||||
|
if ok { |
||||||
|
switch tokenFiltersValue := tokenFiltersValue.(type) { |
||||||
|
case []string: |
||||||
|
tokenFilters, err = getTokenFilters(tokenFiltersValue, cache) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
case []interface{}: |
||||||
|
tokenFiltersNames, err := convertInterfaceSliceToStringSlice(tokenFiltersValue, "token filter") |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
tokenFilters, err = getTokenFilters(tokenFiltersNames, cache) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
default: |
||||||
|
return nil, fmt.Errorf("unsupported type for token_filters, must be slice") |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
rv := analysis.Analyzer{ |
||||||
|
Tokenizer: tokenizer, |
||||||
|
} |
||||||
|
if charFilters != nil { |
||||||
|
rv.CharFilters = charFilters |
||||||
|
} |
||||||
|
if tokenFilters != nil { |
||||||
|
rv.TokenFilters = tokenFilters |
||||||
|
} |
||||||
|
return &rv, nil |
||||||
|
} |
||||||
|
|
||||||
|
func init() { |
||||||
|
registry.RegisterAnalyzer(Name, AnalyzerConstructor) |
||||||
|
} |
||||||
|
|
||||||
|
func getCharFilters(charFilterNames []string, cache *registry.Cache) ([]analysis.CharFilter, error) { |
||||||
|
charFilters := make([]analysis.CharFilter, len(charFilterNames)) |
||||||
|
for i, charFilterName := range charFilterNames { |
||||||
|
charFilter, err := cache.CharFilterNamed(charFilterName) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
charFilters[i] = charFilter |
||||||
|
} |
||||||
|
|
||||||
|
return charFilters, nil |
||||||
|
} |
||||||
|
|
||||||
|
func getTokenFilters(tokenFilterNames []string, cache *registry.Cache) ([]analysis.TokenFilter, error) { |
||||||
|
tokenFilters := make([]analysis.TokenFilter, len(tokenFilterNames)) |
||||||
|
for i, tokenFilterName := range tokenFilterNames { |
||||||
|
tokenFilter, err := cache.TokenFilterNamed(tokenFilterName) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
tokenFilters[i] = tokenFilter |
||||||
|
} |
||||||
|
|
||||||
|
return tokenFilters, nil |
||||||
|
} |
||||||
|
|
||||||
|
func convertInterfaceSliceToStringSlice(interfaceSlice []interface{}, objType string) ([]string, error) { |
||||||
|
stringSlice := make([]string, len(interfaceSlice)) |
||||||
|
for i, interfaceObj := range interfaceSlice { |
||||||
|
stringObj, ok := interfaceObj.(string) |
||||||
|
if ok { |
||||||
|
stringSlice[i] = stringObj |
||||||
|
} else { |
||||||
|
return nil, fmt.Errorf(objType + " name must be a string") |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return stringSlice, nil |
||||||
|
} |
@ -1,46 +0,0 @@ |
|||||||
// Copyright (c) 2014 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 simple |
|
||||||
|
|
||||||
import ( |
|
||||||
"github.com/blevesearch/bleve/analysis" |
|
||||||
"github.com/blevesearch/bleve/analysis/token/lowercase" |
|
||||||
"github.com/blevesearch/bleve/analysis/tokenizer/letter" |
|
||||||
"github.com/blevesearch/bleve/registry" |
|
||||||
) |
|
||||||
|
|
||||||
const Name = "simple" |
|
||||||
|
|
||||||
func AnalyzerConstructor(config map[string]interface{}, cache *registry.Cache) (*analysis.Analyzer, error) { |
|
||||||
tokenizer, err := cache.TokenizerNamed(letter.Name) |
|
||||||
if err != nil { |
|
||||||
return nil, err |
|
||||||
} |
|
||||||
toLowerFilter, err := cache.TokenFilterNamed(lowercase.Name) |
|
||||||
if err != nil { |
|
||||||
return nil, err |
|
||||||
} |
|
||||||
rv := analysis.Analyzer{ |
|
||||||
Tokenizer: tokenizer, |
|
||||||
TokenFilters: []analysis.TokenFilter{ |
|
||||||
toLowerFilter, |
|
||||||
}, |
|
||||||
} |
|
||||||
return &rv, nil |
|
||||||
} |
|
||||||
|
|
||||||
func init() { |
|
||||||
registry.RegisterAnalyzer(Name, AnalyzerConstructor) |
|
||||||
} |
|
@ -0,0 +1,79 @@ |
|||||||
|
// Copyright (c) 2014 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 unicodenorm |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
|
||||||
|
"github.com/blevesearch/bleve/analysis" |
||||||
|
"github.com/blevesearch/bleve/registry" |
||||||
|
"golang.org/x/text/unicode/norm" |
||||||
|
) |
||||||
|
|
||||||
|
const Name = "normalize_unicode" |
||||||
|
|
||||||
|
const NFC = "nfc" |
||||||
|
const NFD = "nfd" |
||||||
|
const NFKC = "nfkc" |
||||||
|
const NFKD = "nfkd" |
||||||
|
|
||||||
|
var forms = map[string]norm.Form{ |
||||||
|
NFC: norm.NFC, |
||||||
|
NFD: norm.NFD, |
||||||
|
NFKC: norm.NFKC, |
||||||
|
NFKD: norm.NFKD, |
||||||
|
} |
||||||
|
|
||||||
|
type UnicodeNormalizeFilter struct { |
||||||
|
form norm.Form |
||||||
|
} |
||||||
|
|
||||||
|
func NewUnicodeNormalizeFilter(formName string) (*UnicodeNormalizeFilter, error) { |
||||||
|
form, ok := forms[formName] |
||||||
|
if !ok { |
||||||
|
return nil, fmt.Errorf("no form named %s", formName) |
||||||
|
} |
||||||
|
return &UnicodeNormalizeFilter{ |
||||||
|
form: form, |
||||||
|
}, nil |
||||||
|
} |
||||||
|
|
||||||
|
func MustNewUnicodeNormalizeFilter(formName string) *UnicodeNormalizeFilter { |
||||||
|
filter, err := NewUnicodeNormalizeFilter(formName) |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
return filter |
||||||
|
} |
||||||
|
|
||||||
|
func (s *UnicodeNormalizeFilter) Filter(input analysis.TokenStream) analysis.TokenStream { |
||||||
|
for _, token := range input { |
||||||
|
token.Term = s.form.Bytes(token.Term) |
||||||
|
} |
||||||
|
return input |
||||||
|
} |
||||||
|
|
||||||
|
func UnicodeNormalizeFilterConstructor(config map[string]interface{}, cache *registry.Cache) (analysis.TokenFilter, error) { |
||||||
|
formVal, ok := config["form"].(string) |
||||||
|
if !ok { |
||||||
|
return nil, fmt.Errorf("must specify form") |
||||||
|
} |
||||||
|
form := formVal |
||||||
|
return NewUnicodeNormalizeFilter(form) |
||||||
|
} |
||||||
|
|
||||||
|
func init() { |
||||||
|
registry.RegisterTokenFilter(Name, UnicodeNormalizeFilterConstructor) |
||||||
|
} |
@ -1,76 +0,0 @@ |
|||||||
// Copyright (c) 2016 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 character |
|
||||||
|
|
||||||
import ( |
|
||||||
"unicode/utf8" |
|
||||||
|
|
||||||
"github.com/blevesearch/bleve/analysis" |
|
||||||
) |
|
||||||
|
|
||||||
type IsTokenRune func(r rune) bool |
|
||||||
|
|
||||||
type CharacterTokenizer struct { |
|
||||||
isTokenRun IsTokenRune |
|
||||||
} |
|
||||||
|
|
||||||
func NewCharacterTokenizer(f IsTokenRune) *CharacterTokenizer { |
|
||||||
return &CharacterTokenizer{ |
|
||||||
isTokenRun: f, |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
func (c *CharacterTokenizer) Tokenize(input []byte) analysis.TokenStream { |
|
||||||
|
|
||||||
rv := make(analysis.TokenStream, 0, 1024) |
|
||||||
|
|
||||||
offset := 0 |
|
||||||
start := 0 |
|
||||||
end := 0 |
|
||||||
count := 0 |
|
||||||
for currRune, size := utf8.DecodeRune(input[offset:]); currRune != utf8.RuneError; currRune, size = utf8.DecodeRune(input[offset:]) { |
|
||||||
isToken := c.isTokenRun(currRune) |
|
||||||
if isToken { |
|
||||||
end = offset + size |
|
||||||
} else { |
|
||||||
if end-start > 0 { |
|
||||||
// build token
|
|
||||||
rv = append(rv, &analysis.Token{ |
|
||||||
Term: input[start:end], |
|
||||||
Start: start, |
|
||||||
End: end, |
|
||||||
Position: count + 1, |
|
||||||
Type: analysis.AlphaNumeric, |
|
||||||
}) |
|
||||||
count++ |
|
||||||
} |
|
||||||
start = offset + size |
|
||||||
end = start |
|
||||||
} |
|
||||||
offset += size |
|
||||||
} |
|
||||||
// if we ended in the middle of a token, finish it
|
|
||||||
if end-start > 0 { |
|
||||||
// build token
|
|
||||||
rv = append(rv, &analysis.Token{ |
|
||||||
Term: input[start:end], |
|
||||||
Start: start, |
|
||||||
End: end, |
|
||||||
Position: count + 1, |
|
||||||
Type: analysis.AlphaNumeric, |
|
||||||
}) |
|
||||||
} |
|
||||||
return rv |
|
||||||
} |
|
@ -1,33 +0,0 @@ |
|||||||
// Copyright (c) 2016 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 letter |
|
||||||
|
|
||||||
import ( |
|
||||||
"unicode" |
|
||||||
|
|
||||||
"github.com/blevesearch/bleve/analysis" |
|
||||||
"github.com/blevesearch/bleve/analysis/tokenizer/character" |
|
||||||
"github.com/blevesearch/bleve/registry" |
|
||||||
) |
|
||||||
|
|
||||||
const Name = "letter" |
|
||||||
|
|
||||||
func TokenizerConstructor(config map[string]interface{}, cache *registry.Cache) (analysis.Tokenizer, error) { |
|
||||||
return character.NewCharacterTokenizer(unicode.IsLetter), nil |
|
||||||
} |
|
||||||
|
|
||||||
func init() { |
|
||||||
registry.RegisterTokenizer(Name, TokenizerConstructor) |
|
||||||
} |
|
@ -1,23 +0,0 @@ |
|||||||
// Copyright (c) 2014 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.
|
|
||||||
|
|
||||||
// +build appengine appenginevm
|
|
||||||
|
|
||||||
package bleve |
|
||||||
|
|
||||||
// in the appengine environment we cannot support disk based indexes
|
|
||||||
// so we do no extra configuration in this method
|
|
||||||
func initDisk() { |
|
||||||
|
|
||||||
} |
|
@ -0,0 +1,137 @@ |
|||||||
|
// Copyright (c) 2017 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 document |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
|
||||||
|
"github.com/blevesearch/bleve/analysis" |
||||||
|
"github.com/blevesearch/bleve/geo" |
||||||
|
"github.com/blevesearch/bleve/numeric" |
||||||
|
) |
||||||
|
|
||||||
|
var GeoPrecisionStep uint = 9 |
||||||
|
|
||||||
|
type GeoPointField struct { |
||||||
|
name string |
||||||
|
arrayPositions []uint64 |
||||||
|
options IndexingOptions |
||||||
|
value numeric.PrefixCoded |
||||||
|
numPlainTextBytes uint64 |
||||||
|
} |
||||||
|
|
||||||
|
func (n *GeoPointField) Name() string { |
||||||
|
return n.name |
||||||
|
} |
||||||
|
|
||||||
|
func (n *GeoPointField) ArrayPositions() []uint64 { |
||||||
|
return n.arrayPositions |
||||||
|
} |
||||||
|
|
||||||
|
func (n *GeoPointField) Options() IndexingOptions { |
||||||
|
return n.options |
||||||
|
} |
||||||
|
|
||||||
|
func (n *GeoPointField) Analyze() (int, analysis.TokenFrequencies) { |
||||||
|
tokens := make(analysis.TokenStream, 0) |
||||||
|
tokens = append(tokens, &analysis.Token{ |
||||||
|
Start: 0, |
||||||
|
End: len(n.value), |
||||||
|
Term: n.value, |
||||||
|
Position: 1, |
||||||
|
Type: analysis.Numeric, |
||||||
|
}) |
||||||
|
|
||||||
|
original, err := n.value.Int64() |
||||||
|
if err == nil { |
||||||
|
|
||||||
|
shift := GeoPrecisionStep |
||||||
|
for shift < 64 { |
||||||
|
shiftEncoded, err := numeric.NewPrefixCodedInt64(original, shift) |
||||||
|
if err != nil { |
||||||
|
break |
||||||
|
} |
||||||
|
token := analysis.Token{ |
||||||
|
Start: 0, |
||||||
|
End: len(shiftEncoded), |
||||||
|
Term: shiftEncoded, |
||||||
|
Position: 1, |
||||||
|
Type: analysis.Numeric, |
||||||
|
} |
||||||
|
tokens = append(tokens, &token) |
||||||
|
shift += GeoPrecisionStep |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
fieldLength := len(tokens) |
||||||
|
tokenFreqs := analysis.TokenFrequency(tokens, n.arrayPositions, n.options.IncludeTermVectors()) |
||||||
|
return fieldLength, tokenFreqs |
||||||
|
} |
||||||
|
|
||||||
|
func (n *GeoPointField) Value() []byte { |
||||||
|
return n.value |
||||||
|
} |
||||||
|
|
||||||
|
func (n *GeoPointField) Lon() (float64, error) { |
||||||
|
i64, err := n.value.Int64() |
||||||
|
if err != nil { |
||||||
|
return 0.0, err |
||||||
|
} |
||||||
|
return geo.MortonUnhashLon(uint64(i64)), nil |
||||||
|
} |
||||||
|
|
||||||
|
func (n *GeoPointField) Lat() (float64, error) { |
||||||
|
i64, err := n.value.Int64() |
||||||
|
if err != nil { |
||||||
|
return 0.0, err |
||||||
|
} |
||||||
|
return geo.MortonUnhashLat(uint64(i64)), nil |
||||||
|
} |
||||||
|
|
||||||
|
func (n *GeoPointField) GoString() string { |
||||||
|
return fmt.Sprintf("&document.GeoPointField{Name:%s, Options: %s, Value: %s}", n.name, n.options, n.value) |
||||||
|
} |
||||||
|
|
||||||
|
func (n *GeoPointField) NumPlainTextBytes() uint64 { |
||||||
|
return n.numPlainTextBytes |
||||||
|
} |
||||||
|
|
||||||
|
func NewGeoPointFieldFromBytes(name string, arrayPositions []uint64, value []byte) *GeoPointField { |
||||||
|
return &GeoPointField{ |
||||||
|
name: name, |
||||||
|
arrayPositions: arrayPositions, |
||||||
|
value: value, |
||||||
|
options: DefaultNumericIndexingOptions, |
||||||
|
numPlainTextBytes: uint64(len(value)), |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func NewGeoPointField(name string, arrayPositions []uint64, lon, lat float64) *GeoPointField { |
||||||
|
return NewGeoPointFieldWithIndexingOptions(name, arrayPositions, lon, lat, DefaultNumericIndexingOptions) |
||||||
|
} |
||||||
|
|
||||||
|
func NewGeoPointFieldWithIndexingOptions(name string, arrayPositions []uint64, lon, lat float64, options IndexingOptions) *GeoPointField { |
||||||
|
mhash := geo.MortonHash(lon, lat) |
||||||
|
prefixCoded := numeric.MustNewPrefixCodedInt64(int64(mhash), 0) |
||||||
|
return &GeoPointField{ |
||||||
|
name: name, |
||||||
|
arrayPositions: arrayPositions, |
||||||
|
value: prefixCoded, |
||||||
|
options: options, |
||||||
|
// not correct, just a place holder until we revisit how fields are
|
||||||
|
// represented and can fix this better
|
||||||
|
numPlainTextBytes: uint64(8), |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,9 @@ |
|||||||
|
# geo support in bleve |
||||||
|
|
||||||
|
First, all of this geo code is a Go adaptation of the [Lucene 5.3.2 sandbox geo support](https://lucene.apache.org/core/5_3_2/sandbox/org/apache/lucene/util/package-summary.html). |
||||||
|
|
||||||
|
## Notes |
||||||
|
|
||||||
|
- All of the APIs will use float64 for lon/lat values. |
||||||
|
- When describing a point in function arguments or return values, we always use the order lon, lat. |
||||||
|
- High level APIs will use TopLeft and BottomRight to describe bounding boxes. This may not map cleanly to min/max lon/lat when crossing the dateline. The lower level APIs will use min/max lon/lat and require the higher-level code to split boxes accordingly. |
@ -0,0 +1,170 @@ |
|||||||
|
// Copyright (c) 2017 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 geo |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"math" |
||||||
|
|
||||||
|
"github.com/blevesearch/bleve/numeric" |
||||||
|
) |
||||||
|
|
||||||
|
// GeoBits is the number of bits used for a single geo point
|
||||||
|
// Currently this is 32bits for lon and 32bits for lat
|
||||||
|
var GeoBits uint = 32 |
||||||
|
|
||||||
|
var minLon = -180.0 |
||||||
|
var minLat = -90.0 |
||||||
|
var maxLon = 180.0 |
||||||
|
var maxLat = 90.0 |
||||||
|
var minLonRad = minLon * degreesToRadian |
||||||
|
var minLatRad = minLat * degreesToRadian |
||||||
|
var maxLonRad = maxLon * degreesToRadian |
||||||
|
var maxLatRad = maxLat * degreesToRadian |
||||||
|
var geoTolerance = 1E-6 |
||||||
|
var lonScale = float64((uint64(0x1)<<GeoBits)-1) / 360.0 |
||||||
|
var latScale = float64((uint64(0x1)<<GeoBits)-1) / 180.0 |
||||||
|
|
||||||
|
// MortonHash computes the morton hash value for the provided geo point
|
||||||
|
// This point is ordered as lon, lat.
|
||||||
|
func MortonHash(lon, lat float64) uint64 { |
||||||
|
return numeric.Interleave(scaleLon(lon), scaleLat(lat)) |
||||||
|
} |
||||||
|
|
||||||
|
func scaleLon(lon float64) uint64 { |
||||||
|
rv := uint64((lon - minLon) * lonScale) |
||||||
|
return rv |
||||||
|
} |
||||||
|
|
||||||
|
func scaleLat(lat float64) uint64 { |
||||||
|
rv := uint64((lat - minLat) * latScale) |
||||||
|
return rv |
||||||
|
} |
||||||
|
|
||||||
|
// MortonUnhashLon extracts the longitude value from the provided morton hash.
|
||||||
|
func MortonUnhashLon(hash uint64) float64 { |
||||||
|
return unscaleLon(numeric.Deinterleave(hash)) |
||||||
|
} |
||||||
|
|
||||||
|
// MortonUnhashLat extracts the latitude value from the provided morton hash.
|
||||||
|
func MortonUnhashLat(hash uint64) float64 { |
||||||
|
return unscaleLat(numeric.Deinterleave(hash >> 1)) |
||||||
|
} |
||||||
|
|
||||||
|
func unscaleLon(lon uint64) float64 { |
||||||
|
return (float64(lon) / lonScale) + minLon |
||||||
|
} |
||||||
|
|
||||||
|
func unscaleLat(lat uint64) float64 { |
||||||
|
return (float64(lat) / latScale) + minLat |
||||||
|
} |
||||||
|
|
||||||
|
// compareGeo will compare two float values and see if they are the same
|
||||||
|
// taking into consideration a known geo tolerance.
|
||||||
|
func compareGeo(a, b float64) float64 { |
||||||
|
compare := a - b |
||||||
|
if math.Abs(compare) <= geoTolerance { |
||||||
|
return 0 |
||||||
|
} |
||||||
|
return compare |
||||||
|
} |
||||||
|
|
||||||
|
// RectIntersects checks whether rectangles a and b intersect
|
||||||
|
func RectIntersects(aMinX, aMinY, aMaxX, aMaxY, bMinX, bMinY, bMaxX, bMaxY float64) bool { |
||||||
|
return !(aMaxX < bMinX || aMinX > bMaxX || aMaxY < bMinY || aMinY > bMaxY) |
||||||
|
} |
||||||
|
|
||||||
|
// RectWithin checks whether box a is within box b
|
||||||
|
func RectWithin(aMinX, aMinY, aMaxX, aMaxY, bMinX, bMinY, bMaxX, bMaxY float64) bool { |
||||||
|
rv := !(aMinX < bMinX || aMinY < bMinY || aMaxX > bMaxX || aMaxY > bMaxY) |
||||||
|
return rv |
||||||
|
} |
||||||
|
|
||||||
|
// BoundingBoxContains checks whether the lon/lat point is within the box
|
||||||
|
func BoundingBoxContains(lon, lat, minLon, minLat, maxLon, maxLat float64) bool { |
||||||
|
return compareGeo(lon, minLon) >= 0 && compareGeo(lon, maxLon) <= 0 && |
||||||
|
compareGeo(lat, minLat) >= 0 && compareGeo(lat, maxLat) <= 0 |
||||||
|
} |
||||||
|
|
||||||
|
const degreesToRadian = math.Pi / 180 |
||||||
|
const radiansToDegrees = 180 / math.Pi |
||||||
|
|
||||||
|
// DegreesToRadians converts an angle in degrees to radians
|
||||||
|
func DegreesToRadians(d float64) float64 { |
||||||
|
return d * degreesToRadian |
||||||
|
} |
||||||
|
|
||||||
|
// RadiansToDegrees converts an angle in radians to degress
|
||||||
|
func RadiansToDegrees(r float64) float64 { |
||||||
|
return r * radiansToDegrees |
||||||
|
} |
||||||
|
|
||||||
|
var earthMeanRadiusMeters = 6371008.7714 |
||||||
|
|
||||||
|
func RectFromPointDistance(lon, lat, dist float64) (float64, float64, float64, float64, error) { |
||||||
|
err := checkLongitude(lon) |
||||||
|
if err != nil { |
||||||
|
return 0, 0, 0, 0, err |
||||||
|
} |
||||||
|
err = checkLatitude(lat) |
||||||
|
if err != nil { |
||||||
|
return 0, 0, 0, 0, err |
||||||
|
} |
||||||
|
radLon := DegreesToRadians(lon) |
||||||
|
radLat := DegreesToRadians(lat) |
||||||
|
radDistance := (dist + 7e-2) / earthMeanRadiusMeters |
||||||
|
|
||||||
|
minLatL := radLat - radDistance |
||||||
|
maxLatL := radLat + radDistance |
||||||
|
|
||||||
|
var minLonL, maxLonL float64 |
||||||
|
if minLatL > minLatRad && maxLatL < maxLatRad { |
||||||
|
deltaLon := asin(sin(radDistance) / cos(radLat)) |
||||||
|
minLonL = radLon - deltaLon |
||||||
|
if minLonL < minLonRad { |
||||||
|
minLonL += 2 * math.Pi |
||||||
|
} |
||||||
|
maxLonL = radLon + deltaLon |
||||||
|
if maxLonL > maxLonRad { |
||||||
|
maxLonL -= 2 * math.Pi |
||||||
|
} |
||||||
|
} else { |
||||||
|
// pole is inside distance
|
||||||
|
minLatL = math.Max(minLatL, minLatRad) |
||||||
|
maxLatL = math.Min(maxLatL, maxLatRad) |
||||||
|
minLonL = minLonRad |
||||||
|
maxLonL = maxLonRad |
||||||
|
} |
||||||
|
|
||||||
|
return RadiansToDegrees(minLonL), |
||||||
|
RadiansToDegrees(maxLatL), |
||||||
|
RadiansToDegrees(maxLonL), |
||||||
|
RadiansToDegrees(minLatL), |
||||||
|
nil |
||||||
|
} |
||||||
|
|
||||||
|
func checkLatitude(latitude float64) error { |
||||||
|
if math.IsNaN(latitude) || latitude < minLat || latitude > maxLat { |
||||||
|
return fmt.Errorf("invalid latitude %f; must be between %f and %f", latitude, minLat, maxLat) |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func checkLongitude(longitude float64) error { |
||||||
|
if math.IsNaN(longitude) || longitude < minLon || longitude > maxLon { |
||||||
|
return fmt.Errorf("invalid longitude %f; must be between %f and %f", longitude, minLon, maxLon) |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
@ -0,0 +1,98 @@ |
|||||||
|
// Copyright (c) 2017 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 geo |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"math" |
||||||
|
"strconv" |
||||||
|
"strings" |
||||||
|
) |
||||||
|
|
||||||
|
type distanceUnit struct { |
||||||
|
conv float64 |
||||||
|
suffixes []string |
||||||
|
} |
||||||
|
|
||||||
|
var inch = distanceUnit{0.0254, []string{"in", "inch"}} |
||||||
|
var yard = distanceUnit{0.9144, []string{"yd", "yards"}} |
||||||
|
var feet = distanceUnit{0.3048, []string{"ft", "feet"}} |
||||||
|
var kilom = distanceUnit{1000, []string{"km", "kilometers"}} |
||||||
|
var nauticalm = distanceUnit{1852.0, []string{"nm", "nauticalmiles"}} |
||||||
|
var millim = distanceUnit{0.001, []string{"mm", "millimeters"}} |
||||||
|
var centim = distanceUnit{0.01, []string{"cm", "centimeters"}} |
||||||
|
var miles = distanceUnit{1609.344, []string{"mi", "miles"}} |
||||||
|
var meters = distanceUnit{1, []string{"m", "meters"}} |
||||||
|
|
||||||
|
var distanceUnits = []*distanceUnit{ |
||||||
|
&inch, &yard, &feet, &kilom, &nauticalm, &millim, ¢im, &miles, &meters, |
||||||
|
} |
||||||
|
|
||||||
|
// ParseDistance attempts to parse a distance string and return distance in
|
||||||
|
// meters. Example formats supported:
|
||||||
|
// "5in" "5inch" "7yd" "7yards" "9ft" "9feet" "11km" "11kilometers"
|
||||||
|
// "3nm" "3nauticalmiles" "13mm" "13millimeters" "15cm" "15centimeters"
|
||||||
|
// "17mi" "17miles" "19m" "19meters"
|
||||||
|
// If the unit cannot be determined, the entire string is parsed and the
|
||||||
|
// unit of meters is assumed.
|
||||||
|
// If the number portion cannot be parsed, 0 and the parse error are returned.
|
||||||
|
func ParseDistance(d string) (float64, error) { |
||||||
|
for _, unit := range distanceUnits { |
||||||
|
for _, unitSuffix := range unit.suffixes { |
||||||
|
if strings.HasSuffix(d, unitSuffix) { |
||||||
|
parsedNum, err := strconv.ParseFloat(d[0:len(d)-len(unitSuffix)], 64) |
||||||
|
if err != nil { |
||||||
|
return 0, err |
||||||
|
} |
||||||
|
return parsedNum * unit.conv, nil |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
// no unit matched, try assuming meters?
|
||||||
|
parsedNum, err := strconv.ParseFloat(d, 64) |
||||||
|
if err != nil { |
||||||
|
return 0, err |
||||||
|
} |
||||||
|
return parsedNum, nil |
||||||
|
} |
||||||
|
|
||||||
|
// ParseDistanceUnit attempts to parse a distance unit and return the
|
||||||
|
// multiplier for converting this to meters. If the unit cannot be parsed
|
||||||
|
// then 0 and the error message is returned.
|
||||||
|
func ParseDistanceUnit(u string) (float64, error) { |
||||||
|
for _, unit := range distanceUnits { |
||||||
|
for _, unitSuffix := range unit.suffixes { |
||||||
|
if u == unitSuffix { |
||||||
|
return unit.conv, nil |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
return 0, fmt.Errorf("unknown distance unit: %s", u) |
||||||
|
} |
||||||
|
|
||||||
|
// Haversin computes the distance between two points.
|
||||||
|
// This implemenation uses the sloppy math implemenations which trade off
|
||||||
|
// accuracy for performance. The distance returned is in kilometers.
|
||||||
|
func Haversin(lon1, lat1, lon2, lat2 float64) float64 { |
||||||
|
x1 := lat1 * degreesToRadian |
||||||
|
x2 := lat2 * degreesToRadian |
||||||
|
h1 := 1 - cos(x1-x2) |
||||||
|
h2 := 1 - cos((lon1-lon2)*degreesToRadian) |
||||||
|
h := (h1 + cos(x1)*cos(x2)*h2) / 2 |
||||||
|
avgLat := (x1 + x2) / 2 |
||||||
|
diameter := earthDiameter(avgLat) |
||||||
|
|
||||||
|
return diameter * asin(math.Min(1, math.Sqrt(h))) |
||||||
|
} |
@ -0,0 +1,140 @@ |
|||||||
|
// Copyright (c) 2017 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 geo |
||||||
|
|
||||||
|
import ( |
||||||
|
"reflect" |
||||||
|
"strings" |
||||||
|
) |
||||||
|
|
||||||
|
// ExtractGeoPoint takes an arbitrary interface{} and tries it's best to
|
||||||
|
// interpret it is as geo point. Supported formats:
|
||||||
|
// Container:
|
||||||
|
// slice length 2 (GeoJSON)
|
||||||
|
// first element lon, second element lat
|
||||||
|
// map[string]interface{}
|
||||||
|
// exact keys lat and lon or lng
|
||||||
|
// struct
|
||||||
|
// w/exported fields case-insensitive match on lat and lon or lng
|
||||||
|
// struct
|
||||||
|
// satisfying Later and Loner or Lnger interfaces
|
||||||
|
//
|
||||||
|
// in all cases values must be some sort of numeric-like thing: int/uint/float
|
||||||
|
func ExtractGeoPoint(thing interface{}) (lon, lat float64, success bool) { |
||||||
|
var foundLon, foundLat bool |
||||||
|
|
||||||
|
thingVal := reflect.ValueOf(thing) |
||||||
|
thingTyp := thingVal.Type() |
||||||
|
|
||||||
|
// is it a slice
|
||||||
|
if thingVal.IsValid() && thingVal.Kind() == reflect.Slice { |
||||||
|
// must be length 2
|
||||||
|
if thingVal.Len() == 2 { |
||||||
|
first := thingVal.Index(0) |
||||||
|
if first.CanInterface() { |
||||||
|
firstVal := first.Interface() |
||||||
|
lon, foundLon = extractNumericVal(firstVal) |
||||||
|
} |
||||||
|
second := thingVal.Index(1) |
||||||
|
if second.CanInterface() { |
||||||
|
secondVal := second.Interface() |
||||||
|
lat, foundLat = extractNumericVal(secondVal) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// is it a map
|
||||||
|
if l, ok := thing.(map[string]interface{}); ok { |
||||||
|
if lval, ok := l["lon"]; ok { |
||||||
|
lon, foundLon = extractNumericVal(lval) |
||||||
|
} else if lval, ok := l["lng"]; ok { |
||||||
|
lon, foundLon = extractNumericVal(lval) |
||||||
|
} |
||||||
|
if lval, ok := l["lat"]; ok { |
||||||
|
lat, foundLat = extractNumericVal(lval) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// now try reflection on struct fields
|
||||||
|
if thingVal.IsValid() && thingVal.Kind() == reflect.Struct { |
||||||
|
for i := 0; i < thingVal.NumField(); i++ { |
||||||
|
fieldName := thingTyp.Field(i).Name |
||||||
|
if strings.HasPrefix(strings.ToLower(fieldName), "lon") { |
||||||
|
if thingVal.Field(i).CanInterface() { |
||||||
|
fieldVal := thingVal.Field(i).Interface() |
||||||
|
lon, foundLon = extractNumericVal(fieldVal) |
||||||
|
} |
||||||
|
} |
||||||
|
if strings.HasPrefix(strings.ToLower(fieldName), "lng") { |
||||||
|
if thingVal.Field(i).CanInterface() { |
||||||
|
fieldVal := thingVal.Field(i).Interface() |
||||||
|
lon, foundLon = extractNumericVal(fieldVal) |
||||||
|
} |
||||||
|
} |
||||||
|
if strings.HasPrefix(strings.ToLower(fieldName), "lat") { |
||||||
|
if thingVal.Field(i).CanInterface() { |
||||||
|
fieldVal := thingVal.Field(i).Interface() |
||||||
|
lat, foundLat = extractNumericVal(fieldVal) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// last hope, some interfaces
|
||||||
|
// lon
|
||||||
|
if l, ok := thing.(loner); ok { |
||||||
|
lon = l.Lon() |
||||||
|
foundLon = true |
||||||
|
} else if l, ok := thing.(lnger); ok { |
||||||
|
lon = l.Lng() |
||||||
|
foundLon = true |
||||||
|
} |
||||||
|
// lat
|
||||||
|
if l, ok := thing.(later); ok { |
||||||
|
lat = l.Lat() |
||||||
|
foundLat = true |
||||||
|
} |
||||||
|
|
||||||
|
return lon, lat, foundLon && foundLat |
||||||
|
} |
||||||
|
|
||||||
|
// extract numeric value (if possible) and returns a float64
|
||||||
|
func extractNumericVal(v interface{}) (float64, bool) { |
||||||
|
val := reflect.ValueOf(v) |
||||||
|
typ := val.Type() |
||||||
|
switch typ.Kind() { |
||||||
|
case reflect.Float32, reflect.Float64: |
||||||
|
return val.Float(), true |
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: |
||||||
|
return float64(val.Int()), true |
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64: |
||||||
|
return float64(val.Uint()), true |
||||||
|
} |
||||||
|
|
||||||
|
return 0, false |
||||||
|
} |
||||||
|
|
||||||
|
// various support interfaces which can be used to find lat/lon
|
||||||
|
type loner interface { |
||||||
|
Lon() float64 |
||||||
|
} |
||||||
|
|
||||||
|
type later interface { |
||||||
|
Lat() float64 |
||||||
|
} |
||||||
|
|
||||||
|
type lnger interface { |
||||||
|
Lng() float64 |
||||||
|
} |
@ -0,0 +1,212 @@ |
|||||||
|
// Copyright (c) 2017 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 geo |
||||||
|
|
||||||
|
import ( |
||||||
|
"math" |
||||||
|
) |
||||||
|
|
||||||
|
var earthDiameterPerLatitude []float64 |
||||||
|
var sinTab []float64 |
||||||
|
var cosTab []float64 |
||||||
|
var asinTab []float64 |
||||||
|
var asinDer1DivF1Tab []float64 |
||||||
|
var asinDer2DivF2Tab []float64 |
||||||
|
var asinDer3DivF3Tab []float64 |
||||||
|
var asinDer4DivF4Tab []float64 |
||||||
|
|
||||||
|
const radiusTabsSize = (1 << 10) + 1 |
||||||
|
const radiusDelta = (math.Pi / 2) / (radiusTabsSize - 1) |
||||||
|
const radiusIndexer = 1 / radiusDelta |
||||||
|
const sinCosTabsSize = (1 << 11) + 1 |
||||||
|
const asinTabsSize = (1 << 13) + 1 |
||||||
|
const oneDivF2 = 1 / 2.0 |
||||||
|
const oneDivF3 = 1 / 6.0 |
||||||
|
const oneDivF4 = 1 / 24.0 |
||||||
|
|
||||||
|
// 1.57079632673412561417e+00 first 33 bits of pi/2
|
||||||
|
var pio2Hi = math.Float64frombits(0x3FF921FB54400000) |
||||||
|
|
||||||
|
// 6.07710050650619224932e-11 pi/2 - PIO2_HI
|
||||||
|
var pio2Lo = math.Float64frombits(0x3DD0B4611A626331) |
||||||
|
|
||||||
|
var asinPio2Hi = math.Float64frombits(0x3FF921FB54442D18) // 1.57079632679489655800e+00
|
||||||
|
var asinPio2Lo = math.Float64frombits(0x3C91A62633145C07) // 6.12323399573676603587e-17
|
||||||
|
var asinPs0 = math.Float64frombits(0x3fc5555555555555) // 1.66666666666666657415e-01
|
||||||
|
var asinPs1 = math.Float64frombits(0xbfd4d61203eb6f7d) // -3.25565818622400915405e-01
|
||||||
|
var asinPs2 = math.Float64frombits(0x3fc9c1550e884455) // 2.01212532134862925881e-01
|
||||||
|
var asinPs3 = math.Float64frombits(0xbfa48228b5688f3b) // -4.00555345006794114027e-02
|
||||||
|
var asinPs4 = math.Float64frombits(0x3f49efe07501b288) // 7.91534994289814532176e-04
|
||||||
|
var asinPs5 = math.Float64frombits(0x3f023de10dfdf709) // 3.47933107596021167570e-05
|
||||||
|
var asinQs1 = math.Float64frombits(0xc0033a271c8a2d4b) // -2.40339491173441421878e+00
|
||||||
|
var asinQs2 = math.Float64frombits(0x40002ae59c598ac8) // 2.02094576023350569471e+00
|
||||||
|
var asinQs3 = math.Float64frombits(0xbfe6066c1b8d0159) // -6.88283971605453293030e-01
|
||||||
|
var asinQs4 = math.Float64frombits(0x3fb3b8c5b12e9282) // 7.70381505559019352791e-02
|
||||||
|
|
||||||
|
var twoPiHi = 4 * pio2Hi |
||||||
|
var twoPiLo = 4 * pio2Lo |
||||||
|
var sinCosDeltaHi = twoPiHi/sinCosTabsSize - 1 |
||||||
|
var sinCosDeltaLo = twoPiLo/sinCosTabsSize - 1 |
||||||
|
var sinCosIndexer = 1 / (sinCosDeltaHi + sinCosDeltaLo) |
||||||
|
var sinCosMaxValueForIntModulo = ((math.MaxInt64 >> 9) / sinCosIndexer) * 0.99 |
||||||
|
var asinMaxValueForTabs = math.Sin(73.0 * degreesToRadian) |
||||||
|
|
||||||
|
var asinDelta = asinMaxValueForTabs / (asinTabsSize - 1) |
||||||
|
var asinIndexer = 1 / asinDelta |
||||||
|
|
||||||
|
func init() { |
||||||
|
// initializes the tables used for the sloppy math functions
|
||||||
|
|
||||||
|
// sin and cos
|
||||||
|
sinTab = make([]float64, sinCosTabsSize) |
||||||
|
cosTab = make([]float64, sinCosTabsSize) |
||||||
|
sinCosPiIndex := (sinCosTabsSize - 1) / 2 |
||||||
|
sinCosPiMul2Index := 2 * sinCosPiIndex |
||||||
|
sinCosPiMul05Index := sinCosPiIndex / 2 |
||||||
|
sinCosPiMul15Index := 3 * sinCosPiIndex / 2 |
||||||
|
for i := 0; i < sinCosTabsSize; i++ { |
||||||
|
// angle: in [0,2*PI].
|
||||||
|
angle := float64(i)*sinCosDeltaHi + float64(i)*sinCosDeltaLo |
||||||
|
sinAngle := math.Sin(angle) |
||||||
|
cosAngle := math.Cos(angle) |
||||||
|
// For indexes corresponding to null cosine or sine, we make sure the value is zero
|
||||||
|
// and not an epsilon. This allows for a much better accuracy for results close to zero.
|
||||||
|
if i == sinCosPiIndex { |
||||||
|
sinAngle = 0.0 |
||||||
|
} else if i == sinCosPiMul2Index { |
||||||
|
sinAngle = 0.0 |
||||||
|
} else if i == sinCosPiMul05Index { |
||||||
|
sinAngle = 0.0 |
||||||
|
} else if i == sinCosPiMul15Index { |
||||||
|
sinAngle = 0.0 |
||||||
|
} |
||||||
|
sinTab[i] = sinAngle |
||||||
|
cosTab[i] = cosAngle |
||||||
|
} |
||||||
|
|
||||||
|
// asin
|
||||||
|
asinTab = make([]float64, asinTabsSize) |
||||||
|
asinDer1DivF1Tab = make([]float64, asinTabsSize) |
||||||
|
asinDer2DivF2Tab = make([]float64, asinTabsSize) |
||||||
|
asinDer3DivF3Tab = make([]float64, asinTabsSize) |
||||||
|
asinDer4DivF4Tab = make([]float64, asinTabsSize) |
||||||
|
for i := 0; i < asinTabsSize; i++ { |
||||||
|
// x: in [0,ASIN_MAX_VALUE_FOR_TABS].
|
||||||
|
x := float64(i) * asinDelta |
||||||
|
asinTab[i] = math.Asin(x) |
||||||
|
oneMinusXSqInv := 1.0 / (1 - x*x) |
||||||
|
oneMinusXSqInv05 := math.Sqrt(oneMinusXSqInv) |
||||||
|
oneMinusXSqInv15 := oneMinusXSqInv05 * oneMinusXSqInv |
||||||
|
oneMinusXSqInv25 := oneMinusXSqInv15 * oneMinusXSqInv |
||||||
|
oneMinusXSqInv35 := oneMinusXSqInv25 * oneMinusXSqInv |
||||||
|
asinDer1DivF1Tab[i] = oneMinusXSqInv05 |
||||||
|
asinDer2DivF2Tab[i] = (x * oneMinusXSqInv15) * oneDivF2 |
||||||
|
asinDer3DivF3Tab[i] = ((1 + 2*x*x) * oneMinusXSqInv25) * oneDivF3 |
||||||
|
asinDer4DivF4Tab[i] = ((5 + 2*x*(2+x*(5-2*x))) * oneMinusXSqInv35) * oneDivF4 |
||||||
|
} |
||||||
|
|
||||||
|
// earth radius
|
||||||
|
a := 6378137.0 |
||||||
|
b := 6356752.31420 |
||||||
|
a2 := a * a |
||||||
|
b2 := b * b |
||||||
|
earthDiameterPerLatitude = make([]float64, radiusTabsSize) |
||||||
|
earthDiameterPerLatitude[0] = 2.0 * a / 1000 |
||||||
|
earthDiameterPerLatitude[radiusTabsSize-1] = 2.0 * b / 1000 |
||||||
|
for i := 1; i < radiusTabsSize-1; i++ { |
||||||
|
lat := math.Pi * float64(i) / (2*radiusTabsSize - 1) |
||||||
|
one := math.Pow(a2*math.Cos(lat), 2) |
||||||
|
two := math.Pow(b2*math.Sin(lat), 2) |
||||||
|
three := math.Pow(float64(a)*math.Cos(lat), 2) |
||||||
|
four := math.Pow(b*math.Sin(lat), 2) |
||||||
|
radius := math.Sqrt((one + two) / (three + four)) |
||||||
|
earthDiameterPerLatitude[i] = 2 * radius / 1000 |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// earthDiameter returns an estimation of the earth's diameter at the specified
|
||||||
|
// latitude in kilometers
|
||||||
|
func earthDiameter(lat float64) float64 { |
||||||
|
index := math.Mod(math.Abs(lat)*radiusIndexer+0.5, float64(len(earthDiameterPerLatitude))) |
||||||
|
if math.IsNaN(index) { |
||||||
|
return 0 |
||||||
|
} |
||||||
|
return earthDiameterPerLatitude[int(index)] |
||||||
|
} |
||||||
|
|
||||||
|
var pio2 = math.Pi / 2 |
||||||
|
|
||||||
|
func sin(a float64) float64 { |
||||||
|
return cos(a - pio2) |
||||||
|
} |
||||||
|
|
||||||
|
// cos is a sloppy math (faster) implementation of math.Cos
|
||||||
|
func cos(a float64) float64 { |
||||||
|
if a < 0.0 { |
||||||
|
a = -a |
||||||
|
} |
||||||
|
if a > sinCosMaxValueForIntModulo { |
||||||
|
return math.Cos(a) |
||||||
|
} |
||||||
|
// index: possibly outside tables range.
|
||||||
|
index := int(a*sinCosIndexer + 0.5) |
||||||
|
delta := (a - float64(index)*sinCosDeltaHi) - float64(index)*sinCosDeltaLo |
||||||
|
// Making sure index is within tables range.
|
||||||
|
// Last value of each table is the same than first, so we ignore it (tabs size minus one) for modulo.
|
||||||
|
index &= (sinCosTabsSize - 2) // index % (SIN_COS_TABS_SIZE-1)
|
||||||
|
indexCos := cosTab[index] |
||||||
|
indexSin := sinTab[index] |
||||||
|
return indexCos + delta*(-indexSin+delta*(-indexCos*oneDivF2+delta*(indexSin*oneDivF3+delta*indexCos*oneDivF4))) |
||||||
|
} |
||||||
|
|
||||||
|
// asin is a sloppy math (faster) implementation of math.Asin
|
||||||
|
func asin(a float64) float64 { |
||||||
|
var negateResult bool |
||||||
|
if a < 0 { |
||||||
|
a = -a |
||||||
|
negateResult = true |
||||||
|
} |
||||||
|
if a <= asinMaxValueForTabs { |
||||||
|
index := int(a*asinIndexer + 0.5) |
||||||
|
delta := a - float64(index)*asinDelta |
||||||
|
result := asinTab[index] + delta*(asinDer1DivF1Tab[index]+delta*(asinDer2DivF2Tab[index]+delta*(asinDer3DivF3Tab[index]+delta*asinDer4DivF4Tab[index]))) |
||||||
|
if negateResult { |
||||||
|
return -result |
||||||
|
} |
||||||
|
return result |
||||||
|
} |
||||||
|
// value > ASIN_MAX_VALUE_FOR_TABS, or value is NaN
|
||||||
|
// This part is derived from fdlibm.
|
||||||
|
if a < 1 { |
||||||
|
t := (1.0 - a) * 0.5 |
||||||
|
p := t * (asinPs0 + t*(asinPs1+t*(asinPs2+t*(asinPs3+t*(asinPs4+t+asinPs5))))) |
||||||
|
q := 1.0 + t*(asinQs1+t*(asinQs2+t*(asinQs3+t*asinQs4))) |
||||||
|
s := math.Sqrt(t) |
||||||
|
z := s + s*(p/q) |
||||||
|
result := asinPio2Hi - ((z + z) - asinPio2Lo) |
||||||
|
if negateResult { |
||||||
|
return -result |
||||||
|
} |
||||||
|
return result |
||||||
|
} |
||||||
|
// value >= 1.0, or value is NaN
|
||||||
|
if a == 1.0 { |
||||||
|
if negateResult { |
||||||
|
return -math.Pi / 2 |
||||||
|
} |
||||||
|
return math.Pi / 2 |
||||||
|
} |
||||||
|
return math.NaN() |
||||||
|
} |
@ -0,0 +1,43 @@ |
|||||||
|
package numeric |
||||||
|
|
||||||
|
var interleaveMagic = []uint64{ |
||||||
|
0x5555555555555555, |
||||||
|
0x3333333333333333, |
||||||
|
0x0F0F0F0F0F0F0F0F, |
||||||
|
0x00FF00FF00FF00FF, |
||||||
|
0x0000FFFF0000FFFF, |
||||||
|
0x00000000FFFFFFFF, |
||||||
|
0xAAAAAAAAAAAAAAAA, |
||||||
|
} |
||||||
|
|
||||||
|
var interleaveShift = []uint{1, 2, 4, 8, 16} |
||||||
|
|
||||||
|
// Interleave the first 32 bits of each uint64
|
||||||
|
// apdated from org.apache.lucene.util.BitUtil
|
||||||
|
// whcih was adapted from:
|
||||||
|
// http://graphics.stanford.edu/~seander/bithacks.html#InterleaveBMN
|
||||||
|
func Interleave(v1, v2 uint64) uint64 { |
||||||
|
v1 = (v1 | (v1 << interleaveShift[4])) & interleaveMagic[4] |
||||||
|
v1 = (v1 | (v1 << interleaveShift[3])) & interleaveMagic[3] |
||||||
|
v1 = (v1 | (v1 << interleaveShift[2])) & interleaveMagic[2] |
||||||
|
v1 = (v1 | (v1 << interleaveShift[1])) & interleaveMagic[1] |
||||||
|
v1 = (v1 | (v1 << interleaveShift[0])) & interleaveMagic[0] |
||||||
|
v2 = (v2 | (v2 << interleaveShift[4])) & interleaveMagic[4] |
||||||
|
v2 = (v2 | (v2 << interleaveShift[3])) & interleaveMagic[3] |
||||||
|
v2 = (v2 | (v2 << interleaveShift[2])) & interleaveMagic[2] |
||||||
|
v2 = (v2 | (v2 << interleaveShift[1])) & interleaveMagic[1] |
||||||
|
v2 = (v2 | (v2 << interleaveShift[0])) & interleaveMagic[0] |
||||||
|
return (v2 << 1) | v1 |
||||||
|
} |
||||||
|
|
||||||
|
// Deinterleave the 32-bit value starting at position 0
|
||||||
|
// to get the other 32-bit value, shift it by 1 first
|
||||||
|
func Deinterleave(b uint64) uint64 { |
||||||
|
b &= interleaveMagic[0] |
||||||
|
b = (b ^ (b >> interleaveShift[0])) & interleaveMagic[1] |
||||||
|
b = (b ^ (b >> interleaveShift[1])) & interleaveMagic[2] |
||||||
|
b = (b ^ (b >> interleaveShift[2])) & interleaveMagic[3] |
||||||
|
b = (b ^ (b >> interleaveShift[3])) & interleaveMagic[4] |
||||||
|
b = (b ^ (b >> interleaveShift[4])) & interleaveMagic[5] |
||||||
|
return b |
||||||
|
} |
@ -0,0 +1,113 @@ |
|||||||
|
// Copyright (c) 2017 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 query |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding/json" |
||||||
|
"fmt" |
||||||
|
|
||||||
|
"github.com/blevesearch/bleve/geo" |
||||||
|
"github.com/blevesearch/bleve/index" |
||||||
|
"github.com/blevesearch/bleve/mapping" |
||||||
|
"github.com/blevesearch/bleve/search" |
||||||
|
"github.com/blevesearch/bleve/search/searcher" |
||||||
|
) |
||||||
|
|
||||||
|
type GeoBoundingBoxQuery struct { |
||||||
|
TopLeft []float64 `json:"top_left,omitempty"` |
||||||
|
BottomRight []float64 `json:"bottom_right,omitempty"` |
||||||
|
FieldVal string `json:"field,omitempty"` |
||||||
|
BoostVal *Boost `json:"boost,omitempty"` |
||||||
|
} |
||||||
|
|
||||||
|
func NewGeoBoundingBoxQuery(topLeftLon, topLeftLat, bottomRightLon, bottomRightLat float64) *GeoBoundingBoxQuery { |
||||||
|
return &GeoBoundingBoxQuery{ |
||||||
|
TopLeft: []float64{topLeftLon, topLeftLat}, |
||||||
|
BottomRight: []float64{bottomRightLon, bottomRightLat}, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoBoundingBoxQuery) SetBoost(b float64) { |
||||||
|
boost := Boost(b) |
||||||
|
q.BoostVal = &boost |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoBoundingBoxQuery) Boost() float64 { |
||||||
|
return q.BoostVal.Value() |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoBoundingBoxQuery) SetField(f string) { |
||||||
|
q.FieldVal = f |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoBoundingBoxQuery) Field() string { |
||||||
|
return q.FieldVal |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoBoundingBoxQuery) Searcher(i index.IndexReader, m mapping.IndexMapping, options search.SearcherOptions) (search.Searcher, error) { |
||||||
|
field := q.FieldVal |
||||||
|
if q.FieldVal == "" { |
||||||
|
field = m.DefaultSearchField() |
||||||
|
} |
||||||
|
|
||||||
|
if q.BottomRight[0] < q.TopLeft[0] { |
||||||
|
// cross date line, rewrite as two parts
|
||||||
|
|
||||||
|
leftSearcher, err := searcher.NewGeoBoundingBoxSearcher(i, -180, q.BottomRight[1], q.BottomRight[0], q.TopLeft[1], field, q.BoostVal.Value(), options, true) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
rightSearcher, err := searcher.NewGeoBoundingBoxSearcher(i, q.TopLeft[0], q.BottomRight[1], 180, q.TopLeft[1], field, q.BoostVal.Value(), options, true) |
||||||
|
if err != nil { |
||||||
|
_ = leftSearcher.Close() |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
return searcher.NewDisjunctionSearcher(i, []search.Searcher{leftSearcher, rightSearcher}, 0, options) |
||||||
|
} |
||||||
|
|
||||||
|
return searcher.NewGeoBoundingBoxSearcher(i, q.TopLeft[0], q.BottomRight[1], q.BottomRight[0], q.TopLeft[1], field, q.BoostVal.Value(), options, true) |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoBoundingBoxQuery) Validate() error { |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoBoundingBoxQuery) UnmarshalJSON(data []byte) error { |
||||||
|
tmp := struct { |
||||||
|
TopLeft interface{} `json:"top_left,omitempty"` |
||||||
|
BottomRight interface{} `json:"bottom_right,omitempty"` |
||||||
|
FieldVal string `json:"field,omitempty"` |
||||||
|
BoostVal *Boost `json:"boost,omitempty"` |
||||||
|
}{} |
||||||
|
err := json.Unmarshal(data, &tmp) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
// now use our generic point parsing code from the geo package
|
||||||
|
lon, lat, found := geo.ExtractGeoPoint(tmp.TopLeft) |
||||||
|
if !found { |
||||||
|
return fmt.Errorf("geo location top_left not in a valid format") |
||||||
|
} |
||||||
|
q.TopLeft = []float64{lon, lat} |
||||||
|
lon, lat, found = geo.ExtractGeoPoint(tmp.BottomRight) |
||||||
|
if !found { |
||||||
|
return fmt.Errorf("geo location bottom_right not in a valid format") |
||||||
|
} |
||||||
|
q.BottomRight = []float64{lon, lat} |
||||||
|
q.FieldVal = tmp.FieldVal |
||||||
|
q.BoostVal = tmp.BoostVal |
||||||
|
return nil |
||||||
|
} |
@ -0,0 +1,100 @@ |
|||||||
|
// Copyright (c) 2017 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 query |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding/json" |
||||||
|
"fmt" |
||||||
|
|
||||||
|
"github.com/blevesearch/bleve/geo" |
||||||
|
"github.com/blevesearch/bleve/index" |
||||||
|
"github.com/blevesearch/bleve/mapping" |
||||||
|
"github.com/blevesearch/bleve/search" |
||||||
|
"github.com/blevesearch/bleve/search/searcher" |
||||||
|
) |
||||||
|
|
||||||
|
type GeoDistanceQuery struct { |
||||||
|
Location []float64 `json:"location,omitempty"` |
||||||
|
Distance string `json:"distance,omitempty"` |
||||||
|
FieldVal string `json:"field,omitempty"` |
||||||
|
BoostVal *Boost `json:"boost,omitempty"` |
||||||
|
} |
||||||
|
|
||||||
|
func NewGeoDistanceQuery(lon, lat float64, distance string) *GeoDistanceQuery { |
||||||
|
return &GeoDistanceQuery{ |
||||||
|
Location: []float64{lon, lat}, |
||||||
|
Distance: distance, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoDistanceQuery) SetBoost(b float64) { |
||||||
|
boost := Boost(b) |
||||||
|
q.BoostVal = &boost |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoDistanceQuery) Boost() float64 { |
||||||
|
return q.BoostVal.Value() |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoDistanceQuery) SetField(f string) { |
||||||
|
q.FieldVal = f |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoDistanceQuery) Field() string { |
||||||
|
return q.FieldVal |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoDistanceQuery) Searcher(i index.IndexReader, m mapping.IndexMapping, |
||||||
|
options search.SearcherOptions) (search.Searcher, error) { |
||||||
|
field := q.FieldVal |
||||||
|
if q.FieldVal == "" { |
||||||
|
field = m.DefaultSearchField() |
||||||
|
} |
||||||
|
|
||||||
|
dist, err := geo.ParseDistance(q.Distance) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
return searcher.NewGeoPointDistanceSearcher(i, q.Location[0], q.Location[1], |
||||||
|
dist, field, q.BoostVal.Value(), options) |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoDistanceQuery) Validate() error { |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (q *GeoDistanceQuery) UnmarshalJSON(data []byte) error { |
||||||
|
tmp := struct { |
||||||
|
Location interface{} `json:"location,omitempty"` |
||||||
|
Distance string `json:"distance,omitempty"` |
||||||
|
FieldVal string `json:"field,omitempty"` |
||||||
|
BoostVal *Boost `json:"boost,omitempty"` |
||||||
|
}{} |
||||||
|
err := json.Unmarshal(data, &tmp) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
// now use our generic point parsing code from the geo package
|
||||||
|
lon, lat, found := geo.ExtractGeoPoint(tmp.Location) |
||||||
|
if !found { |
||||||
|
return fmt.Errorf("geo location not in a valid format") |
||||||
|
} |
||||||
|
q.Location = []float64{lon, lat} |
||||||
|
q.Distance = tmp.Distance |
||||||
|
q.FieldVal = tmp.FieldVal |
||||||
|
q.BoostVal = tmp.BoostVal |
||||||
|
return nil |
||||||
|
} |
@ -0,0 +1,80 @@ |
|||||||
|
// Copyright (c) 2014 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 query |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding/json" |
||||||
|
"fmt" |
||||||
|
|
||||||
|
"github.com/blevesearch/bleve/index" |
||||||
|
"github.com/blevesearch/bleve/mapping" |
||||||
|
"github.com/blevesearch/bleve/search" |
||||||
|
"github.com/blevesearch/bleve/search/searcher" |
||||||
|
) |
||||||
|
|
||||||
|
type MultiPhraseQuery struct { |
||||||
|
Terms [][]string `json:"terms"` |
||||||
|
Field string `json:"field,omitempty"` |
||||||
|
BoostVal *Boost `json:"boost,omitempty"` |
||||||
|
} |
||||||
|
|
||||||
|
// NewMultiPhraseQuery creates a new Query for finding
|
||||||
|
// term phrases in the index.
|
||||||
|
// It is like PhraseQuery, but each position in the
|
||||||
|
// phrase may be satisfied by a list of terms
|
||||||
|
// as opposed to just one.
|
||||||
|
// At least one of the terms must exist in the correct
|
||||||
|
// order, at the correct index offsets, in the
|
||||||
|
// specified field. Queried field must have been indexed with
|
||||||
|
// IncludeTermVectors set to true.
|
||||||
|
func NewMultiPhraseQuery(terms [][]string, field string) *MultiPhraseQuery { |
||||||
|
return &MultiPhraseQuery{ |
||||||
|
Terms: terms, |
||||||
|
Field: field, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (q *MultiPhraseQuery) SetBoost(b float64) { |
||||||
|
boost := Boost(b) |
||||||
|
q.BoostVal = &boost |
||||||
|
} |
||||||
|
|
||||||
|
func (q *MultiPhraseQuery) Boost() float64 { |
||||||
|
return q.BoostVal.Value() |
||||||
|
} |
||||||
|
|
||||||
|
func (q *MultiPhraseQuery) Searcher(i index.IndexReader, m mapping.IndexMapping, options search.SearcherOptions) (search.Searcher, error) { |
||||||
|
return searcher.NewMultiPhraseSearcher(i, q.Terms, q.Field, options) |
||||||
|
} |
||||||
|
|
||||||
|
func (q *MultiPhraseQuery) Validate() error { |
||||||
|
if len(q.Terms) < 1 { |
||||||
|
return fmt.Errorf("phrase query must contain at least one term") |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (q *MultiPhraseQuery) UnmarshalJSON(data []byte) error { |
||||||
|
type _mphraseQuery MultiPhraseQuery |
||||||
|
tmp := _mphraseQuery{} |
||||||
|
err := json.Unmarshal(data, &tmp) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
q.Terms = tmp.Terms |
||||||
|
q.Field = tmp.Field |
||||||
|
q.BoostVal = tmp.BoostVal |
||||||
|
return nil |
||||||
|
} |
@ -0,0 +1,95 @@ |
|||||||
|
// Copyright (c) 2017 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 query |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
|
||||||
|
"github.com/blevesearch/bleve/index" |
||||||
|
"github.com/blevesearch/bleve/mapping" |
||||||
|
"github.com/blevesearch/bleve/search" |
||||||
|
"github.com/blevesearch/bleve/search/searcher" |
||||||
|
) |
||||||
|
|
||||||
|
type TermRangeQuery struct { |
||||||
|
Min string `json:"min,omitempty"` |
||||||
|
Max string `json:"max,omitempty"` |
||||||
|
InclusiveMin *bool `json:"inclusive_min,omitempty"` |
||||||
|
InclusiveMax *bool `json:"inclusive_max,omitempty"` |
||||||
|
FieldVal string `json:"field,omitempty"` |
||||||
|
BoostVal *Boost `json:"boost,omitempty"` |
||||||
|
} |
||||||
|
|
||||||
|
// NewTermRangeQuery creates a new Query for ranges
|
||||||
|
// of text term values.
|
||||||
|
// Either, but not both endpoints can be nil.
|
||||||
|
// The minimum value is inclusive.
|
||||||
|
// The maximum value is exclusive.
|
||||||
|
func NewTermRangeQuery(min, max string) *TermRangeQuery { |
||||||
|
return NewTermRangeInclusiveQuery(min, max, nil, nil) |
||||||
|
} |
||||||
|
|
||||||
|
// NewTermRangeInclusiveQuery creates a new Query for ranges
|
||||||
|
// of numeric values.
|
||||||
|
// Either, but not both endpoints can be nil.
|
||||||
|
// Control endpoint inclusion with inclusiveMin, inclusiveMax.
|
||||||
|
func NewTermRangeInclusiveQuery(min, max string, minInclusive, maxInclusive *bool) *TermRangeQuery { |
||||||
|
return &TermRangeQuery{ |
||||||
|
Min: min, |
||||||
|
Max: max, |
||||||
|
InclusiveMin: minInclusive, |
||||||
|
InclusiveMax: maxInclusive, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (q *TermRangeQuery) SetBoost(b float64) { |
||||||
|
boost := Boost(b) |
||||||
|
q.BoostVal = &boost |
||||||
|
} |
||||||
|
|
||||||
|
func (q *TermRangeQuery) Boost() float64 { |
||||||
|
return q.BoostVal.Value() |
||||||
|
} |
||||||
|
|
||||||
|
func (q *TermRangeQuery) SetField(f string) { |
||||||
|
q.FieldVal = f |
||||||
|
} |
||||||
|
|
||||||
|
func (q *TermRangeQuery) Field() string { |
||||||
|
return q.FieldVal |
||||||
|
} |
||||||
|
|
||||||
|
func (q *TermRangeQuery) Searcher(i index.IndexReader, m mapping.IndexMapping, options search.SearcherOptions) (search.Searcher, error) { |
||||||
|
field := q.FieldVal |
||||||
|
if q.FieldVal == "" { |
||||||
|
field = m.DefaultSearchField() |
||||||
|
} |
||||||
|
var minTerm []byte |
||||||
|
if q.Min != "" { |
||||||
|
minTerm = []byte(q.Min) |
||||||
|
} |
||||||
|
var maxTerm []byte |
||||||
|
if q.Max != "" { |
||||||
|
maxTerm = []byte(q.Max) |
||||||
|
} |
||||||
|
return searcher.NewTermRangeSearcher(i, minTerm, maxTerm, q.InclusiveMin, q.InclusiveMax, field, q.BoostVal.Value(), options) |
||||||
|
} |
||||||
|
|
||||||
|
func (q *TermRangeQuery) Validate() error { |
||||||
|
if q.Min == "" && q.Min == q.Max { |
||||||
|
return fmt.Errorf("term range query must specify min or max") |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
@ -0,0 +1,88 @@ |
|||||||
|
// Copyright (c) 2017 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 ( |
||||||
|
"github.com/blevesearch/bleve/index" |
||||||
|
"github.com/blevesearch/bleve/search" |
||||||
|
) |
||||||
|
|
||||||
|
// FilterFunc defines a function which can filter documents
|
||||||
|
// returning true means keep the document
|
||||||
|
// returning false means do not keep the document
|
||||||
|
type FilterFunc func(d *search.DocumentMatch) bool |
||||||
|
|
||||||
|
// FilteringSearcher wraps any other searcher, but checks any Next/Advance
|
||||||
|
// call against the supplied FilterFunc
|
||||||
|
type FilteringSearcher struct { |
||||||
|
child search.Searcher |
||||||
|
accept FilterFunc |
||||||
|
} |
||||||
|
|
||||||
|
func NewFilteringSearcher(s search.Searcher, filter FilterFunc) *FilteringSearcher { |
||||||
|
return &FilteringSearcher{ |
||||||
|
child: s, |
||||||
|
accept: filter, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FilteringSearcher) Next(ctx *search.SearchContext) (*search.DocumentMatch, error) { |
||||||
|
next, err := f.child.Next(ctx) |
||||||
|
for next != nil && err == nil { |
||||||
|
if f.accept(next) { |
||||||
|
return next, nil |
||||||
|
} |
||||||
|
next, err = f.child.Next(ctx) |
||||||
|
} |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FilteringSearcher) Advance(ctx *search.SearchContext, ID index.IndexInternalID) (*search.DocumentMatch, error) { |
||||||
|
adv, err := f.child.Advance(ctx, ID) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
if adv == nil { |
||||||
|
return nil, nil |
||||||
|
} |
||||||
|
if f.accept(adv) { |
||||||
|
return adv, nil |
||||||
|
} |
||||||
|
return f.Next(ctx) |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FilteringSearcher) Close() error { |
||||||
|
return f.child.Close() |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FilteringSearcher) Weight() float64 { |
||||||
|
return f.child.Weight() |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FilteringSearcher) SetQueryNorm(n float64) { |
||||||
|
f.child.SetQueryNorm(n) |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FilteringSearcher) Count() uint64 { |
||||||
|
return f.child.Count() |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FilteringSearcher) Min() int { |
||||||
|
return f.child.Min() |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FilteringSearcher) DocumentMatchPoolSize() int { |
||||||
|
return f.child.DocumentMatchPoolSize() |
||||||
|
} |
@ -0,0 +1,173 @@ |
|||||||
|
// Copyright (c) 2017 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 ( |
||||||
|
"github.com/blevesearch/bleve/document" |
||||||
|
"github.com/blevesearch/bleve/geo" |
||||||
|
"github.com/blevesearch/bleve/index" |
||||||
|
"github.com/blevesearch/bleve/numeric" |
||||||
|
"github.com/blevesearch/bleve/search" |
||||||
|
) |
||||||
|
|
||||||
|
func NewGeoBoundingBoxSearcher(indexReader index.IndexReader, minLon, minLat, |
||||||
|
maxLon, maxLat float64, field string, boost float64, |
||||||
|
options search.SearcherOptions, checkBoundaries bool) ( |
||||||
|
search.Searcher, error) { |
||||||
|
|
||||||
|
// track list of opened searchers, for cleanup on early exit
|
||||||
|
var openedSearchers []search.Searcher |
||||||
|
cleanupOpenedSearchers := func() { |
||||||
|
for _, s := range openedSearchers { |
||||||
|
_ = s.Close() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// do math to produce list of terms needed for this search
|
||||||
|
onBoundaryTerms, notOnBoundaryTerms := ComputeGeoRange(0, (geo.GeoBits<<1)-1, |
||||||
|
minLon, minLat, maxLon, maxLat, checkBoundaries) |
||||||
|
|
||||||
|
var onBoundarySearcher search.Searcher |
||||||
|
if len(onBoundaryTerms) > 0 { |
||||||
|
rawOnBoundarySearcher, err := NewMultiTermSearcherBytes(indexReader, |
||||||
|
onBoundaryTerms, field, boost, options, false) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
// add filter to check points near the boundary
|
||||||
|
onBoundarySearcher = NewFilteringSearcher(rawOnBoundarySearcher, |
||||||
|
buildRectFilter(indexReader, field, minLon, minLat, maxLon, maxLat)) |
||||||
|
openedSearchers = append(openedSearchers, onBoundarySearcher) |
||||||
|
} |
||||||
|
|
||||||
|
var notOnBoundarySearcher search.Searcher |
||||||
|
if len(notOnBoundaryTerms) > 0 { |
||||||
|
var err error |
||||||
|
notOnBoundarySearcher, err = NewMultiTermSearcherBytes(indexReader, |
||||||
|
notOnBoundaryTerms, field, boost, options, false) |
||||||
|
if err != nil { |
||||||
|
cleanupOpenedSearchers() |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
openedSearchers = append(openedSearchers, notOnBoundarySearcher) |
||||||
|
} |
||||||
|
|
||||||
|
if onBoundarySearcher != nil && notOnBoundarySearcher != nil { |
||||||
|
rv, err := NewDisjunctionSearcher(indexReader, |
||||||
|
[]search.Searcher{ |
||||||
|
onBoundarySearcher, |
||||||
|
notOnBoundarySearcher, |
||||||
|
}, |
||||||
|
0, options) |
||||||
|
if err != nil { |
||||||
|
cleanupOpenedSearchers() |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
return rv, nil |
||||||
|
} else if onBoundarySearcher != nil { |
||||||
|
return onBoundarySearcher, nil |
||||||
|
} else if notOnBoundarySearcher != nil { |
||||||
|
return notOnBoundarySearcher, nil |
||||||
|
} |
||||||
|
|
||||||
|
return NewMatchNoneSearcher(indexReader) |
||||||
|
} |
||||||
|
|
||||||
|
var geoMaxShift = document.GeoPrecisionStep * 4 |
||||||
|
var geoDetailLevel = ((geo.GeoBits << 1) - geoMaxShift) / 2 |
||||||
|
|
||||||
|
func ComputeGeoRange(term uint64, shift uint, |
||||||
|
sminLon, sminLat, smaxLon, smaxLat float64, |
||||||
|
checkBoundaries bool) ( |
||||||
|
onBoundary [][]byte, notOnBoundary [][]byte) { |
||||||
|
split := term | uint64(0x1)<<shift |
||||||
|
var upperMax uint64 |
||||||
|
if shift < 63 { |
||||||
|
upperMax = term | ((uint64(1) << (shift + 1)) - 1) |
||||||
|
} else { |
||||||
|
upperMax = 0xffffffffffffffff |
||||||
|
} |
||||||
|
lowerMax := split - 1 |
||||||
|
onBoundary, notOnBoundary = relateAndRecurse(term, lowerMax, shift, |
||||||
|
sminLon, sminLat, smaxLon, smaxLat, checkBoundaries) |
||||||
|
plusOnBoundary, plusNotOnBoundary := relateAndRecurse(split, upperMax, shift, |
||||||
|
sminLon, sminLat, smaxLon, smaxLat, checkBoundaries) |
||||||
|
onBoundary = append(onBoundary, plusOnBoundary...) |
||||||
|
notOnBoundary = append(notOnBoundary, plusNotOnBoundary...) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
func relateAndRecurse(start, end uint64, res uint, |
||||||
|
sminLon, sminLat, smaxLon, smaxLat float64, |
||||||
|
checkBoundaries bool) ( |
||||||
|
onBoundary [][]byte, notOnBoundary [][]byte) { |
||||||
|
minLon := geo.MortonUnhashLon(start) |
||||||
|
minLat := geo.MortonUnhashLat(start) |
||||||
|
maxLon := geo.MortonUnhashLon(end) |
||||||
|
maxLat := geo.MortonUnhashLat(end) |
||||||
|
|
||||||
|
level := ((geo.GeoBits << 1) - res) >> 1 |
||||||
|
|
||||||
|
within := res%document.GeoPrecisionStep == 0 && |
||||||
|
geo.RectWithin(minLon, minLat, maxLon, maxLat, |
||||||
|
sminLon, sminLat, smaxLon, smaxLat) |
||||||
|
if within || (level == geoDetailLevel && |
||||||
|
geo.RectIntersects(minLon, minLat, maxLon, maxLat, |
||||||
|
sminLon, sminLat, smaxLon, smaxLat)) { |
||||||
|
if !within && checkBoundaries { |
||||||
|
return [][]byte{ |
||||||
|
numeric.MustNewPrefixCodedInt64(int64(start), res), |
||||||
|
}, nil |
||||||
|
} |
||||||
|
return nil, |
||||||
|
[][]byte{ |
||||||
|
numeric.MustNewPrefixCodedInt64(int64(start), res), |
||||||
|
} |
||||||
|
} else if level < geoDetailLevel && |
||||||
|
geo.RectIntersects(minLon, minLat, maxLon, maxLat, |
||||||
|
sminLon, sminLat, smaxLon, smaxLat) { |
||||||
|
return ComputeGeoRange(start, res-1, sminLon, sminLat, smaxLon, smaxLat, |
||||||
|
checkBoundaries) |
||||||
|
} |
||||||
|
return nil, nil |
||||||
|
} |
||||||
|
|
||||||
|
func buildRectFilter(indexReader index.IndexReader, field string, |
||||||
|
minLon, minLat, maxLon, maxLat float64) FilterFunc { |
||||||
|
return func(d *search.DocumentMatch) bool { |
||||||
|
var lon, lat float64 |
||||||
|
var found bool |
||||||
|
err := indexReader.DocumentVisitFieldTerms(d.IndexInternalID, |
||||||
|
[]string{field}, func(field string, term []byte) { |
||||||
|
// only consider the values which are shifted 0
|
||||||
|
prefixCoded := numeric.PrefixCoded(term) |
||||||
|
shift, err := prefixCoded.Shift() |
||||||
|
if err == nil && shift == 0 { |
||||||
|
var i64 int64 |
||||||
|
i64, err = prefixCoded.Int64() |
||||||
|
if err == nil { |
||||||
|
lon = geo.MortonUnhashLon(uint64(i64)) |
||||||
|
lat = geo.MortonUnhashLat(uint64(i64)) |
||||||
|
found = true |
||||||
|
} |
||||||
|
} |
||||||
|
}) |
||||||
|
if err == nil && found { |
||||||
|
return geo.BoundingBoxContains(lon, lat, |
||||||
|
minLon, minLat, maxLon, maxLat) |
||||||
|
} |
||||||
|
return false |
||||||
|
} |
||||||
|
} |
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue