Question

我想为函数

的未指定数量的参数创建一些函数

例如

scala> def test(fx: (String*) => Boolean, arg: String*): Boolean = fx(arg: _*)
test: (fx: String* => Boolean, arg: String*)Boolean

scala> def AA(arg1: String, arg2: String) :Boolean = {
       println ("Arg1 : " + arg1 + " Arg2 : " + arg2)
       true}
AA: (arg1: String, arg2: String)Boolean

scala> test(AA,"ASDF","BBBB")
<console>:10: error: type mismatch;
 found   : (String, String) => Boolean
 required: String* => Boolean
            test(AA,"ASDF","BBBB")
                  ^

我该如何解决这个问题？

Answer 1

这可以使用ProductArgs的无形和类似于我answer的另一个问题来完成。

import shapeless.{HList, ProductArgs}
import shapeless.ops.hlist.IsHCons
import shapeless.ops.function.FnToProduct
import shapeless.syntax.std.function._

object test extends ProductArgs {
  def applyProduct[L <: HList, NarrowArgs <: HList, Args <: HList, F, R](
    l: L
  )(implicit 
    ihc: IsHCons.Aux[L, F, NarrowArgs],
    ftp: FnToProduct.Aux[F, Args => R],
    ev: NarrowArgs <:< Args
  ): R = {
    val (func, args) = (l.head, l.tail)
    func.toProduct(args)
  }
}

您可以将其用作：

def aa(s1: String) = s1.length
def bb(s1: String, s2: String) = s1 * s2.length

test(aa _, "foo")         // Int = 3
test(bb _, "foo", "bar")  // String = foofoofoo
// test(aa _, "foo", "bar") doesn't compile

将ProductArgs转换或test(aa _, "foo")（实际为test.apply(aa _, "foo")）扩展为test.applyProduct((aa _) :: "foo" :: HNil)。在applyProduct中，我们检查HList是否包含函数和有效参数。

我们不应该使用NarrowArgs <:< Args，但ProductArgs似乎与SingletonProductArgs的结果相同。

Answer 2

这是因为AA不接受可变数量的参数，将其更改为：

def AA(args: String*) :Boolean

如何在Scala中为未指定数量的参数函数创建函数？

2 个答案: