168 lines
4.2 KiB
Go
168 lines
4.2 KiB
Go
// 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 upside_down
|
|
|
|
import (
|
|
"github.com/blevesearch/bleve/index"
|
|
"github.com/blevesearch/bleve/index/store"
|
|
)
|
|
|
|
type UpsideDownCouchTermFieldReader struct {
|
|
indexReader *IndexReader
|
|
iterator store.KVIterator
|
|
count uint64
|
|
term []byte
|
|
field uint16
|
|
}
|
|
|
|
func newUpsideDownCouchTermFieldReader(indexReader *IndexReader, term []byte, field uint16) (*UpsideDownCouchTermFieldReader, error) {
|
|
dictionaryRow := NewDictionaryRow(term, field, 0)
|
|
val, err := indexReader.kvreader.Get(dictionaryRow.Key())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if val == nil {
|
|
return &UpsideDownCouchTermFieldReader{
|
|
count: 0,
|
|
term: term,
|
|
field: field,
|
|
}, nil
|
|
}
|
|
|
|
err = dictionaryRow.parseDictionaryV(val)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
tfr := NewTermFrequencyRow(term, field, []byte{}, 0, 0)
|
|
it := indexReader.kvreader.PrefixIterator(tfr.Key())
|
|
|
|
return &UpsideDownCouchTermFieldReader{
|
|
indexReader: indexReader,
|
|
iterator: it,
|
|
count: dictionaryRow.count,
|
|
term: term,
|
|
field: field,
|
|
}, nil
|
|
}
|
|
|
|
func (r *UpsideDownCouchTermFieldReader) Count() uint64 {
|
|
return r.count
|
|
}
|
|
|
|
func (r *UpsideDownCouchTermFieldReader) Next() (*index.TermFieldDoc, error) {
|
|
if r.iterator != nil {
|
|
key, val, valid := r.iterator.Current()
|
|
if valid {
|
|
tfr, err := NewTermFrequencyRowKV(key, val)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rv := index.TermFieldDoc{
|
|
ID: string(tfr.doc),
|
|
Freq: tfr.freq,
|
|
Norm: float64(tfr.norm),
|
|
Vectors: r.indexReader.index.termFieldVectorsFromTermVectors(tfr.vectors),
|
|
}
|
|
r.iterator.Next()
|
|
return &rv, nil
|
|
}
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (r *UpsideDownCouchTermFieldReader) Advance(docID string) (*index.TermFieldDoc, error) {
|
|
if r.iterator != nil {
|
|
tfr := NewTermFrequencyRow(r.term, r.field, []byte(docID), 0, 0)
|
|
r.iterator.Seek(tfr.Key())
|
|
key, val, valid := r.iterator.Current()
|
|
if valid {
|
|
tfr, err := NewTermFrequencyRowKV(key, val)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rv := index.TermFieldDoc{
|
|
ID: string(tfr.doc),
|
|
Freq: tfr.freq,
|
|
Norm: float64(tfr.norm),
|
|
Vectors: r.indexReader.index.termFieldVectorsFromTermVectors(tfr.vectors),
|
|
}
|
|
r.iterator.Next()
|
|
return &rv, nil
|
|
}
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (r *UpsideDownCouchTermFieldReader) Close() error {
|
|
if r.iterator != nil {
|
|
return r.iterator.Close()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type UpsideDownCouchDocIDReader struct {
|
|
indexReader *IndexReader
|
|
iterator store.KVIterator
|
|
}
|
|
|
|
func newUpsideDownCouchDocIDReader(indexReader *IndexReader, start, end string) (*UpsideDownCouchDocIDReader, error) {
|
|
startBytes := []byte(start)
|
|
if start == "" {
|
|
startBytes = []byte{0x0}
|
|
}
|
|
endBytes := []byte(end)
|
|
if end == "" {
|
|
endBytes = []byte{0xff}
|
|
}
|
|
bisr := NewBackIndexRow(startBytes, nil, nil)
|
|
bier := NewBackIndexRow(endBytes, nil, nil)
|
|
it := indexReader.kvreader.RangeIterator(bisr.Key(), bier.Key())
|
|
|
|
return &UpsideDownCouchDocIDReader{
|
|
indexReader: indexReader,
|
|
iterator: it,
|
|
}, nil
|
|
}
|
|
|
|
func (r *UpsideDownCouchDocIDReader) Next() (string, error) {
|
|
key, val, valid := r.iterator.Current()
|
|
if valid {
|
|
br, err := NewBackIndexRowKV(key, val)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
rv := string(br.doc)
|
|
r.iterator.Next()
|
|
return rv, nil
|
|
}
|
|
return "", nil
|
|
}
|
|
|
|
func (r *UpsideDownCouchDocIDReader) Advance(docID string) (string, error) {
|
|
bir := NewBackIndexRow([]byte(docID), nil, nil)
|
|
r.iterator.Seek(bir.Key())
|
|
key, val, valid := r.iterator.Current()
|
|
if valid {
|
|
br, err := NewBackIndexRowKV(key, val)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
rv := string(br.doc)
|
|
r.iterator.Next()
|
|
return rv, nil
|
|
}
|
|
return "", nil
|
|
}
|
|
|
|
func (r *UpsideDownCouchDocIDReader) Close() error {
|
|
return r.iterator.Close()
|
|
}
|