From 7c8ab2fe66d508cf6e8d7a728f599a5f71abca96 Mon Sep 17 00:00:00 2001
From: CVL-GitHub <jupiter.hu@monash.edu>
Date: Mon, 9 Mar 2015 13:39:41 +1100
Subject: [PATCH] add slurm provision
---
.../templates/slurm_provision.sh.j2.v1 | 34 +++++++++++++++++++
1 file changed, 34 insertions(+)
create mode 100644 roles/slurm-provision/templates/slurm_provision.sh.j2.v1
diff --git a/roles/slurm-provision/templates/slurm_provision.sh.j2.v1 b/roles/slurm-provision/templates/slurm_provision.sh.j2.v1
new file mode 100644
index 00000000..2b8bf1f1
--- /dev/null
+++ b/roles/slurm-provision/templates/slurm_provision.sh.j2.v1
@@ -0,0 +1,34 @@
+#!/bin/sh
+
+CVL_HOME="/cvl/home"
+user_list=($(getent passwd | cut -d ":" -f1))
+log_file="/root/slurm.log"
+
+for user in ${user_list[*]}; do
+ uid=$(id -u ${user})
+ gid=$(id -g ${user})
+ user_home=${CVL_HOME}/${user}
+ if [ ! -d ${user_home} ]; then
+ mkdir -p ${user_home}
+ cp -r /etc/skel/* ${user_home}
+ chown -R ${uid}:${gid} ${user_home}
+ chmod 700 ${user_home}
+
+ account={{ projectname }}
+ cluster={{ clustername }}
+ find=$(sacctmgr list cluster ${cluster} | grep ${cluster})
+ if [ -z "${find}" ]; then
+ su slurm -c "sacctmgr -i add cluster ${cluster}" || { echo "error to create cluster ${cluster}" >> ${log_file} && exit 1; }
+ fi
+ find=$(sacctmgr list account ${account} | grep ${account})
+ if [ -z "${find}" ]; then
+ su slurm -c "sacctmgr -i add account ${account} Description=CVL Organization=monash cluster=${cluster}" || { echo "error to create account ${account}" >> ${log_file} && exit 1; }
+ fi
+ find=$(sacctmgr list user ${username} | grep ${username})
+ if [ -z "${find}" ]; then
+ su slurm -c "sacctmgr -i add user ${username} account=${account} cluster=${cluster}" || { echo "error to create user ${username}" >> ${log_file} && exit 1; }
+ fi
+ fi
+done
+
+
--
GitLab