From a04f6c8212c055bd1978c7fa6dc6106bad56119f Mon Sep 17 00:00:00 2001
From: see24 <sarahendicott4@gmail.com>
Date: Tue, 2 Jul 2024 16:34:43 -0400
Subject: [PATCH] working on autoscaling

---
 analyses/scripts/run_azure_autoscale.sh    | 49 ++++++++++++++++++++++
 cloud_config/template_task_hello_mult.json |  2 +-
 2 files changed, 50 insertions(+), 1 deletion(-)
 create mode 100644 analyses/scripts/run_azure_autoscale.sh

diff --git a/analyses/scripts/run_azure_autoscale.sh b/analyses/scripts/run_azure_autoscale.sh
new file mode 100644
index 0000000..fcdb958
--- /dev/null
+++ b/analyses/scripts/run_azure_autoscale.sh
@@ -0,0 +1,49 @@
+#!/bin/bash
+
+az login
+
+az batch account login -g EcDc-WLS-rg -n ecdcwlsbatch
+
+poolName=test_pool_json_cli
+
+mkdir cloud_config/task_jsons
+
+for rowi in {1..5}
+do
+ sed 's,<taskid>,'$rowi',g' cloud_config/template_task_hello_mult.json > \
+ cloud_config/task_jsons/task_to_use_$rowi.json
+done
+
+# We create the pool which will start with 1 target dedicated node
+az batch pool create --json-file cloud_config/template_pool_cli.json
+az batch job create --pool-id test_pool_json_cli --id "test_job"
+
+
+az batch pool list \
+--query "[].{name:id, vmSize:vmSize,  curNodes: currentDedicatedNodes, \
+tarNodes: targetDedicatedNodes, allocState:allocationState}" --output table
+
+for rowi in {1..5}
+do
+az batch task create --json-file cloud_config/task_jsons/task_to_use_$rowi.json --job-id test_job
+done
+
+az batch job task-counts show --job-id test_job
+
+az batch pool autoscale enable --pool-id test_pool_json_cli --auto-scale-evaluation-interval "PT5M"\
+ --auto-scale-formula 'percentage = 70;
+ maxNumberofVMs = 25;
+ span = TimeInterval_Minute * 2;
+ $samples = $ActiveTasks.GetSamplePercent(span);
+ $tasks = $samples < percentage ? max(0,$ActiveTasks.GetSample(1)) : max( $ActiveTasks.GetSample(1), avg($ActiveTasks.GetSample(span)));
+ multiplier = 1;
+ $cores = $TargetDedicatedNodes;
+ $extraVMs = (($tasks - $cores) + 0) * multiplier;
+ $targetVMs = ($CurrentDedicatedNodes + $extraVMs);
+ $TargetDedicatedNodes = max(0, min($targetVMs, maxNumberofVMs));
+ $NodeDeallocationOption = taskcompletion;'
+
+# modified a little but still not really working. Need to figure out 
+# a better way to set starting number of tasks
+
+az batch node list --pool-id $poolName --query "{nodes: [].[id, state][]}" --output json
diff --git a/cloud_config/template_task_hello_mult.json b/cloud_config/template_task_hello_mult.json
index c59cd52..259e99e 100644
--- a/cloud_config/template_task_hello_mult.json
+++ b/cloud_config/template_task_hello_mult.json
@@ -1,6 +1,6 @@
 {
     "id": "sampleTask<taskid>",
-    "commandLine": "R -e 'print(<taskid>);mean(mtcars$mpg);Sys.sleep(200)'",
+    "commandLine": "R -e 'print(<taskid>);mean(mtcars$mpg);Sys.sleep(500)'",
 	"containerSettings": {
     "imageName": "rocker/r-bspm:jammy",
     "containerRunOptions": "--rm"