隐式类

在Scala中,可以通过编写隐式转换,并在需要时将它们引入作用域,从而向封闭类添加新功能。这种方法的一个主要好处是,不必扩展现有的类来添加新功能。

从Scala 2.10开始,可以定义一个隐式类,然后在该类中定义方法来实现向String类添加自己的方法。

隐式类必须在允许方法定义的范围内定义(不是在顶层)。这意味着隐式类必须在class、object或package object中定义。

implicit class StringImprovements(val s: String) {
    def increment = s.map(c => (c + 1).toChar)
    def decrement = s.map(c => (c - 1).toChar)
    def hideAll = s.replaceAll(".", "*")
}

"ro`qj".increment         // spark

注意,可以在隐式类中定义任意数量的方法。建议对隐式方法定义的返回类型进行注释。如果遇到编译器找不到隐式方法的情况,或者在声明方法时只想显式声明,请将返回类型添加到方法定义中。

修改上面的代码,在increment、decrement和hideAll方法中,显式指定返回类型是String:

implicit class StringImprovements(val s: String) {
    // 明确每个方法返回一个字符串
    def increment: String = s.map(c => (c + 1).toChar)
    def decrement: String = s.map(c => (c − 1).toChar)
    def hideAll: String = s.replaceAll(".", "*")
}

尽管到目前为止显示的所有方法都返回了一个String,但是可以从方法中返回所需的任何类型。下面的类演示了几种不同类型的字符串转换方法:

implicit class StringImprovements(val s: String) {
    def increment = s.map(c => (c + 1).toChar)
    def decrement = s.map(c => (c − 1).toChar)
    def hideAll: String = s.replaceAll(".", "*")
    def plusOne = s.toInt + 1
    def asBoolean = s match {
        case "0" | "zero" | "" | " " => false
        case _ => true
    }
}

使用这些新方法,除了前面显示的String转换外,现在还可以执行Int和Boolean转换:

"4".plusOne       	// res0: Int = 5
"0".asBoolean      	// res1: Boolean = false
"1".asBoolean		// res2: Boolean = true

【示例】定义和使用隐式类。

// case类
case class Fruit(name:String, price:Double, productCode:Option[Long] = None)

// 定义隐式类
object FruitImplicits{
  implicit class AugmentedFruit(fruit: Fruit){
    def uuid:String = s"${fruit.name}-${fruit.productCode.getOrElse(1234)}"
  }
}

// 主类
object ImplicitClassDemo {
  def main(args: Array[String]): Unit = {
    val fruit = Fruit("苹果", 4.50, Some(1001))

    println(s"水果名称 = ${fruit.name}")
    println(s"水果价格 = ${fruit.price}")
    println(s"水果编程 = ${fruit.productCode.getOrElse(1000)}")

    import FruitImplicits._
    println(s"水果uuid = ${fruit.uuid}")
  }
}

执行以上代码,输出结果如下:

水果名称 = 苹果
水果价格 = 4.5
水果编程 = 1001
水果uuid = 苹果-1001

《PySpark原理深入与编程实战》