helper/tmpdir-setup: Add cleanup command

mariux64 · Apr 20, 2022 · ce29daf · ce29daf
1 parent 08b5694
commit ce29daf
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/helper/tmpdir-setup b/helper/tmpdir-setup
@@ -4,6 +4,7 @@ usage() {
     cat <<EOF >&2
 usage:
     $0 create JOBID SIZE UID  # create SIZE GiB /dev/shm/mxqd/mnt/job/$JOBID
+    $0 cleanup JOBID          # cleanup
 EOF
     exit 1
 }
@@ -47,13 +48,26 @@ cmd_create() {
 	exit $status
 }
 
+cmd_cleanup() {
+    (( $# == 1 )) || usage
+    MXQ_JOBID=$1
+
+    mountpoint=$mntdir/$MXQ_JOBID
+
+    umount $mountpoint
+    rmdir $mountpoint
+}
+
 (( $# > 0 )) || usage
 cmd="$1"
 shift;
 case "$cmd" in
 	create)
 		cmd_create "$@"
 		;;
+    cleanup)
+        cmd_cleanup "$@"
+        ;;
 	*)
 		usage
 		;;