通过不获取类型，不匹配错误来生成火花组

Question

I am trying to get this variable GroupsByP to have certain type: GroupsByP is defined out of db connection select/collect statement which has 3 fields: 2 strings ( p and id ) and an int ( order ).

Expected result should be of the form Map[p,Set[(Id,Order)]]

val GroupsByP = db.pLinkGroups.collect()
  .groupBy(_.p)
  .map(group => group._1 -> (group._2.map(_.Id -> group._2.map(_.Order)).toSet))

my desired type for this variable is

Map[String, Set[(String, Int)]]

but actual is Map[String, Set[(String, Array[Int])]],

Answer 1

If I got your question right, this should do it:

 val GroupsByP: Map[String, Set[(String, Int)]] = input.collect()
    .groupBy(_.p)
    .map(group => group._1 -> group._2.map(record => (record.Id, record.Order)).toSet)

You should be mapping each record into a (Id, Order) tuple.

A very similar but perhaps more readable implementation might be:

val GroupsByP: Map[String, Set[(String, Int)]] = input.collect()
    .groupBy(_.p)
    .mapValues(_.map(record => (record.Id, record.Order)).toSet)