R/worker_log.R

In javierluraschi/sparkworker: R Worker for Apache Spark

log_env <- new.env()

log_session <- function(sessionId) {
  assign('sessionId', sessionId, envir = log_env)
}

log_format <- function(message, level = "INFO") {
  paste(
    format(Sys.time(), "%y/%m/%d %H:%M:%S"),
    " ",
    level,
    " sparklyr: RScript (",
    log_env$sessionId,
    ") ",
    message,
    sep = "")
}

log_level <- function(..., level) {
  args = list(...)
  message <- paste(args, sep = "", collapse = "")
  formatted <- log_format(message, level)
  cat(formatted, "\n")
}

log <- function(...) {
  log_level(..., level = "INFO")
}

log_warning<- function(...) {
  log_level(..., level = "WARN")
}

log_error <- function(...) {
  log_level(..., level = "ERROR")
}

unlockBinding("stop",  as.environment("package:base"))
assign("stop", function(...) {
  log_error(...)
  quit(status = -1)
}, as.environment("package:base"))
lockBinding("stop",  as.environment("package:base"))