mirror of
https://github.com/ClusterCockpit/cc-metric-collector.git
synced 2024-11-10 04:27:25 +01:00
140 lines
4.3 KiB
Bash
Executable File
140 lines
4.3 KiB
Bash
Executable File
#!/bin/bash -l
|
|
|
|
|
|
# Some settings for scripting with less headache
|
|
|
|
# when a command fails, bash exits instead of continuing
|
|
set -o errexit
|
|
|
|
# make the script fail, when accessing an unset variable
|
|
# use "${VARNAME-}" instead of "$VARNAME" when you want to access
|
|
# a variable that may or may not have been set
|
|
set -o nounset
|
|
|
|
# ensure that a pipeline command is treated as failed, even if one command in the pipeline fails
|
|
set -o pipefail
|
|
|
|
# enable debug mode, by running your script as TRACE=1 ./script.sh
|
|
if [[ "${TRACE-0}" == "1" ]]; then
|
|
set -o xtrace
|
|
fi
|
|
|
|
|
|
# Default values for variables
|
|
: ${UID=$(id -u)}
|
|
: ${VERBOSITY=0}
|
|
: ${DELETE=0}
|
|
: ${LIST=0}
|
|
: ${JOBID="random"}
|
|
: ${BASE=./slurmJobDetector-sys-fs-cgroup}
|
|
|
|
# Print usage if needed
|
|
usage()
|
|
{
|
|
echo "
|
|
Usage: $(basename $0) <opts>
|
|
[ -h | --help ]
|
|
[ -v | --verbosity ]
|
|
[ -u | --uid <UID> (default: ${UID}) ]
|
|
[ -j | --jobid <JOBID> (default: ${JOBID}) ]
|
|
[ -b | --basedir <JOBID> (default: ${BASE}) ]
|
|
[ -d | --delete ]
|
|
[ -l | --list ]
|
|
"
|
|
exit $1;
|
|
}
|
|
|
|
cd "$(dirname "$0")"
|
|
|
|
main() {
|
|
PARSED_ARGUMENTS=$(getopt -a -n $(basename $0) -o hj:u:vb:dl --long help,verbosity,uid:,jobid:,basedir:,delete,list -- "$@")
|
|
VALID_ARGUMENTS=$?
|
|
# Parsing failed
|
|
if [[ "$VALID_ARGUMENTS" != "0" ]]; then
|
|
usage 2
|
|
fi
|
|
# No argument (comment out if command should work without any arguments)
|
|
# if [[ "${PARSED_ARGUMENTS}" == " --" ]]; then
|
|
# usage 0
|
|
# fi
|
|
# Evaluate arguments
|
|
eval set -- "$PARSED_ARGUMENTS"
|
|
while :
|
|
do
|
|
case "$1" in
|
|
-h | --help) usage 0; shift ;;
|
|
-v | --verbosity) VERBOSITY=1; shift ;;
|
|
-d | --delete) DELETE=1; shift ;;
|
|
-l | --list) LIST=1; shift ;;
|
|
-u | --uid) UID=$2 ; shift 2 ;;
|
|
-j | --jobid) JOBID=$2 ; shift 2 ;;
|
|
-b | --basedir) BASE=$2 ; shift 2 ;;
|
|
--) shift; break ;;
|
|
*) echo "Unexpected option: $1 - this should not happen."
|
|
usage 2;;
|
|
esac
|
|
done
|
|
|
|
if [[ ${LIST} -eq 1 ]]; then
|
|
for F in $(ls -d ${BASE}/cpuset/slurm/uid_*/job_*); do
|
|
JOBID=$(echo "$F" | rev | cut -d '/' -f 1 | rev | cut -d '_' -f 2)
|
|
MYUID=$(echo "$F" | rev | cut -d '/' -f 2 | rev | cut -d '_' -f 2)
|
|
echo "UID ${MYUID} JOBID ${JOBID}"
|
|
done
|
|
exit 0
|
|
fi
|
|
|
|
|
|
if [[ ${JOBID} == "random" ]]; then
|
|
if [[ ${DELETE} -eq 1 ]]; then
|
|
echo "Cannot use random JOBID for deletion"
|
|
exit 1
|
|
else
|
|
JOBID=$RANDOM
|
|
fi
|
|
fi
|
|
|
|
FOLDERS="cpuset cpuacct memory devices"
|
|
|
|
if [[ ${DELETE} -eq 1 ]]; then
|
|
for F in ${FOLDERS}; do
|
|
rm -r --force "${BASE}/${F}/slurm/uid_${UID}/job_${JOBID}"
|
|
done
|
|
else
|
|
for F in ${FOLDERS}; do
|
|
if [[ $VERBOSITY -eq 1 ]]; then
|
|
echo "${BASE}/${F}/slurm/uid_${UID}/job_${JOBID}"
|
|
fi
|
|
mkdir -p "${BASE}/${F}/slurm/uid_${UID}/job_${JOBID}"
|
|
done
|
|
|
|
echo "0-71" > "${BASE}/cpuset/slurm/uid_${UID}/job_${JOBID}/cpuset.effective_cpus"
|
|
echo "0-3" > "${BASE}/cpuset/slurm/uid_${UID}/job_${JOBID}/cpuset.effective_mems"
|
|
|
|
echo "249036800000" > "${BASE}/memory/slurm/uid_${UID}/job_${JOBID}/memory.limit_in_bytes"
|
|
echo "249036800000" > "${BASE}/memory/slurm/uid_${UID}/job_${JOBID}/memory.soft_limit_in_bytes"
|
|
echo "13987840" > "${BASE}/memory/slurm/uid_${UID}/job_${JOBID}/memory.usage_in_bytes"
|
|
echo "14966784" > "${BASE}/memory/slurm/uid_${UID}/job_${JOBID}/memory.max_usage_in_bytes"
|
|
echo "60" > "${BASE}/memory/slurm/uid_${UID}/job_${JOBID}/memory.swappiness"
|
|
|
|
echo "474140369" > "${BASE}/cpuacct/slurm/uid_${UID}/job_${JOBID}/cpuacct.usage"
|
|
echo "169078878" > "${BASE}/cpuacct/slurm/uid_${UID}/job_${JOBID}/cpuacct.usage_user"
|
|
echo "315684619" > "${BASE}/cpuacct/slurm/uid_${UID}/job_${JOBID}/cpuacct.usage_sys"
|
|
|
|
echo "a *:* rwm" > "${BASE}/devices/slurm/uid_${UID}/job_${JOBID}/devices.list"
|
|
#memory.numa_stat
|
|
#total=0 N0=0 N1=0 N2=0 N3=0
|
|
#file=0 N0=0 N1=0 N2=0 N3=0
|
|
#anon=0 N0=0 N1=0 N2=0 N3=0
|
|
#unevictable=0 N0=0 N1=0 N2=0 N3=0
|
|
#hierarchical_total=958 N0=28 N1=579 N2=180 N3=171
|
|
#hierarchical_file=194 N0=0 N1=194 N2=0 N3=0
|
|
#hierarchical_anon=764 N0=28 N1=385 N2=180 N3=171
|
|
#hierarchical_unevictable=0 N0=0 N1=0 N2=0 N3=0
|
|
|
|
fi
|
|
}
|
|
|
|
main "$@"
|
|
|