Scala 中的 Python 风格装饰器

Question

在 Python 中我可以做这样的事情:

def wrap(f):
    def wrapper(*args, **kwargs):
        print "args: ", args, kwargs
        res = f(*args, **kwargs)
        print "result: ", res
        return res
    return wrapper

这让我可以包装任何函数，而不管它们采用什么参数。例如:

In [8]: def f(thing):
    print "in f:", thing
    return 3

In [9]: wrapped_f = wrap(f)

In [10]: wrapped_f(2)
args:  (2,) {}
in f: 2
result:  3
Out[10]: 3

有没有办法在 Scala 中做类似的事情(编写一个可以应用于任何函数的包装器，无论其输入/输出类型如何)？

Answer 1

您当然可以使用宏来做到这一点。您可以将方法调用转换为具有部分应用程序的函数:

object Foo {
  def bar(i: Int): Int = i + 1
}

val fn = Foo.bar _

defined object Foo
fn: Int => Int = <function1>

现在您有一个对象，在本例中为 Function1[Int, Int] 类型，您可以将其传递给 Scala 宏，它可能是这样的(未测试):

object DecoratorMacros {

  import reflect.macros.blackbox
  import language.experimental.macros

  def decorate[A <: Function](fn: A): [A] = macro decorate_impl[A]

  def decorate_impl[A: c.WeakTypeTag](c: blackbox.Context) = {
    import c.universe._
    val type = weakTypeOf[A]
    ...
  }
}

在宏的主体中，您可以检查 fn: A 的整个类型签名，其中将包括参数。然后您可以编写代码来实现您想要的副作用，并返回一个您可以调用的函数。像这样:

DecoratorMacros.decorate(Foo.bar _)(42)

宏相当复杂，但如果您认为这是一条您想走的路，我可以详细说明。