From 050fbf10d1b7888c59d39bd3f9fe7301f9f7920c Mon Sep 17 00:00:00 2001
From: Dom G <domgarguilo@apache.org>
Date: Tue, 15 Sep 2026 10:26:19 -0400
Subject: [PATCH] Allow separate datanode hosts for agitator

---
 bin/agitator | 3 ++-
 conf/env.sh  | 2 ++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/bin/agitator b/bin/agitator
index a4fbe973..6d21347e 100755
--- a/bin/agitator
+++ b/bin/agitator
@@ -203,7 +203,8 @@ function read_cluster_conf() {
 function get_app_hosts() {
   case "$1" in
     manager) echo -n "$MANAGER_HOSTS" ;;
-    tserver | datanode) echo -n "$TSERVER_HOSTS" ;;
+    tserver) echo -n "$TSERVER_HOSTS" ;;
+    datanode) echo -n "${AGTR_DATANODE_HOSTS:-$TSERVER_HOSTS}" ;;
     *) return 1 ;;
   esac
 }
diff --git a/conf/env.sh b/conf/env.sh
index 5907d7f6..e50ed3a1 100644
--- a/conf/env.sh
+++ b/conf/env.sh
@@ -106,6 +106,8 @@ export AGTR_DATANODE_RESTART_SLEEP_TIME=10
 # Min and max number of datanodes the agitator will kill at once
 export AGTR_DATANODE_MIN_KILL=1
 export AGTR_DATANODE_MAX_KILL=1
+# Datanode hosts if different from tservers
+#export AGTR_DATANODE_HOSTS="host1 host2"
 # HDFS agitation
 AGTR_HDFS_USER=$(whoami)
 export AGTR_HDFS_USER