优雅地返回地图

Question

我是Scala的新手，从Java过来的，我无法从此函数优雅地返回Map。什么是重写此函数的优雅方法，所以只有一个退出点？

foo(): Map[TypeDef, DataFrame]
if (someTest) {
    if (someTest2) {
         val a = A()
         val b = B()
         Map(a -> b)
    } else {
       Map.empty
    }
} else {
    Map.empty
}

Answer 1

您可以这样写：

def foo(): Map[TypeDef, DataFrame] = 
    if (someTest && someTest2) Map(A() -> B()) else Map.empty