#!/usr/bin/env bash
# Set the desired task count on the experiment service.
#
# Usage: ./set_desired.sh <n>

set -euo pipefail

N="${1:?Usage: $0 <desired_count>}"
CLUSTER="claude-ecs-task-protection-experiment"
SERVICE="claude-ecs-task-protection-experiment"

aws ecs update-service \
    --cluster "${CLUSTER}" \
    --service "${SERVICE}" \
    --desired-count "${N}" \
    --output json > /dev/null

echo "Set desired_count=${N}"