Transformation eine Liste auf eine Karte

Question

ich die folgenden Typen haben:

case class Category(id: Int, name: String) 
case class Product(id: Int, location: Location, categoryIdList: Option[List[Int]]) 

eine Liste der Produkte gegeben

val products:List[Product] = loadProducts() 

Wie kann ich Produkt eine Karte von categoryId zu Ort?

Map[categoryId, Location] 

So würde meine Methode wie folgt aussehen:

def getMap(products: List[Product]): Map[Int, Location] = { 
    ?? 
} 

Ich muss irgendwie über die optionale Liste von categoryIdList laufen und dann eine Karte aus, dass mit der Location-Eigenschaft erstellen.

Answer 1

Um Seq zu einem Map wir es ein Seq[(Int,Location)], das ist ein Seq ein Tuple2 zu sein, zuerst konvertieren müssen zu konvertieren. Nur dann ist die toMap Methode tatsächlich verfügbar.

Bearbeiten: Okay, hier ist eine Implementierung basierend auf jeder categoryId auf der Liste, beachten Sie, dass Sie eine Option einer Liste nicht verwenden sollten, da ein leerer Zustand für eine Liste nur eine leere Liste ist.

def getMap(products: List[Product]): Map[Int, Location] = { 
    products.flatMap(toListTuple2).toMap 
} 

def toListTuple2(product: Product): List[(Int, Location)] = { 
    product.categoryIdList 
    .getOrElse(List()) 
    .map(category => (category, product.location)) 
} 

So, hier wir zuerst unser Produkt in eine Liste von categoryId s drehen und Locations und dann flatmap es zu einem List von (Int, Location), die dann in eine Map gedreht werden kann durch toMap aufrufen.

Answer 2

Dies sollte das tun, was Sie fordern jedoch Lösung keine Probleme in den Kommentaren versehen Adresse:

def getMap(products: List[Product]): Map[Int, Location] = { 
    val locations = scala.collection.mutable.Map[Int, Location]() 
    for { 
    product <- products 
    if product.categoryIdList.nonEmpty 
    category <- product.categoryIdList.get 
    } { 
    locations(category) = product.location 
    } 
    locations.toMap 
} 

Answer 3

def getMap(products: List[Product]) = { 
     products.map(p => (p.categoryIdList.getOrElse(List.empty), p.location)) 
       .flatMap(x => x._1.map(_ -> x._2)) 
       .toMap 
}