如何在Scala表达式中保持功能风格

Question

当我需要将对象的多个参数收集到List中时,我一直在努力寻找一种方法来保持表达式的功能样式.

例如,假设我有一个Notification对象,它既有fromId(通知来自的用户ID),也有objectOwnerId(创建原始对象的用户的id).这些在facebook风格通知中可能有所不同("X也在Y的帖子上发表评论").

我可以使用for表达式收集userIds

val userIds = for { notification <- notifications } yield notification.fromId

但是说我想将fromIds和objectOwnerIds收集到一个列表中,有没有办法在没有vars用户的单个表达式中执行此操作？

我过去做过类似的事情:

var ids = List()
for {
    notification <- notifications
    ids = ids ++ List(notification.fromId, notification.objectOwnerId)
}
ids = ids.distinct

但感觉必须有更好的方法.使用var,以及在完成收集后调用distinct的需要都很难看.我可以避免与一些条件有所不同,但我正在努力学习正确的功能方法来做事.

在此先感谢您的帮助!

Answer 1

对于这种情况,有foldLeft:

(notifications foldLeft Set.empty[Id]) { (set, notification) =>
  set ++ Seq(notification.fromId, notification.ownerId)
}

或者简短形式:

(Set.empty[Id] /: notifications) { (set, notification) =>
  set ++ Seq(notification.fromId, notification.ownerId)
}

集合不会重复.折叠后,您可以根据需要将该集转换为另一个集合.

Answer 2

val userIds = for { 
  notification <- notifications 
  id <- List(notification.fromId, notification.objectOwnerId)
} yield id

应用distinct如果需要的话之后.如果只能在单个通知上复制id,则可以distinct在第二个生成器上应用.