org.graylog.plugins.views.search.elasticsearch.IndexRangeContainsOneOfStreams 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.graylog.plugins.views.search.elasticsearch;
import com.google.common.collect.ImmutableSet;
import org.graylog2.indexer.IndexSet;
import org.graylog2.indexer.ranges.IndexRange;
import org.graylog2.plugin.streams.Stream;
import java.util.Collections;
import java.util.Set;
import java.util.function.Predicate;
import java.util.stream.Collectors;
public class IndexRangeContainsOneOfStreams implements Predicate {
private final Set validIndexSets;
private final Set validStreamIds;
public IndexRangeContainsOneOfStreams(Set validStreams) {
this.validStreamIds = validStreams.stream().map(Stream::getId).collect(Collectors.toSet());
this.validIndexSets = validStreams.stream().map(Stream::getIndexSet).collect(Collectors.toSet());
}
IndexRangeContainsOneOfStreams(Stream... validStreams) {
this(ImmutableSet.copyOf(validStreams));
}
@Override
public boolean test(IndexRange indexRange) {
if (validIndexSets.isEmpty() && validStreamIds.isEmpty()) {
return false;
}
// If index range is incomplete, check the prefix against the valid index sets.
if (indexRange.streamIds() == null) {
return validIndexSets.stream().anyMatch(indexSet -> indexSet.isManagedIndex(indexRange.indexName()));
}
// Otherwise check if the index range contains any of the valid stream ids.
return !Collections.disjoint(indexRange.streamIds(), validStreamIds);
}
}