another round of golint cleanup
This commit is contained in:
parent
e1b77956d4
commit
e21935f850
@ -18,28 +18,28 @@ type op struct {
|
||||
v []byte
|
||||
}
|
||||
|
||||
type BoltDBBatch struct {
|
||||
store *BoltDBStore
|
||||
type Batch struct {
|
||||
store *Store
|
||||
ops []op
|
||||
}
|
||||
|
||||
func newBoltDBBatch(store *BoltDBStore) *BoltDBBatch {
|
||||
rv := BoltDBBatch{
|
||||
func newBatch(store *Store) *Batch {
|
||||
rv := Batch{
|
||||
store: store,
|
||||
ops: make([]op, 0),
|
||||
}
|
||||
return &rv
|
||||
}
|
||||
|
||||
func (i *BoltDBBatch) Set(key, val []byte) {
|
||||
func (i *Batch) Set(key, val []byte) {
|
||||
i.ops = append(i.ops, op{key, val})
|
||||
}
|
||||
|
||||
func (i *BoltDBBatch) Delete(key []byte) {
|
||||
func (i *Batch) Delete(key []byte) {
|
||||
i.ops = append(i.ops, op{key, nil})
|
||||
}
|
||||
|
||||
func (i *BoltDBBatch) Execute() error {
|
||||
func (i *Batch) Execute() error {
|
||||
return i.store.db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte(i.store.bucket))
|
||||
|
||||
@ -59,6 +59,6 @@ func (i *BoltDBBatch) Execute() error {
|
||||
})
|
||||
}
|
||||
|
||||
func (i *BoltDBBatch) Close() error {
|
||||
func (i *Batch) Close() error {
|
||||
return nil
|
||||
}
|
||||
|
@ -13,8 +13,8 @@ import (
|
||||
"github.com/boltdb/bolt"
|
||||
)
|
||||
|
||||
type BoltDBIterator struct {
|
||||
store *BoltDBStore
|
||||
type Iterator struct {
|
||||
store *Store
|
||||
tx *bolt.Tx
|
||||
cursor *bolt.Cursor
|
||||
valid bool
|
||||
@ -22,52 +22,52 @@ type BoltDBIterator struct {
|
||||
val []byte
|
||||
}
|
||||
|
||||
func newBoltDBIterator(store *BoltDBStore) *BoltDBIterator {
|
||||
func newIterator(store *Store) *Iterator {
|
||||
tx, _ := store.db.Begin(false)
|
||||
b := tx.Bucket([]byte(store.bucket))
|
||||
cursor := b.Cursor()
|
||||
|
||||
return &BoltDBIterator{
|
||||
return &Iterator{
|
||||
store: store,
|
||||
tx: tx,
|
||||
cursor: cursor,
|
||||
}
|
||||
}
|
||||
|
||||
func (i *BoltDBIterator) SeekFirst() {
|
||||
func (i *Iterator) SeekFirst() {
|
||||
i.key, i.val = i.cursor.First()
|
||||
|
||||
i.valid = (i.key != nil)
|
||||
}
|
||||
|
||||
func (i *BoltDBIterator) Seek(k []byte) {
|
||||
func (i *Iterator) Seek(k []byte) {
|
||||
i.key, i.val = i.cursor.Seek(k)
|
||||
|
||||
i.valid = (i.key != nil)
|
||||
}
|
||||
|
||||
func (i *BoltDBIterator) Next() {
|
||||
func (i *Iterator) Next() {
|
||||
i.key, i.val = i.cursor.Next()
|
||||
|
||||
i.valid = (i.key != nil)
|
||||
}
|
||||
|
||||
func (i *BoltDBIterator) Current() ([]byte, []byte, bool) {
|
||||
func (i *Iterator) Current() ([]byte, []byte, bool) {
|
||||
return i.key, i.val, i.valid
|
||||
}
|
||||
|
||||
func (i *BoltDBIterator) Key() []byte {
|
||||
func (i *Iterator) Key() []byte {
|
||||
return i.key
|
||||
}
|
||||
|
||||
func (i *BoltDBIterator) Value() []byte {
|
||||
func (i *Iterator) Value() []byte {
|
||||
return i.val
|
||||
}
|
||||
|
||||
func (i *BoltDBIterator) Valid() bool {
|
||||
func (i *Iterator) Valid() bool {
|
||||
return i.valid
|
||||
}
|
||||
|
||||
func (i *BoltDBIterator) Close() {
|
||||
func (i *Iterator) Close() {
|
||||
i.tx.Rollback()
|
||||
}
|
||||
|
@ -19,14 +19,14 @@ import (
|
||||
|
||||
const Name = "boltdb"
|
||||
|
||||
type BoltDBStore struct {
|
||||
type Store struct {
|
||||
path string
|
||||
bucket string
|
||||
db *bolt.DB
|
||||
}
|
||||
|
||||
func Open(path string, bucket string) (*BoltDBStore, error) {
|
||||
rv := BoltDBStore{
|
||||
func Open(path string, bucket string) (*Store, error) {
|
||||
rv := Store{
|
||||
path: path,
|
||||
bucket: bucket,
|
||||
}
|
||||
@ -49,7 +49,7 @@ func Open(path string, bucket string) (*BoltDBStore, error) {
|
||||
return &rv, nil
|
||||
}
|
||||
|
||||
func (bs *BoltDBStore) Get(key []byte) ([]byte, error) {
|
||||
func (bs *Store) Get(key []byte) ([]byte, error) {
|
||||
var rv []byte
|
||||
|
||||
err := bs.db.View(func(tx *bolt.Tx) error {
|
||||
@ -61,34 +61,34 @@ func (bs *BoltDBStore) Get(key []byte) ([]byte, error) {
|
||||
return rv, err
|
||||
}
|
||||
|
||||
func (bs *BoltDBStore) Set(key, val []byte) error {
|
||||
func (bs *Store) Set(key, val []byte) error {
|
||||
return bs.db.Update(func(tx *bolt.Tx) error {
|
||||
return tx.Bucket([]byte(bs.bucket)).Put(key, val)
|
||||
})
|
||||
}
|
||||
|
||||
func (bs *BoltDBStore) Delete(key []byte) error {
|
||||
func (bs *Store) Delete(key []byte) error {
|
||||
return bs.db.Update(func(tx *bolt.Tx) error {
|
||||
return tx.Bucket([]byte(bs.bucket)).Delete(key)
|
||||
})
|
||||
}
|
||||
|
||||
func (bs *BoltDBStore) Commit() error {
|
||||
func (bs *Store) Commit() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (bs *BoltDBStore) Close() error {
|
||||
func (bs *Store) Close() error {
|
||||
return bs.db.Close()
|
||||
}
|
||||
|
||||
func (bs *BoltDBStore) Iterator(key []byte) store.KVIterator {
|
||||
rv := newBoltDBIterator(bs)
|
||||
func (bs *Store) Iterator(key []byte) store.KVIterator {
|
||||
rv := newIterator(bs)
|
||||
rv.Seek(key)
|
||||
return rv
|
||||
}
|
||||
|
||||
func (bs *BoltDBStore) NewBatch() store.KVBatch {
|
||||
return newBoltDBBatch(bs)
|
||||
func (bs *Store) NewBatch() store.KVBatch {
|
||||
return newBatch(bs)
|
||||
}
|
||||
|
||||
func StoreConstructor(config map[string]interface{}) (store.KVStore, error) {
|
||||
|
@ -16,7 +16,7 @@ import (
|
||||
"github.com/blevesearch/bleve/index/store/test"
|
||||
)
|
||||
|
||||
func TestBoltDBStore(t *testing.T) {
|
||||
func TestStore(t *testing.T) {
|
||||
s, err := Open("test", "bleve")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
|
@ -9,14 +9,14 @@
|
||||
|
||||
package inmem
|
||||
|
||||
type InMemBatch struct {
|
||||
store *InMemStore
|
||||
type Batch struct {
|
||||
store *Store
|
||||
keys [][]byte
|
||||
vals [][]byte
|
||||
}
|
||||
|
||||
func newInMemBatch(store *InMemStore) *InMemBatch {
|
||||
rv := InMemBatch{
|
||||
func newBatch(store *Store) *Batch {
|
||||
rv := Batch{
|
||||
store: store,
|
||||
keys: make([][]byte, 0),
|
||||
vals: make([][]byte, 0),
|
||||
@ -24,17 +24,17 @@ func newInMemBatch(store *InMemStore) *InMemBatch {
|
||||
return &rv
|
||||
}
|
||||
|
||||
func (i *InMemBatch) Set(key, val []byte) {
|
||||
func (i *Batch) Set(key, val []byte) {
|
||||
i.keys = append(i.keys, key)
|
||||
i.vals = append(i.vals, val)
|
||||
}
|
||||
|
||||
func (i *InMemBatch) Delete(key []byte) {
|
||||
func (i *Batch) Delete(key []byte) {
|
||||
i.keys = append(i.keys, key)
|
||||
i.vals = append(i.vals, nil)
|
||||
}
|
||||
|
||||
func (i *InMemBatch) Execute() error {
|
||||
func (i *Batch) Execute() error {
|
||||
for index, key := range i.keys {
|
||||
val := i.vals[index]
|
||||
if val == nil {
|
||||
@ -46,6 +46,6 @@ func (i *InMemBatch) Execute() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (i *InMemBatch) Close() error {
|
||||
func (i *Batch) Close() error {
|
||||
return nil
|
||||
}
|
||||
|
@ -13,57 +13,57 @@ import (
|
||||
"github.com/ryszard/goskiplist/skiplist"
|
||||
)
|
||||
|
||||
type InMemIterator struct {
|
||||
store *InMemStore
|
||||
type Iterator struct {
|
||||
store *Store
|
||||
iterator skiplist.Iterator
|
||||
valid bool
|
||||
}
|
||||
|
||||
func newInMemIterator(store *InMemStore) *InMemIterator {
|
||||
rv := InMemIterator{
|
||||
func newIterator(store *Store) *Iterator {
|
||||
rv := Iterator{
|
||||
store: store,
|
||||
iterator: store.list.Iterator(),
|
||||
}
|
||||
return &rv
|
||||
}
|
||||
|
||||
func (i *InMemIterator) SeekFirst() {
|
||||
func (i *Iterator) SeekFirst() {
|
||||
i.Seek([]byte{0})
|
||||
}
|
||||
|
||||
func (i *InMemIterator) Seek(k []byte) {
|
||||
func (i *Iterator) Seek(k []byte) {
|
||||
i.valid = i.iterator.Seek(string(k))
|
||||
}
|
||||
|
||||
func (i *InMemIterator) Next() {
|
||||
func (i *Iterator) Next() {
|
||||
i.valid = i.iterator.Next()
|
||||
}
|
||||
|
||||
func (i *InMemIterator) Current() ([]byte, []byte, bool) {
|
||||
func (i *Iterator) Current() ([]byte, []byte, bool) {
|
||||
if i.valid {
|
||||
return []byte(i.Key()), []byte(i.Value()), true
|
||||
}
|
||||
return nil, nil, false
|
||||
}
|
||||
|
||||
func (i *InMemIterator) Key() []byte {
|
||||
func (i *Iterator) Key() []byte {
|
||||
if i.valid {
|
||||
return []byte(i.iterator.Key().(string))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (i *InMemIterator) Value() []byte {
|
||||
func (i *Iterator) Value() []byte {
|
||||
if i.valid {
|
||||
return []byte(i.iterator.Value().(string))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (i *InMemIterator) Valid() bool {
|
||||
func (i *Iterator) Valid() bool {
|
||||
return i.valid
|
||||
}
|
||||
|
||||
func (i *InMemIterator) Close() {
|
||||
func (i *Iterator) Close() {
|
||||
i.iterator.Close()
|
||||
}
|
||||
|
@ -17,27 +17,27 @@ import (
|
||||
|
||||
const Name = "mem"
|
||||
|
||||
type InMemStore struct {
|
||||
type Store struct {
|
||||
list *skiplist.SkipList
|
||||
}
|
||||
|
||||
func Open() (*InMemStore, error) {
|
||||
rv := InMemStore{
|
||||
func Open() (*Store, error) {
|
||||
rv := Store{
|
||||
list: skiplist.NewStringMap(),
|
||||
}
|
||||
|
||||
return &rv, nil
|
||||
}
|
||||
|
||||
func MustOpen() *InMemStore {
|
||||
rv := InMemStore{
|
||||
func MustOpen() *Store {
|
||||
rv := Store{
|
||||
list: skiplist.NewStringMap(),
|
||||
}
|
||||
|
||||
return &rv
|
||||
}
|
||||
|
||||
func (i *InMemStore) Get(key []byte) ([]byte, error) {
|
||||
func (i *Store) Get(key []byte) ([]byte, error) {
|
||||
val, ok := i.list.Get(string(key))
|
||||
if ok {
|
||||
return []byte(val.(string)), nil
|
||||
@ -45,32 +45,32 @@ func (i *InMemStore) Get(key []byte) ([]byte, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (i *InMemStore) Set(key, val []byte) error {
|
||||
func (i *Store) Set(key, val []byte) error {
|
||||
i.list.Set(string(key), string(val))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (i *InMemStore) Delete(key []byte) error {
|
||||
func (i *Store) Delete(key []byte) error {
|
||||
i.list.Delete(string(key))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (i *InMemStore) Commit() error {
|
||||
func (i *Store) Commit() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (i *InMemStore) Close() error {
|
||||
func (i *Store) Close() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (i *InMemStore) Iterator(key []byte) store.KVIterator {
|
||||
rv := newInMemIterator(i)
|
||||
func (i *Store) Iterator(key []byte) store.KVIterator {
|
||||
rv := newIterator(i)
|
||||
rv.Seek(key)
|
||||
return rv
|
||||
}
|
||||
|
||||
func (i *InMemStore) NewBatch() store.KVBatch {
|
||||
return newInMemBatch(i)
|
||||
func (i *Store) NewBatch() store.KVBatch {
|
||||
return newBatch(i)
|
||||
}
|
||||
|
||||
func StoreConstructor(config map[string]interface{}) (store.KVStore, error) {
|
||||
|
@ -15,7 +15,7 @@ import (
|
||||
"github.com/blevesearch/bleve/index/store/test"
|
||||
)
|
||||
|
||||
func TestInMemStore(t *testing.T) {
|
||||
func TestStore(t *testing.T) {
|
||||
s, err := Open()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
|
@ -19,17 +19,17 @@ const Name = "ansi"
|
||||
|
||||
const DefaultAnsiHighlight = bgYellow
|
||||
|
||||
type ANSIFragmentFormatter struct {
|
||||
type FragmentFormatter struct {
|
||||
color string
|
||||
}
|
||||
|
||||
func NewANSIFragmentFormatter() *ANSIFragmentFormatter {
|
||||
return &ANSIFragmentFormatter{
|
||||
func NewFragmentFormatter() *FragmentFormatter {
|
||||
return &FragmentFormatter{
|
||||
color: DefaultAnsiHighlight,
|
||||
}
|
||||
}
|
||||
|
||||
func (a *ANSIFragmentFormatter) Format(f *highlight.Fragment, tlm search.TermLocationMap) string {
|
||||
func (a *FragmentFormatter) Format(f *highlight.Fragment, tlm search.TermLocationMap) string {
|
||||
orderedTermLocations := highlight.OrderTermLocations(tlm)
|
||||
rv := ""
|
||||
curr := f.Start
|
||||
@ -86,7 +86,7 @@ const (
|
||||
)
|
||||
|
||||
func Constructor(config map[string]interface{}, cache *registry.Cache) (highlight.FragmentFormatter, error) {
|
||||
return NewANSIFragmentFormatter(), nil
|
||||
return NewFragmentFormatter(), nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
|
@ -20,19 +20,19 @@ const Name = "html"
|
||||
const defaultHTMLHighlightBefore = "<b>"
|
||||
const defaultHTMLHighlightAfter = "</b>"
|
||||
|
||||
type HTMLFragmentFormatter struct {
|
||||
type FragmentFormatter struct {
|
||||
before string
|
||||
after string
|
||||
}
|
||||
|
||||
func NewHTMLFragmentFormatter(before, after string) *HTMLFragmentFormatter {
|
||||
return &HTMLFragmentFormatter{
|
||||
func NewFragmentFormatter(before, after string) *FragmentFormatter {
|
||||
return &FragmentFormatter{
|
||||
before: before,
|
||||
after: after,
|
||||
}
|
||||
}
|
||||
|
||||
func (a *HTMLFragmentFormatter) Format(f *highlight.Fragment, tlm search.TermLocationMap) string {
|
||||
func (a *FragmentFormatter) Format(f *highlight.Fragment, tlm search.TermLocationMap) string {
|
||||
orderedTermLocations := highlight.OrderTermLocations(tlm)
|
||||
rv := ""
|
||||
curr := f.Start
|
||||
@ -71,7 +71,7 @@ func Constructor(config map[string]interface{}, cache *registry.Cache) (highligh
|
||||
if ok {
|
||||
after = afterVal
|
||||
}
|
||||
return NewHTMLFragmentFormatter(before, after), nil
|
||||
return NewFragmentFormatter(before, after), nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
|
@ -41,7 +41,7 @@ func TestHTMLFragmentFormatter1(t *testing.T) {
|
||||
},
|
||||
}
|
||||
|
||||
emHTMLFormatter := NewHTMLFragmentFormatter("<b>", "</b>")
|
||||
emHTMLFormatter := NewFragmentFormatter("<b>", "</b>")
|
||||
for _, test := range tests {
|
||||
result := emHTMLFormatter.Format(test.fragment, test.tlm)
|
||||
if result != test.output {
|
||||
@ -75,7 +75,7 @@ func TestHTMLFragmentFormatter2(t *testing.T) {
|
||||
},
|
||||
}
|
||||
|
||||
emHTMLFormatter := NewHTMLFragmentFormatter("<em>", "</em>")
|
||||
emHTMLFormatter := NewFragmentFormatter("<em>", "</em>")
|
||||
for _, test := range tests {
|
||||
result := emHTMLFormatter.Format(test.fragment, test.tlm)
|
||||
if result != test.output {
|
||||
|
@ -18,17 +18,17 @@ const Name = "simple"
|
||||
|
||||
const defaultFragmentSize = 200
|
||||
|
||||
type SimpleFragmenter struct {
|
||||
type Fragmenter struct {
|
||||
fragmentSize int
|
||||
}
|
||||
|
||||
func NewSimpleFragmenter(fragmentSize int) *SimpleFragmenter {
|
||||
return &SimpleFragmenter{
|
||||
func NewFragmenter(fragmentSize int) *Fragmenter {
|
||||
return &Fragmenter{
|
||||
fragmentSize: fragmentSize,
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SimpleFragmenter) Fragment(orig []byte, ot highlight.TermLocations) []*highlight.Fragment {
|
||||
func (s *Fragmenter) Fragment(orig []byte, ot highlight.TermLocations) []*highlight.Fragment {
|
||||
rv := make([]*highlight.Fragment, 0)
|
||||
|
||||
maxbegin := 0
|
||||
@ -91,7 +91,7 @@ func Constructor(config map[string]interface{}, cache *registry.Cache) (highligh
|
||||
if ok {
|
||||
size = int(sizeVal)
|
||||
}
|
||||
return NewSimpleFragmenter(size), nil
|
||||
return NewFragmenter(size), nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
|
@ -178,7 +178,7 @@ func TestSimpleFragmenter(t *testing.T) {
|
||||
},
|
||||
}
|
||||
|
||||
fragmenter := NewSimpleFragmenter(100)
|
||||
fragmenter := NewFragmenter(100)
|
||||
for _, test := range tests {
|
||||
fragments := fragmenter.Fragment(test.orig, test.ot)
|
||||
if !reflect.DeepEqual(fragments, test.fragments) {
|
||||
@ -228,7 +228,7 @@ func TestSimpleFragmenterWithSize(t *testing.T) {
|
||||
},
|
||||
}
|
||||
|
||||
fragmenter := NewSimpleFragmenter(5)
|
||||
fragmenter := NewFragmenter(5)
|
||||
for _, test := range tests {
|
||||
fragments := fragmenter.Fragment(test.orig, test.ot)
|
||||
if !reflect.DeepEqual(fragments, test.fragments) {
|
||||
|
@ -14,20 +14,20 @@ import (
|
||||
"github.com/blevesearch/bleve/search/highlight"
|
||||
)
|
||||
|
||||
// SimpleFragmentScorer will score fragments by how many
|
||||
// FragmentScorer will score fragments by how many
|
||||
// unique terms occur in the fragment with no regard for
|
||||
// any boost values used in the original query
|
||||
type SimpleFragmentScorer struct {
|
||||
type FragmentScorer struct {
|
||||
tlm search.TermLocationMap
|
||||
}
|
||||
|
||||
func NewSimpleFragmentScorer(tlm search.TermLocationMap) *SimpleFragmentScorer {
|
||||
return &SimpleFragmentScorer{
|
||||
func NewFragmentScorer(tlm search.TermLocationMap) *FragmentScorer {
|
||||
return &FragmentScorer{
|
||||
tlm: tlm,
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SimpleFragmentScorer) Score(f *highlight.Fragment) {
|
||||
func (s *FragmentScorer) Score(f *highlight.Fragment) {
|
||||
score := 0.0
|
||||
OUTER:
|
||||
for _, locations := range s.tlm {
|
||||
|
@ -67,7 +67,7 @@ func TestSimpleFragmentScorer(t *testing.T) {
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
scorer := NewSimpleFragmentScorer(test.tlm)
|
||||
scorer := NewFragmentScorer(test.tlm)
|
||||
scorer.Score(test.fragment)
|
||||
if test.fragment.Score != test.score {
|
||||
t.Errorf("expected score %f, got %f", test.score, test.fragment.Score)
|
||||
|
@ -22,45 +22,45 @@ import (
|
||||
const Name = "simple"
|
||||
const defaultSeparator = "…"
|
||||
|
||||
type SimpleHighlighter struct {
|
||||
type Highlighter struct {
|
||||
fragmenter highlight.Fragmenter
|
||||
formatter highlight.FragmentFormatter
|
||||
sep string
|
||||
}
|
||||
|
||||
func NewSimpleHighlighter(fragmenter highlight.Fragmenter, formatter highlight.FragmentFormatter, separator string) *SimpleHighlighter {
|
||||
return &SimpleHighlighter{
|
||||
func NewHighlighter(fragmenter highlight.Fragmenter, formatter highlight.FragmentFormatter, separator string) *Highlighter {
|
||||
return &Highlighter{
|
||||
fragmenter: fragmenter,
|
||||
formatter: formatter,
|
||||
sep: separator,
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SimpleHighlighter) Fragmenter() highlight.Fragmenter {
|
||||
func (s *Highlighter) Fragmenter() highlight.Fragmenter {
|
||||
return s.fragmenter
|
||||
}
|
||||
|
||||
func (s *SimpleHighlighter) SetFragmenter(f highlight.Fragmenter) {
|
||||
func (s *Highlighter) SetFragmenter(f highlight.Fragmenter) {
|
||||
s.fragmenter = f
|
||||
}
|
||||
|
||||
func (s *SimpleHighlighter) FragmentFormatter() highlight.FragmentFormatter {
|
||||
func (s *Highlighter) FragmentFormatter() highlight.FragmentFormatter {
|
||||
return s.formatter
|
||||
}
|
||||
|
||||
func (s *SimpleHighlighter) SetFragmentFormatter(f highlight.FragmentFormatter) {
|
||||
func (s *Highlighter) SetFragmentFormatter(f highlight.FragmentFormatter) {
|
||||
s.formatter = f
|
||||
}
|
||||
|
||||
func (s *SimpleHighlighter) Separator() string {
|
||||
func (s *Highlighter) Separator() string {
|
||||
return s.sep
|
||||
}
|
||||
|
||||
func (s *SimpleHighlighter) SetSeparator(sep string) {
|
||||
func (s *Highlighter) SetSeparator(sep string) {
|
||||
s.sep = sep
|
||||
}
|
||||
|
||||
func (s *SimpleHighlighter) BestFragmentInField(dm *search.DocumentMatch, doc *document.Document, field string) string {
|
||||
func (s *Highlighter) BestFragmentInField(dm *search.DocumentMatch, doc *document.Document, field string) string {
|
||||
fragments := s.BestFragmentsInField(dm, doc, field, 1)
|
||||
if len(fragments) > 0 {
|
||||
return fragments[0]
|
||||
@ -68,10 +68,10 @@ func (s *SimpleHighlighter) BestFragmentInField(dm *search.DocumentMatch, doc *d
|
||||
return ""
|
||||
}
|
||||
|
||||
func (s *SimpleHighlighter) BestFragmentsInField(dm *search.DocumentMatch, doc *document.Document, field string, num int) []string {
|
||||
func (s *Highlighter) BestFragmentsInField(dm *search.DocumentMatch, doc *document.Document, field string, num int) []string {
|
||||
tlm := dm.Locations[field]
|
||||
orderedTermLocations := highlight.OrderTermLocations(tlm)
|
||||
scorer := NewSimpleFragmentScorer(dm.Locations[field])
|
||||
scorer := NewFragmentScorer(dm.Locations[field])
|
||||
|
||||
// score the fragments and put them into a priority queue ordered by score
|
||||
fq := make(FragmentQueue, 0)
|
||||
@ -197,7 +197,7 @@ func Constructor(config map[string]interface{}, cache *registry.Cache) (highligh
|
||||
return nil, fmt.Errorf("error building fragment formatter: %v", err)
|
||||
}
|
||||
|
||||
return NewSimpleHighlighter(fragmenter, formatter, separator), nil
|
||||
return NewHighlighter(fragmenter, formatter, separator), nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
|
@ -25,9 +25,9 @@ const (
|
||||
)
|
||||
|
||||
func TestSimpleHighlighter(t *testing.T) {
|
||||
fragmenter := sfrag.NewSimpleFragmenter(100)
|
||||
formatter := ansi.NewANSIFragmentFormatter()
|
||||
highlighter := NewSimpleHighlighter(fragmenter, formatter, defaultSeparator)
|
||||
fragmenter := sfrag.NewFragmenter(100)
|
||||
formatter := ansi.NewFragmentFormatter()
|
||||
highlighter := NewHighlighter(fragmenter, formatter, defaultSeparator)
|
||||
|
||||
docMatch := search.DocumentMatch{
|
||||
ID: "a",
|
||||
@ -152,9 +152,9 @@ Etiam vel augue vel nisl commodo suscipit et ac nisl. Quisque eros diam, porttit
|
||||
"… accumsan. Vivamus eros felis, rhoncus vel " + DefaultAnsiHighlight + "interdum" + reset + " bibendum, imperdiet nec diam. Etiam sed eros sed…",
|
||||
}
|
||||
|
||||
fragmenter := sfrag.NewSimpleFragmenter(100)
|
||||
formatter := ansi.NewANSIFragmentFormatter()
|
||||
highlighter := NewSimpleHighlighter(fragmenter, formatter, defaultSeparator)
|
||||
fragmenter := sfrag.NewFragmenter(100)
|
||||
formatter := ansi.NewFragmentFormatter()
|
||||
highlighter := NewHighlighter(fragmenter, formatter, defaultSeparator)
|
||||
fragments := highlighter.BestFragmentsInField(&docMatch, doc, "full", 5)
|
||||
|
||||
if !reflect.DeepEqual(fragments, expectedFragments) {
|
||||
|
Loading…
Reference in New Issue
Block a user