forked from Shiloh/githaven
95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
|
// Copyright (c) 2019 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 bleve
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/blevesearch/bleve/document"
|
||
|
|
"github.com/blevesearch/bleve/index"
|
||
|
|
"github.com/blevesearch/bleve/index/scorch"
|
||
|
|
"github.com/blevesearch/bleve/mapping"
|
||
|
|
)
|
||
|
|
|
||
|
|
type builderImpl struct {
|
||
|
|
b index.IndexBuilder
|
||
|
|
m mapping.IndexMapping
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *builderImpl) Index(id string, data interface{}) error {
|
||
|
|
if id == "" {
|
||
|
|
return ErrorEmptyID
|
||
|
|
}
|
||
|
|
|
||
|
|
doc := document.NewDocument(id)
|
||
|
|
err := b.m.MapDocument(doc, data)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
err = b.b.Index(doc)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *builderImpl) Close() error {
|
||
|
|
return b.b.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
func newBuilder(path string, mapping mapping.IndexMapping, config map[string]interface{}) (Builder, error) {
|
||
|
|
if path == "" {
|
||
|
|
return nil, fmt.Errorf("builder requires path")
|
||
|
|
}
|
||
|
|
|
||
|
|
err := mapping.Validate()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
if config == nil {
|
||
|
|
config = map[string]interface{}{}
|
||
|
|
}
|
||
|
|
|
||
|
|
// the builder does not have an API to interact with internal storage
|
||
|
|
// however we can pass k/v pairs through the config
|
||
|
|
mappingBytes, err := json.Marshal(mapping)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
config["internal"] = map[string][]byte{
|
||
|
|
string(mappingInternalKey): mappingBytes,
|
||
|
|
}
|
||
|
|
|
||
|
|
// do not use real config, as these are options for the builder,
|
||
|
|
// not the resulting index
|
||
|
|
meta := newIndexMeta(scorch.Name, scorch.Name, map[string]interface{}{})
|
||
|
|
err = meta.Save(path)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
config["path"] = indexStorePath(path)
|
||
|
|
|
||
|
|
b, err := scorch.NewBuilder(config)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
rv := &builderImpl{
|
||
|
|
b: b,
|
||
|
|
m: mapping,
|
||
|
|
}
|
||
|
|
|
||
|
|
return rv, nil
|
||
|
|
}
|