Scala编程与实践spark

scala中HashMap的map方法

2017-02-19  本文已影响140人  牛轧糖123

scala真是一门有趣的语言,虽然编译后的程序跟java一样也运行在JVM之上,但是scala可真不像java那样易于理解,前两天我就遇到了一个有趣的语法故事,是和scala中的集合类HashMap(Map也是一样的)的map方法有关的,下面我将记录下我的感受。

现象描述

在scala中有一个scala.collection.mutable.HashMap类,它可以存储k-v对,现在我新建了两个Map,变量名字是ss和sse,里面存储了一些内容

object ScalaAPITest {
  def main(args: Array[String]): Unit = {
    val ss = Map("hh" -> "sss", "kk" -> "mmm")
    val sse = Map("hh" -> "sss", "kk" -> "mmm")
  }
}

现在我想要对这个ss中的内容进行更新操作,在ss中的每一个v的后面再加上一个字符串“ooooo”,然后将ss的内容打印出来,于是代码变成了下面这样

object ScalaAPITest {
  def main(args: Array[String]): Unit = {
    val ss = Map("hh" -> "sss", "kk" -> "mmm")
    val sse = Map("hh" -> "sss", "kk" -> "mmm")
    val gg = ss.map(updateFun)
    gg.foreach(showContent)
  }
  def updateFun(v: (String, String)):(String, String) ={
    (v._1, v._2 + "ooooo")
  }
  def showContent(v: (String, String)):Unit ={
    println(v._1 + "----" + v._2)
  }
}

以上代码可以正常运行,并打印预期的结果,但是现在我要改变我想做的事情,我想将ss和sse中相同k的v相加起来存到ss中,这时再采用以上的方式似乎就做不到了,因为在updateFun中不能调用sse这个变量(虽然它是不变的,但是习惯了这么叫),于是就有了下面这种写法:

object ScalaAPITest {
  def main(args: Array[String]): Unit = {
    val ss = Map("hh" -> "sss", "kk" -> "mmm")
    val sse = Map("hh" -> "sss", "kk" -> "mmm")
    val gg = ss.map{
      case (k, v) => k -> {
          val ve = sse.getOrElse(k, ("random",  "random")) //在这里可以调用sse
          v + ve
        }
    }
    gg.foreach(showContent)
  }
  def showContent(v: (String, String)):Unit ={
    println(v._1 + "----" + v._2)
  }
}

大家可以从上面的代码看出这样做的好处,就是利用了模式匹配的方式是的可以在ss.map{}这个函数体里面调用sse,而在ss.map()中是做不到的。
这是一个很让我兴奋的地方,但是还不止这一个,还有另一个地方,请看下面一段代码:

object ScalaAPITest {
  def main(args: Array[String]): Unit = {
    val ss = Map("hh" -> "sss", "kk" -> "mmm")
    val sse = Map("hh" -> "sss", "kk" -> "mmm")
    val gg = ss.map{
      case (k, v) => {
          val ve = sse.getOrElse(k, ("random",  "random")) //在这里可以调用sse
          (k, v + ve)
        }
    }
    gg.foreach(showContent)
  }
  def showContent(v: (String, String)):Unit ={
    println(v._1 + "----" + v._2)
  }
}

请注意观察这段代码和上一段代码中不一样的地方,差异出现在case语句和它的方法体中,而这两段代码的达成的结果是相同的。

上一篇下一篇

猜你喜欢

热点阅读