Scala不可变列表添加了一个'单元'元件

时间:2012-09-13 08:20:14

标签: list scala functional-programming immutability

我是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. 并且2.请告诉我正确的方法来做这个Scala。如果我在一系列条件上进行迭代,我可以使用理解。

4 个答案:

答案 0 :(得分:4)

我宁愿使用一元或空列表,而不是过滤单位:

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 " )
}

答案 1 :(得分:3)

if - else是Scala中的表达式。你写的内容变成:

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

如您所见,常见的分支类型为Unit

整个表达式的类型为List[Any],因为这是StringUnit的常见超类型。

实现目标的一些方法:

// #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)
} 

答案 2 :(得分:2)

您可以这样写:

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)结束,按类型过滤。单位不满足集合中的类型条件,因此结果将为空字符串列表

好处是,与filter方法不同,collect选择尽可能紧密的类型(因此您将获得字符串序列 - 这将来自于收集中的函数)。

另一种可能性是将.collect { ... }替换为flatten,但您将丢失类型信息。

答案 3 :(得分:1)

另一种观点是构建一个选项列表,然后展平列表。

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