在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中做类似的事情(编写一个可以应用于任何函数的包装器,而不管其输入/输出类型如何)?
解决方法:
您当然可以使用宏来执行此操作.您可以使用部分应用程序将方法调用转换为函数:
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)
宏相当复杂,但是如果您认为这是您想走的路,我可以详细说明.