Examples with NodesSniffer org.elasticsearch.client.sniff.NodesSniffer used on opensource projects

Example 1 with NodesSniffer

use of org.elasticsearch.client.sniff.NodesSniffer in project graylog2-server by Graylog2.

the class FilteredElasticsearchNodesSnifferTest method returnsAllNodesIfFilterMatchesAll.

@Test
void returnsAllNodesIfFilterMatchesAll() throws Exception {
    final List<Node> nodes = mockNodes();
    final NodesSniffer nodesSniffer = new FilteredElasticsearchNodesSniffer(mockSniffer(nodes), "always", "true");
    assertThat(nodesSniffer.sniff()).isEqualTo(nodes);
}

Example 2 with NodesSniffer

use of org.elasticsearch.client.sniff.NodesSniffer in project graylog2-server by Graylog2.

the class FilteredElasticsearchNodesSnifferTest method doesNotFilterNodesIfNoFilterIsSet.

@Test
void doesNotFilterNodesIfNoFilterIsSet() throws Exception {
    final List<Node> nodes = mockNodes();
    final NodesSniffer nodesSniffer = new FilteredElasticsearchNodesSniffer(mockSniffer(nodes), null, null);
    assertThat(nodesSniffer.sniff()).isEqualTo(nodes);
}

Example 3 with NodesSniffer

use of org.elasticsearch.client.sniff.NodesSniffer in project graylog2-server by Graylog2.

the class FilteredElasticsearchNodesSnifferTest method returnsMatchingNodesIfGivenAttributeIsInList.

@Test
void returnsMatchingNodesIfGivenAttributeIsInList() throws Exception {
    final Node matchingNode = mockNode(ImmutableMap.of("something", ImmutableList.of("somevalue", "42", "pi")));
    final List<Node> nodes = Collections.singletonList(matchingNode);
    final NodesSniffer nodesSniffer = new FilteredElasticsearchNodesSniffer(mockSniffer(nodes), "something", "42");
    assertThat(nodesSniffer.sniff()).isEqualTo(nodes);
}

Example 4 with NodesSniffer

use of org.elasticsearch.client.sniff.NodesSniffer in project graylog2-server by Graylog2.

the class FilteredElasticsearchNodesSnifferTest method mockSniffer.

private NodesSniffer mockSniffer(List<Node> nodes) throws IOException {
    final NodesSniffer mockSniffer = mock(NodesSniffer.class);
    when(mockSniffer.sniff()).thenReturn(nodes);
    return mockSniffer;
}

Example 5 with NodesSniffer

use of org.elasticsearch.client.sniff.NodesSniffer in project graylog2-server by Graylog2.

the class FilteredElasticsearchNodesSniffer method create.

static FilteredElasticsearchNodesSniffer create(RestClient restClient, long sniffRequestTimeoutMillis, ElasticsearchNodesSniffer.Scheme scheme, String filter) {
    final String attribute;
    final String value;
    if (!Strings.isNullOrEmpty(filter)) {
        final String[] conditions = filter.split(":");
        if (conditions.length < 2) {
            throw new IllegalArgumentException("Invalid filter specified for ES node discovery: " + filter);
        }
        attribute = conditions[0].trim();
        value = conditions[1].trim();
    } else {
        attribute = null;
        value = null;
    }
    final NodesSniffer nodesSniffer = new ElasticsearchNodesSniffer(restClient, sniffRequestTimeoutMillis, scheme);
    return new FilteredElasticsearchNodesSniffer(nodesSniffer, attribute, value);
}