Как получить доступ и обновить значение в изменяемой карте карты карт - PullRequest
9 голосов
/ 01 декабря 2010

У меня есть трехуровневая структура данных (отступы и разрывы строк для удобства чтения):

scala> import scala.collection.mutable.Map
import scala.collection.mutable.Map

scala> val m = Map("normal" -> Map("home" -> Map("wins" -> 0, "scores" -> 0),
                                   "away" -> Map("wins" -> 0, "scores" -> 0)))
m: scala.collection.mutable.Map[java.lang.String,
   scala.collection.mutable.Map[java.lang.String,
   scala.collection.mutable.Map[java.lang.String,Int]]] = 
Map((normal,Map(away -> Map(wins -> 0, scores -> 0),
     home -> Map(wins -> 0, scores -> 0))))

Для доступа к самым внутренним данным (счетам) требуется много ввода:

import org.scalatest.{Assertions, FunSuite}

class MapExamplesSO extends FunSuite with Assertions {
  test("Update values in a mutable map of map of maps") {
    import scala.collection.mutable.Map
    // The m map is essentially an accumulator
    val m = Map("normal" -> 
                Map("home" -> Map("wins" -> 0, "scores" -> 0),
                    "away" -> Map("wins" -> 0, "scores" -> 0)
                  )
          )
    //
    // Is there a less verbose way to increment the scores ?
    //
    assert(m("normal").apply("home").apply("scores") === 0)

    val s1 = m("normal").apply("home").apply("scores") + 1
    m("normal").apply("home").update("scores", s1)

    assert(m("normal").apply("home").apply("scores") === 1)

    val s2 = m("normal").apply("home").apply("scores") + 2
    m("normal").apply("home").update("scores", s2)

    assert(m("normal").apply("home").apply("scores") === 3)
  }
}

Есть ли менее подробный способ изменить значение баллов?

Я новичок в Scala, поэтому приветствуются и все другие наблюдения кода, приведенного выше.

Ответы [ 4 ]

21 голосов
/ 01 декабря 2010

Вам не нужно использовать «apply», просто сделайте это нормально с «()»

m("normal")("home")("scores") = 1
10 голосов
/ 01 декабря 2010

Вы можете написать

m("normal").apply("home").apply("scores")

в

m("normal")("home")("scores")

Однако я не уверен, что такая структура - хорошая идея. Может быть, вам стоит рассмотреть инкапсуляцию этой функциональности в специализированном классе.

3 голосов
/ 01 декабря 2010

Меньше многословно:

assert(m("normal")("home")("scores") === 0)

val s1 = m("normal")("home")("scores") + 1
m("normal")("home")("scores") = s1

assert(m("normal")("home")("scores") === 1)

val s2 = m("normal")("home")("scores") + 2
m("normal")("home")("scores") = s2

assert(m("normal")("home")("scores") === 3)

Это использует тот факт, что как apply, так и update имеют синтаксические сахара для них, как показано выше. Короче еще:

// On REPL, put both these definitions inside an object instead
// of entering them on different lines
def scores = m("normal")("home")("scores")
def scores_=(n: Int) = m("normal")("home")("scores") = n

assert(scores === 0)

val s1 = scores + 1
scores = s1

assert(scores === 1)

val s2 = scores + 2
scores = s2

// Just so you see these updates are being made to the map:
assert(m("normal")("home")("scores") === 3)

Что использует синтаксический сахар для добытчиков и сеттеров (определение получателя должно существовать, чтобы определение установщика работало).

3 голосов
/ 01 декабря 2010

Добавление локальной вспомогательной функции - это всегда хороший способ уменьшить дублирование кода:

class MapExamplesSO {
  def test {
    import scala.collection.mutable.Map
    // The m map is essentially an accumulator
    var m = Map("normal" -> 
                Map("home" -> Map("wins" -> 0, "scores" -> 0),
                    "away" -> Map("wins" -> 0, "scores" -> 0)))


    //Local Helper returns (Old, New)
    def updateScore(k1 : String,k2 : String,k3 : String)
                   (f : Int => Int) : (Int, Int) = {
      val old = m(k1)(k2)(k3)
      m(k1)(k2)(k3) = f(old)
      (old, m(k1)(k2)(k3))
    }

    assert(m("normal")(home")("scores") === 0)
    assert(updateScore("normal","home","scores")(_+1)._2 === 1)
    assert(updateScore("normal","home","scores")(_+2)._2 === 3)
  }
}

[Редактировать сделанный код более плотным]

...