如何用不可變集合和適當的語句寫適當的Scala？

Question

我想下面的函數：採取以下參數：

• def filters: Map[String, String]
• （從的Request.QueryString）
• def key: String

..如果queryString包含key在參數中指定def queryString: Map[String, Seq[String]] ，我想將相應的值添加到filters。

的工作職能是：

private def getUpdatedFiltersIfQueryStringContains(filters: Map[String, String], queryString: Map[String, Seq[String]], key: String): Map[String, String] = { 
    var updatedFilters: Map[String, String] = filters 

    if (queryString.contains(key)) { 
    updatedFilters = updatedFilters ++ Map(key -> queryString.get(key).get.head) 
    } 

    updatedFilters 
} 

這看起來可怕的，對不對？

由於if語句在Scala中返回的東西，我期待的東西沿着這些線路更多：

private def getUpdatedFiltersIfQueryStringContains(filters: Map[String, String], queryString: Map[String, Seq[String]], key: String): Map[String, String] = { 
    if (queryString.contains(key)) { 
    filters ++ Map(key -> queryString.get(key).get.head) 
    } 
} 

但是，這並不編譯，錯誤的是：

類型不匹配;發現：單位要求：地圖[字符串，字符串]

Answer 1

你的函數簽名指出Map[String, String]應返回，但...等一下，想象，如果的queryString不包含的關鍵是什麼？好吧，看起來什麼都不會返回，並且單元在斯卡拉中什麼也不是（至少沒有任何一種）。

要解決，你需要提供這個問題的選擇：

private def getUpdatedFiltersIfQueryStringContains(filters: Map[String, String], queryString: Map[String, Seq[String]], key: String): Map[String, String] = { 
    if (queryString.contains(key)) { 
     filters ++ Map(key -> queryString.get(key).get.head) 
    } else filters 
    } 

的方式，函數名真讓我害怕

Answer 2

您可以使用更多的單子

queryString.get(key) map { v => filters ++ Map(key -> v.head) } getOrElse filters

這是你想要的。

Answer 3

這裏是一個更地道的實現不使用if：

def getUpdatedFiltersIfQueryStringContains(filters: Map[String, String], queryString: Map[String, Seq[String]], key: String): Map[String, String] = 
    filters ++ queryString.get(key).map(key -> _.head)