0
0

fix api changes and test after merge

This commit is contained in:
Marty Schoch 2016-09-09 14:11:28 -04:00
parent 1b68c4ec5b
commit 36000f1a1b
2 changed files with 2 additions and 15 deletions

View File

@ -98,7 +98,7 @@ func (i *IndexReader) Document(id string) (doc *document.Document, err error) {
return return
} }
func (i *IndexReader) DocumentFieldTerms(id index.IndexInternalID) (index.FieldTerms, error) { func (i *IndexReader) DocumentFieldTerms(id index.IndexInternalID, fields []string) (index.FieldTerms, error) {
back, err := i.index.backIndexRowForDoc(i, id, "") back, err := i.index.backIndexRowForDoc(i, id, "")
if err != nil { if err != nil {
return nil, err return nil, err
@ -106,19 +106,6 @@ func (i *IndexReader) DocumentFieldTerms(id index.IndexInternalID) (index.FieldT
if back == nil { if back == nil {
return nil, nil return nil, nil
} }
rv := make(index.FieldTerms, len(back.termsEntries))
for _, entry := range back.termsEntries {
fieldName := i.index.fieldCache.FieldIndexed(uint16(*entry.Field))
rv[fieldName] = entry.Terms
}
return rv, nil
}
func (i *IndexReader) DocumentFieldTermsForFields(id index.IndexInternalID, fields []string) (index.FieldTerms, error) {
back, err := i.index.backIndexRowForDoc(i, id, "")
if err != nil {
return nil, err
}
rv := make(index.FieldTerms, len(fields)) rv := make(index.FieldTerms, len(fields))
fieldsMap := make(map[uint16]string, len(fields)) fieldsMap := make(map[uint16]string, len(fields))
for _, f := range fields { for _, f := range fields {

View File

@ -1229,7 +1229,7 @@ func TestIndexDocumentFieldTerms(t *testing.T) {
} }
}() }()
fieldTerms, err := indexReader.DocumentFieldTerms(EncodeUvarintAscending(nil, 1)) fieldTerms, err := indexReader.DocumentFieldTerms(EncodeUvarintAscending(nil, 1), []string{"_id", "name", "title"})
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} }