Print plan() for 'batchtools_slurm' reports on Slurm version

Author: HenrikBengtsson

DESCRIPTION

@@ -1,5 +1,5 @@
 Package: future.batchtools
-Version: 0.12.2-9969
+Version: 0.12.2-9970
 Depends:
     R (>= 3.2.0),
     parallelly,

NAMESPACE

@@ -15,6 +15,7 @@ S3method(nbrOfWorkers,BatchtoolsUniprocessFutureBackend)
 S3method(print,BatchtoolsFuture)
 S3method(print,BatchtoolsFutureBackend)
 S3method(print,BatchtoolsSGEFutureBackend)
+S3method(print,BatchtoolsSlurmFutureBackend)
 S3method(registerFuture,BatchtoolsFuture)
 S3method(registerFuture,BatchtoolsUniprocessFuture)
 S3method(registerFuture,default)

NEWS.md

@@ -33,7 +33,8 @@
    specified, corresponding `module load <name>` entries are injected
    to the job script.
 
- * Print `plan()` for `batchtools_sge` reports on SGE version.
+ * Calling `plan()` on `batchtools_sge` and `batchtools_slurm` reports
+   on the version of the scheduler.
 
 ## Documentation
 

R/batchtools_slurm.R

@@ -7,6 +7,15 @@ BatchtoolsSlurmFutureBackend <- function(...) {
 }
 
 
+#' @export
+#' @keywords internal
+print.BatchtoolsSlurmFutureBackend <- function(x, ...) {  
+  NextMethod()
+  printf("Slurm version: %s\n", slurm_version())
+  invisible(x)
+}
+
+
 #' A batchtools slurm backend resolves futures in parallel via a Slurm job scheduler
 #'
 #' @inheritParams BatchtoolsTemplateFutureBackend
@@ -80,3 +89,29 @@ attr(batchtools_slurm, "tweakable") <- c(
 )
 attr(batchtools_slurm, "init") <- TRUE
 attr(batchtools_slurm, "factory") <- BatchtoolsSlurmFutureBackend
+
+
+slurm_version <- local({
+  version <- NULL
+  
+  function() {
+    if (is.null(version)) {
+      out <- tryCatch(system2("scontrol", args = c("--version"), stdout = TRUE, stderr = TRUE), error = identity)
+      if (inherits(out, "error")) {
+        version <<- "N/A (unexpected output from 'scontrol --version')"
+      } else {
+        status <- attr(out, "status")
+        if (!is.null(status) && status != 0) {
+          version <<- "N/A (unexpected output from 'scontrol --version')"
+        } else {
+          out <- gsub("(^[[:blank:]]+|[[:blank:]]+$)", "", out)
+          out <- out[nzchar(out)]
+          if (length(out) >= 1) {
+            version <<- out[1]
+          }
+        }
+      }
+    }
+    version
+  }
+})