在Rlang中进行嵌套懒惰评估的干净方法

Question

假设我有一个函数f，该函数需要一堆参数以及一个可选的额外参数。

f <- function(..., extra)
{
    arglst <- lapply(quos(...), get_expr)
    if(!missing(extra))
    {
        extra <- get_expr(enquo(extra))
        arglst <- c(arglst, extra=extra)
    }
    arglst
    ## ... do something with argument list ... ##
}

f(a, extra=foo)
# [[1]]
# a
# 
# $extra
# foo

请注意，我不想这样评估参数，但是我确实想要获取传入的表达式，以供其他代码进行评估。

新的rlang软件包（为dplyr的下一个版本提供支持，该版本将在CRAN Real Soon Now上发布）提供了用于懒惰评估的广泛工具，而我在f上面已经使用过。例如quos，get_expr和enquo都是rlang的函数。

在中f，我处理的部分extra实际上是样板代码：我想在其他函数中执行此操作，而不仅仅是在f。我不想每次都重写它，所以我想将它放入自己的函数中：

doExtra <- function(arglst, extra)
{
    if(!missing(extra))
    {
        extra <- get_expr(enquo(extra))
        arglst <- c(arglst, extra=extra)
    }
    arglst
}

f2 <- function(..., extra)
{
    arglst <- lapply(quos(...), get_expr)
    arglst <- doExtra(arglst, extra)
    arglst
}

问题是当我这样做时extra，doExtra看到的值是从中传入的值f2，而不是原始值：

f2(a, extra=foo)
# [[1]]
# a
#
# $extra
# extra

如何修改f以隔离样板代码，而不会得到错误的结果？我可以做一些事情，例如直接操纵doExtra调用框架的环境，但这将非常丑陋。

Answer 1

• 要将命名参数转发给另一个引用函数，您必须先引用然后取消引用：!! enquo(arg)。如果您只是通过enquo(arg)，引号功能将看到：enquo(arg)。如果您传递参数符号，那也将是它。这就是为什么您需要在其捕获的参数内取消引用。

  !! enquo(arg)触发的计算enquo(arg)，该计算返回提供给arg参数的表达式。然后在函数捕获的参数中将其取消引用。

• 如果您引用的是可能丢失的参数，最好先对其进行引用，然后使用来检查是否缺少参数quo_is_missing()。引用缺少的参数会创建通过quo()不带参数调用返回的相同对象。

• 如果您不需要担保，则可以使用exprs()和enexpr()。但是，您正在失去环境，并且使进一步的评估变得脆弱。

  如果您要通过其他方式捕获环境以对其进行评估base::eval()或类似评估，请注意，担保可能包含其他担保。只有eval_tidy()了解这些嵌套的方法。

IIUC您的问题，这是关于传递一个应被引用给另一个函数的参数。一种方法是捕获第一个函数，然后按值传递给第二个函数：

library("purrr")
library("rlang")

f <- function(..., extra) {
  exprs <- exprs(...)

  # Pass the enquoted argument by value
  exprs <- extra_by_value(exprs, enexpr(extra))

  exprs
}
extra_by_value <- function(exprs, extra) {
  if (!is_missing(extra)) {
    c(exprs, extra = extra)
  } else {
    exprs
  }
}

如果第二个函数必须按表达式而不是按值（也许是因为它是另一个面向用户的动词），则必须取消对引用的表达式的引用：

f <- function(..., extra) {
  exprs <- exprs(...)

  # Since the argument is captured by the function, we need
  # to unquote the relevant expression into the argument:
  exprs <- extra_by_expression(exprs, !! enexpr(extra))

  exprs
}
extra_by_expression <- function(exprs, extra) {
  extra <- enexpr(extra)
  if (!is_missing(extra)) {
    c(exprs, extra = extra)
  } else {
    exprs
  }
}

所有这些概念都适用于担保。这是等效的代码：

f <- function(..., extra) {
  quos <- quos(...)

  # Since the argument is captured by the function, we need
  # to unquote the relevant expression into the argument:
  quos <- extra_by_expression(quos, !! enquo(extra))

  quos
}
extra_by_expression <- function(quos, extra) {
  extra <- enquo(extra)
  if (!quo_is_missing(extra)) {
    c(quos, extra = extra)
  } else {
    quos
  }
}

与原始表达式相比，使用quaquos通常总是更好，因为它们可以跟踪其上下文。