R/main.R

In vimc/orderly.server: Serve Orderly

main <- function(args = commandArgs(TRUE)) {
  args <- main_args(args)
  do.call("server", args)
}

main_args <- function(args) {
  doc <- "Usage:
  orderly.server [options] <path>

Options:
  --port=PORT                    Port to run on [default: 8321]
  --host=HOST                    IP address owned by this server [default: 0.0.0.0]
  --go-signal=PATH               Relative path for go signal
  --identity=NAME                Identity to use
  --queue-id=ID                  rrq ID
  --workers=WORKERS              Number of workers to spawn [default: 0]
  --backup-period=BACKUP_PERIOD  How frequently should backup be run, 0 or negative for no backup [default: 600]
  --log-level=LEVEL              Detail level of logs info or trace [default: info]"
  res <- docopt_parse(doc, args)

  backup_period <- as.integer(res[["backup_period"]])
  if (backup_period <= 0) {
    backup_period <- NULL
  }

  list(path = res[["path"]],
       port = as.integer(res[["port"]]),
       host = res[["host"]],
       identity = res[["identity"]],
       go_signal = res[["go_signal"]],
       queue_id = res[["queue_id"]],
       workers = as.integer(res[["workers"]]),
       backup_period = backup_period,
       log_level = res[["log_level"]])
}

main_worker_args <- function(args = commandArgs(TRUE)) {
  usage <- "Usage:
orderly_worker [options] [<queue_id>]

Options:
  --go-signal=PATH  Path for go signal"
  dat <- docopt_parse(usage, args)
  list(queue_id = dat[["queue_id"]],
       go_signal = dat[["go_signal"]])
}

main_worker <- function(args = commandArgs(TRUE)) {
  # nocov start
  args <- main_worker_args(args)
  wait_for_go_signal(NULL, args$go_signal)
  w <- rrq::rrq_worker_from_config(orderly_queue_id(args$queue_id, TRUE),
                                   timeout = 30)
  w$loop()
  invisible(TRUE)
  # nocov end
}

docopt_parse <- function(doc, args) {
  dat <- docopt::docopt(doc, args)
  names(dat) <- gsub("-", "_", names(dat), fixed = TRUE)
  dat
}

write_script <- function(path, code, versioned = FALSE) {
  if (!isTRUE(file.info(path, extra_cols = FALSE)$isdir)) {
    stop("'path' must be a directory")
  }
  if (versioned) {
    rscript <- file.path(R.home(), "bin", "Rscript")
  } else {
    rscript <- "/usr/bin/env Rscript"
  }
  code <- c(sprintf("#!%s", rscript), code)
  path_bin <- file.path(path, "orderly.server")
  writeLines(code, path_bin)
  Sys.chmod(path_bin, "755")
  invisible(path_bin)
}