Scala不可變列表添加“單元”元素

Question

我是scala的初學者，我正在尋找最好/慣用的方式去做我打算在這里做的事情。

這就是我想要做的

def someMethod(obj:MyObj):List[String] = {
     List[String]() +: 
      {if (somecondition is satisfied) .. " element"} +: 
      { if (another condition) .. " something else " }


}

也就是說，該方法檢查輸入參數對象的某些屬性，並將元素添加到List（即要返回）。 如果沒有滿足任何條件，則應返回空列表。

1. 當然代碼不能編譯。 但不知何故，我覺得List [T] + Unit應該返回List [T]。 為什么我錯了？

並且2.請告訴我正確的方法來做Scala。 如果我在一系列條件上進行迭代，我可以使用理解。

Answer 1

我寧願使用一元或空列表，而不是過濾單位：

def someMethod(obj:MyObj): List[String] = {
    Nil ++
    ( if (somecondition is satisfied) List(" element") else Nil ) ++
    ( if (another condition) .. List(" something else ") else Nil ) ++
}

編輯：關於下面的評論，如果你發現上面的代碼太冗長和難以維護，你仍然可以創建一個輔助函數：

def optElem[T]( condition: Boolean)( value: => T ): Option[T] = if ( condition ) Option( value ) else None
def someMethod(obj:MyObj): List[String] = {
  Nil ++
  optElem (somecondition is satisfied)( " element") ++
  optElem (another condition)(" something else " )
}

Answer 2

if - else是Scala中的表達式。 你寫的內容變成了：

List[String]() +: 
      {if (somecondition is satisfied) {" element"; () } else () }+: 
      { if (another condition) { " something else "; () } else () }

如您所見，常見的分支類型是Unit 。

整個表達式的類型是List[Any]因為這是String和Unit的常見超類型。

一些方法來實現你想要的：

// #1. Ugly.
def someMethod(obj:MyObj):List[String] = {
  val xs = List[String]() 
  val xs1 = if (somecondition is satisfied) xs :+ " element" else xs
  val xs2 = if (another condition) xs1 :+ " something else" else xs1
  xs2
}

// #2. Better, but uses mutable builder.
def someMethod(obj:MyObj):List[String] = {
  val b = List.newBuilder[String]
  if (somecondition is satisfied) b += " element"
  if (another condition) b += " something else"
  b.result
} 

// #3. Best way IMO, but computationally expensive.
def someMethod(obj:MyObj):List[String] = {
  List[String]() ++
  Option("element").filter(some condition) ++ // for more correct semantics
                                              // use LazyOption from Scalaz.
  Option("something else").filter(another condition)
} 

Answer 3

你可以這樣寫：

def someMethod(obj:MyObj):List[String] = {
     List(
      if (somecondition is satisfied) " element", 
      if (another condition) " something else "
     ) collect{ case x: String => x }
}

假設所有條件都失敗了：您將首先按List(Unit,Unit)結束List(Unit,Unit)然后按類型過濾 。 單位不滿足收集內的類型條件，因此結果將為空列表

好處是，與filter方法不同， collect選擇盡可能緊密的類型（因此你將獲得字符串序列 - 這將來自收集中的函數）。

另一種可能性是用flatten替換.collect { ... } ，但你會丟失類型信息。

Answer 4

另一種觀點是構建一個選項列表，然后展平列表。

def f(obj: MyObj): List[String] = {
  List(if (cond1(obj)) Some("element")        else None,
       if (cond2(obj)) Some("something else") else None).flatten
}