#!/usr/bin/env bash

. ${MAIN_PATH}/lib/yandex.sh

usage() {
	echo """
Usage: ya tool sre pqos [opts]

Available opts:

  -h | help  - print help
       sys  - for whole system
       numa  - splitted by numa
"""
}

cmd_sys() {
	range_cpus=$(cat /sys/devices/system/cpu/online)
	cmd_run_sudo_yatool pqos -I -m "all:[$range_cpus]"
}

cmd_numa() {
	args=""
	for i in  $(ls -d /sys/devices/system/node/node*); do
		range_cpus=$(cat $i/cpulist)
		args+=" -m all:[$range_cpus]"
	done
	cmd_run_sudo_yatool pqos -I $args
}

echo $1
case "$1" in
  sys) shift; cmd_sys;;
  numa) shift; cmd_numa;;
  *) usage;;
esac
