org.graylog2.indexer.results.HighlightParser Maven / Gradle / Ivy
/**
* This file is part of Graylog.
*
* Graylog is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Graylog is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Graylog. If not, see .
*/
package org.graylog2.indexer.results;
import com.google.common.collect.ImmutableListMultimap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Multimap;
import com.google.common.collect.Range;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
final class HighlightParser {
private static final String startToken = "";
private static final String endToken = "";
private static final Pattern highlightPattern = Pattern.compile(startToken + "(.*?)" + endToken);
private static final Integer startTokenLength = startToken.length();
private static final Integer endTokenLength = endToken.length();
private HighlightParser() {}
static Multimap> extractHighlightRanges(Map> highlight) {
if (highlight == null || highlight.isEmpty()) {
return ImmutableListMultimap.of();
}
final ImmutableListMultimap.Builder> builder = ImmutableListMultimap.builder();
highlight.forEach((key, value) -> extractRange(value).forEach(range -> builder.put(key, range)));
return builder.build();
}
private static Set> extractRange(List highlights) {
final ImmutableSet.Builder> builder = ImmutableSet.builder();
highlights.forEach(highlight -> {
final Matcher matcher = highlightPattern.matcher(highlight);
Integer count = -1;
while (matcher.find()) {
count++;
final Integer start = matcher.start() - count * (startTokenLength + endTokenLength);
final Integer end = start + (matcher.end(1) - matcher.start(1));
builder.add(Range.closed(start, end));
}
});
return builder.build();
}
}