Capture Arbitrary Conditions with `withCallingHandlers`

后端 未结 4 1389
梦毁少年i
梦毁少年i 2020-12-31 18:05

The Problem

I\'m trying to write a function that will evaluate code and store the results, including any possible conditions signaled in the code. I\'ve got this

4条回答
  •  盖世英雄少女心
    2020-12-31 18:31

    I may be a bit late, but I've been digging into the condition-system as well, and I think I've found some other solutions.

    But first: some reasons why this is necessarily a hard problem, not something that can easily be solved generally.
    The question is which function is signalling a condition, and whether this function can continue execution if it throws a condition. Errors are implemented as "just a condition" as well, but most functions don't expect to be continued after they've thrown a stop().
    And some functions may pass on a condition, expecting not be bothered by it again.
    Normally, this means that control can only be returned after a stop if a function has explicitly said it can accept that: with a restart provided. There may also be other serious conditions that can be signalled, and if a function expects such a condition to always be caught, and you force it to return execution, things break badly.
    What should happen when you would have written it as follows and execution would resume?

    myfun <- function(deleteFiles=NULL) {
      if (!all(haveRights(deleteFiles))) stop("Access denied")
      file.remove(deleteFiles)
    }
    withCallingHandlers(val <- eval(myfun(myProtectedFiles)),
      error=function(e) message("I'm just going to ignore everything..."))
    

    If no other handlers are called (which alert the user that stop has been called), the files would be removed, even though this function has a (small) safeguard against that. In the case of an error this is clear, but there could be also cases for other conditions, so I think that's the main reason R doesn't really support it if you stop the passing on of conditions, unless it means halting.

    Nonetheless, I think I've found 2 ways of hacking your problem. The first is simply executing expr step by step, which is quite close to Martin Morgans solution, but moves the withRestarts into your function:

    evalcapt <- function(expr) {
      conds <- list()
      for (i in seq_along(expr)) {
        withCallingHandlers(
          val <- withRestarts(
            eval(expr[[i]]),
            muffleCustom = function()
              NULL
          ),
          custom = function(e) {
            message("Caught condition of class ", deparse(class(e)))
            conds <<- c(conds, list(e))
            invokeRestart(findRestart("muffleCustom"))
          })
      }
      list(val = val, conditions = conds)
    }
    

    The main disadvantage is that this doesn't dig into functions, expr is executed for each instruction at the level it is called.
    So if you call evalcapt(myfun()), the for-loop sees this as one instruction. And this one instruction throws a condition --> so does not return --> so you can't see any output that would have been there would you not have been catching anything. OTOH, evalcapt(expression(signalCondition(myCondition), 25)) does work as requested, as this is an expression with 2 elements, each of which is called.

    If you want to go hardcore, I think you could try evaluating myfun() step-by-step, but there is always the question how deep you want to go. If myfun() calls myotherfun(), which calls myotherotherfun(), do you want to return control to the point where myfun failed, or myotherfun, or myotherotherfun?
    Basically, it's just a guess about what level you want to halt execution, and where you want to resume.

    So a second solution: hijack any call to signalCondition. This means you'll probably end up at a quite deep level, although not the very deepest (no primitives, or code that calls .signalCondition).
    I think this works best if you're really sure that your custom condition is only thrown by code that is written by you: it means that execution resumes directly after signalCondition.
    Which gives me this function:

    evalcapt <- function(expr) {
      if(exists('conds', parent.frame(), inherits=FALSE)) {
        conds_backup <- get('conds', parent.frame(), inherits=FALSE)
        on.exit(assign('conds', conds_backup, parent.frame(), inherits=FALSE), add=TRUE)
      } else {
        on.exit(rm('conds', pos=parent.frame(), inherits=FALSE), add=TRUE)
      }
      assign('conds', list(), parent.frame(), inherits=FALSE)
      origsignalCondition <- signalCondition
      if(exists('signalCondition', parent.frame(), inherits=FALSE)) {
        signal_backup <- get('signalCondition', parent.frame(), inherits=FALSE)
        on.exit(assign('signalCondition', signal_backup, parent.frame(), inherits=FALSE), add=TRUE)
      } else {
        on.exit(rm('signalCondition', pos=parent.frame(), inherits=FALSE), add=TRUE)
      }
      assign('signalCondition', function(e) {
        if(is(e, 'custom')) {
          message("Caught condition of class ", deparse(class(e)))
          conds <<- c(conds, list(e))
        } else {
          origsignalCondition(e)
        }
      }, parent.frame())
      val <- eval(expr, parent.frame())
      list(val=val, conditions=conds)
    }
    

    It looks way messier, but that's mostly because there are more issues with which environment to use. The differences are that here, I use the calling environment as context, and to hijack signalCondition() that needs to be there too. And afterwards we need to clean up.
    But the main use is overwriting signalCondition: if we see a custom error we log it, and return control. If it's another condition, we pass on control.

    Here there may be some smaller disadvantages:

    • You may end up in a deeper function, where the bug is the way myfun calls myotherfun, but you end up in myotherfun (or deeper).
    • It only catches occurrences where signalCondition is called. If you call e.g. warning(myCondition), nothing is caught.
    • If a function in another package/another environment calls signalCondition, then it uses its own searchpath, meaning our signalCondition might be bypassed, and base::signalCondition is used instead.
    • When debugging, it's a lot uglier. Variables are assigned in environments where you don't expect them (and then disappear when you exit a function), the scope for different functions may be unclear, parent.frame() might give others results then you'd expect, etc.
    • And as said before: all functions must be able to handle re-entrance after throwing a condition.

提交回复
热议问题