Optionally adding items to a Scala Map
Asked Answered
M

3

35

I am looking for an idiomatic solution to this problem.

I am building a val Scala (immutable) Map and would like to optionally add one or more items:

val aMap =
  Map(key1 -> value1,
      key2 -> value2,
      (if (condition) (key3 -> value3) else ???))

How can this be done without using a var? What should replace the ???? Is it better to use the + operator?

val aMap =
  Map(key1 -> value1,
      key2 -> value2) +
  (if (condition) (key3 -> value3) else ???))

One possible solution is:

val aMap =
  Map(key1 -> value1,
      key2 -> value2,
      (if (condition) (key3 -> value3) else (null, null))).filter {
        case (k, v) => k != null && v != null
      }

Is this the best way?

Meteoritics answered 18/5, 2011 at 17:46 Comment(0)
T
35

How about something along the lines of

val optional = if(condition) Seq((key3 -> value3)) else Nil
val entities = Seq(key1 -> value1, key2 -> value2) ++ optional
val aMap = entities.toMap
Trembly answered 18/5, 2011 at 17:54 Comment(1)
Nice! I can probably also do Map(...) ++ (if(condition) Seq((key3 -> value3)) else Nil)Meteoritics
T
27

Another possibility is to take advantage of the iterable nature of Option.

A non-empty value o:

scala> val o = Some('z' -> 3)
scala> (Seq('x' -> 1, 'y' -> 2) ++ o).toMap
res1: scala.collection.immutable.Map[Char,Int] = Map(x -> 1, y -> 2, z -> 3)

An empty value o:

scala> val o = None
scala> (Seq('x' -> 1, 'y' -> 2) ++ o).toMap
res2: scala.collection.immutable.Map[Char,Int] = Map(x -> 1, y -> 2)
Twombly answered 18/10, 2015 at 20:27 Comment(0)
R
15

You can add directly to the map:

scala> val map = Map(1 -> 2, 3 -> 4)
scala> val some = Some(5 -> 6)
scala> val none = None
scala> val combinedMap = map ++ some ++ none
combinedMap: scala.collection.immutable.Map[Int,Int] = Map(1 -> 2, 3 -> 4, 5 -> 6)
Rolfrolfe answered 3/6, 2017 at 16:27 Comment(1)
Nice. This is essentially using options iterable nature. Here ++ is an alias concat and you can concat any IterableOnce of Tuple2 to Map where types in Tuple2 match.Omnipotence

© 2022 - 2024 — McMap. All rights reserved.