c7443fe52b
more things can return error now in a couple of places we had to swallow errors because they didn't fit the existing API. in these case and proactively in a few others we now return error as well. also the batch API has been updated to allow performing set/delete internal within the batch
120 lines
2.5 KiB
Go
120 lines
2.5 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 index
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/blevesearch/bleve/document"
|
|
)
|
|
|
|
type Index interface {
|
|
Open() error
|
|
Close() error
|
|
|
|
DocCount() (uint64, error)
|
|
|
|
Update(doc *document.Document) error
|
|
Delete(id string) error
|
|
Batch(batch *Batch) error
|
|
|
|
SetInternal(key, val []byte) error
|
|
DeleteInternal(key []byte) error
|
|
|
|
DumpAll() chan interface{}
|
|
DumpDoc(id string) chan interface{}
|
|
DumpFields() chan interface{}
|
|
|
|
Reader() (IndexReader, error)
|
|
|
|
Stats() json.Marshaler
|
|
}
|
|
|
|
type IndexReader interface {
|
|
TermFieldReader(term []byte, field string) (TermFieldReader, error)
|
|
DocIDReader(start, end string) (DocIDReader, error)
|
|
|
|
FieldReader(field string, startTerm []byte, endTerm []byte) (FieldReader, error)
|
|
|
|
Document(id string) (*document.Document, error)
|
|
DocumentFieldTerms(id string) (FieldTerms, error)
|
|
|
|
Fields() ([]string, error)
|
|
|
|
GetInternal(key []byte) ([]byte, error)
|
|
|
|
DocCount() uint64
|
|
|
|
Close()
|
|
}
|
|
|
|
type FieldTerms map[string][]string
|
|
|
|
type TermFieldVector struct {
|
|
Field string
|
|
Pos uint64
|
|
Start uint64
|
|
End uint64
|
|
}
|
|
|
|
type TermFieldDoc struct {
|
|
Term string
|
|
ID string
|
|
Freq uint64
|
|
Norm float64
|
|
Vectors []*TermFieldVector
|
|
}
|
|
|
|
type TermFieldReader interface {
|
|
Next() (*TermFieldDoc, error)
|
|
Advance(ID string) (*TermFieldDoc, error)
|
|
Count() uint64
|
|
Close()
|
|
}
|
|
|
|
type FieldReader interface {
|
|
Next() (*TermFieldDoc, error)
|
|
Close()
|
|
}
|
|
|
|
type DocIDReader interface {
|
|
Next() (string, error)
|
|
Advance(ID string) (string, error)
|
|
Close()
|
|
}
|
|
|
|
type Batch struct {
|
|
IndexOps map[string]*document.Document
|
|
InternalOps map[string][]byte
|
|
}
|
|
|
|
func NewBatch() *Batch {
|
|
return &Batch{
|
|
IndexOps: make(map[string]*document.Document),
|
|
InternalOps: make(map[string][]byte),
|
|
}
|
|
}
|
|
|
|
func (b Batch) Update(doc *document.Document) {
|
|
b.IndexOps[doc.ID] = doc
|
|
}
|
|
|
|
func (b Batch) Delete(id string) {
|
|
b.IndexOps[id] = nil
|
|
}
|
|
|
|
func (b Batch) SetInternal(key, val []byte) {
|
|
b.InternalOps[string(key)] = val
|
|
}
|
|
|
|
func (b Batch) DeleteInternal(key []byte) {
|
|
b.InternalOps[string(key)] = nil
|
|
}
|