1 files changed, 55 insertions, 0 deletions
diff --git a/src/main/java/org/elasticsearch/river/cluster/RiverNodeHelper.java b/src/main/java/org/elasticsearch/river/cluster/RiverNodeHelper.java
new file mode 100644
index 0000000..6300751
--- /dev/null
+++ b/src/main/java/org/elasticsearch/river/cluster/RiverNodeHelper.java
@@ -0,0 +1,55 @@
+/*
+ * Licensed to Elasticsearch under one or more contributor
+ * license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright
+ * ownership. Elasticsearch licenses this file to you under
+ * the Apache License, Version 2.0 (the "License"); you may
+ * not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package org.elasticsearch.river.cluster;
+
+import org.elasticsearch.cluster.node.DiscoveryNode;
+import org.elasticsearch.river.RiverName;
+
+/**
+ *
+ */
+public class RiverNodeHelper {
+
+    public static boolean isRiverNode(DiscoveryNode node) {
+        // we don't allocate rivers on client nodes
+        if (node.clientNode()) {
+            return false;
+        }
+        String river = node.attributes().get("river");
+        // by default, if not set, it's a river node (better OOB exp)
+        if (river == null) {
+            return true;
+        }
+        if ("_none_".equals(river)) {
+            return false;
+        }
+        // there is at least one river settings, we need it
+        return true;
+    }
+
+    public static boolean isRiverNode(DiscoveryNode node, RiverName riverName) {
+        if (!isRiverNode(node)) {
+            return false;
+        }
+        String river = node.attributes().get("river");
+        // by default, if not set, its an river node (better OOB exp)
+        return river == null || river.contains(riverName.type()) || river.contains(riverName.name());
+    }
+}