org.wikimedia.highlighter.experimental.elasticsearch.OffsetSnippetFormatter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of experimental-highlighter-elasticsearch-plugin Show documentation

Experimental Highlighter plugin

There is a newer version: 7.10.2

Show newest version

package org.wikimedia.highlighter.experimental.elasticsearch;

import java.util.Iterator;

import org.wikimedia.search.highlighter.experimental.Snippet;
import org.wikimedia.search.highlighter.experimental.Snippet.Hit;
import org.wikimedia.search.highlighter.experimental.SnippetFormatter;

public class OffsetSnippetFormatter implements SnippetFormatter {
    @Override
    public String format(Snippet snippet) {
        StringBuilder b = new StringBuilder();
        b.append(snippet.startOffset()).append(':');
        Iterator itr = snippet.hits().iterator();
        if (itr.hasNext()) {
            Hit hit = itr.next();
            b.append(hit.startOffset()).append('-').append(hit.endOffset());
            while(itr.hasNext()){
                hit = itr.next();
                b.append(',').append(hit.startOffset()).append('-').append(hit.endOffset());
            }
        }
        b.append(':').append(snippet.endOffset());
        return b.toString();
    }
}