элегантно вернуть карту

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