R/app_entry.R

In Wang-Cankun/rDeepMAPS: IRIS3-api

#' Start IRIS3 R api server
#' @import plumber
#' @param port default: 8000
#' @param host default: "127.0.0.1"
#' @return NULL
#' @examples
#' \dontrun{
#' start_server(port = 8000)
#' }
#' @export
start_server <-
  function(port = 8000,
           host = "127.0.0.1") {
    message(paste("Starting IRIS3 R server at host =", host, ", port =", port))
    tryCatch(
      {
        plumber::plumb(dir = as.character(system.file("endpoints",
          package = "iris3api"
        ))) %>%
          plumber::pr_hook("preroute", function(req, res) {
            tictoc::tic()
            logger::log_info(
              glue(
                '{convert_empty(req$REMOTE_ADDR)} \\
              "{convert_empty(req$HTTP_USER_AGENT)}"\\
               {convert_empty(req$REQUEST_METHOD)}\\
               {convert_empty(req$PATH_INFO)}\\
               {convert_empty(res$status)}'
              )
            )
          }) %>%
          plumber::pr_hook("postroute", function(req, res) {
            # nolint start
            end <- tictoc::toc(quiet = TRUE)
            # nolint end
            logger::log_info(
              glue(
                "
               {convert_empty(req$REQUEST_METHOD)} \\
               {convert_empty(req$PATH_INFO)} \\
               {convert_empty(res$status)} \\
               Time elapsed: \\
               {round(end$toc-end$tic,digits=5)}s"
              )
            )
          }) %>%
          plumber::pr_set_docs(FALSE) %>%
          plumber::pr_run(
            host = host,
            port = as.numeric(port)
          )
      },
      interrupt = function(i) {
        disconnect_socketio()
        message("Server Exited")
      }
    )
    return(NULL)
  }


#' Start debug server
#' @import plumber
#' @param port default: 8000
#' @param host default: "127.0.0.1"
#' @return NULL
#' @examples
#' \dontrun{
#' start_debug_server(port = 8000)
#' }
#' @export
start_debug_server <-
  function(port = 8000,
           host = "127.0.0.1") {
    print(getwd())
    #init_socketio()
    message(paste("Starting debug R server at host =", host, ", port =", port))
    tryCatch(
      {
        plumber::plumb(dir = as.character(system.file("endpoints",
          package = "iris3api"
        ))) %>%
          plumber::pr_hook("preroute", function(req, res) {
            tictoc::tic()
          }) %>%
          plumber::pr_hook("postroute", function(req, value) {
            end <- tictoc::toc(quiet = TRUE)
            utils::str(
              list(
                type = as.character(
                  glue("postroute: \\
                {req$REQUEST_METHOD} \\
                     {req$PATH_INFO}")
                ),
                query_str = req$QUERY_STRING,
                post_str = req$body,
                value = value,
                elapsed = as.character(round(end$toc - end$tic, digits = 7))
              )
            )
            return(value)
          }) %>%
          plumber::pr_set_docs(FALSE) %>%
          plumber::pr_run(
            host = host,
            port = as.numeric(port)
          )
      },
      interrupt = function(i) {
        message("Server Exited")
      }
    )
    return(NULL)
  }