码迷,mamicode.com
首页 > 编程语言 > 详细

Scala 枚举: Enumeration 概述

时间:2016-11-21 23:17:11      阅读:3669      评论:0      收藏:0      [点我收藏+]

标签:初始化   xtend   lte   类型   功能   spark   没有   gre   而不是   

Spark Scala 枚举


* Java 中的那套枚举并不能直接使用到 Scala 中
* Scala 中的枚举使用轻量级 Enumeration 进行实现
* Scala 中的枚举其实是一个伴随对象
* Scala 中的枚举没有方法重写功能
* Scala 中的枚举其实都是 Enumeration.Value 这个对象

和Java和C++不同,Scala并没有枚举类型。不过,Spark标准类库提供了一个 Enumeration类型,可以用于生成枚举,定义一个枚举类的对象,并以Value方法调用枚举中所有可能值,例如:

object WeekDay extends Enumeration {

   val Mon, Tue, Wed, Thu, Fri, Sat, Sun = Value    

     //在这里定义具体的枚举实例

  }

 

这里我们定义了7个字段:Mon, Tue, Wed, Thu, Fri, Sat, Sun,然后用Value将他们初始化,代码还可以如下表述:

val Mon = Value

val Tue = Value

val Wed = Value

val Thu = Value

val Fri = Value

val Sat= Value

val Sun = Value

 

也可以向Value方法传入ID,name,或两个参数都传入

 

val Mon = Value(0,”meeting”)

val Tue = Value(1)

val Wed = Value

val Thu = Value

val Fri = Value

val Sat= Value(10,”shopping”)

val Sun = Value

 

如果不指定ID值,则ID在前一个枚举值基础上+1,从零开始,缺省name字段为字段名

这时可以用WeekDay.Mon , WeekDay.Tue 来引用枚举值了

这里枚举的类型为WeekDay.Value而不是WeekDay,后者代表对象

可以添加一个类型名称这样定义

object WeekDay extends Enumeration {

    type WeekDay = Value                              

    //这里仅仅是为了将Enumration.Value的类型暴露出来给外界使用而已

    val Mon, Tue, Wed, Thu, Fri, Sat, Sun = Value 

    //在这里定义具体的枚举实例

  }

 

现在枚举类型变成了WeekDay.WeekDay

引用枚举类型的方法:

  import WeekDay._

  def isWorkingDay(d: WeekDay) = ! (d == Sat || d == Sun)

  WeekDay.values filter isWorkingDay foreach println      //使用语法糖进行输出

枚举值可以通过ID或name值进行调用

println(WeekDay(0)) // 输出 meeting

println(WeekDay.Mon)      //直接取枚举值 meeting

println(WeekDay.Mon.id) //取枚举值所在序号 0

println(WeekDay.maxId) //枚举值的个数 7

println(WeekDay.withName("meeting")) //通过字符串获取枚举(这里是不需要反射的) meeting

 

进一步通过如下代码理解枚举类型的构造与使用

举例:

—————————————定义 TrafficLightColor 对象—————————————-

object TrafficLightColor extends Enumeration {

  type TrafficLightColor = Value

  val Red = Value(0, "Stop")

  val Yellow = Value(10)

  val Green = Value("Go")

}

————————定义 Margin 对象以及Margin. Margin Enumeration类型———————

object Margin extends Enumeration {

  type Margin = Value

  val TOP, BOTTOM, LEFT, RIGHT = Value

}

 

————————————————使用方法——————————————————

import TrafficLightColor.TrafficLightColor

object Driver extends App {

  println(Margin.BOTTOM, Margin.BOTTOM.id)

  //输出 (BOTTOM,1)

  def doWhat(color: TrafficLightColor) = {

    if (color == TrafficLightColor.Red) "stop"

    else if (color == TrafficLightColor.Yellow) "hurry up" else "go"

  } 

  //使用match匹配

  def doWhat2(color: TrafficLightColor) = color match {

    case TrafficLightColor.Red    => "stop" 

    case TrafficLightColor.Yellow => "hurry up" 

    case _      => "go"

  }

  // load Red

  val red = TrafficLightColor(0) // Calls Enumeration.apply

   println(red, red.id)

  //输出:  (Stop,0)

    println(doWhat(red))

  //输出:  Stop

  println(doWhat2(TrafficLightColor.Yellow))

  //输出:  hurry up

  //打印出所有枚举

  Margin.values.foreach { v => println(v,v.id)}

  //输出:   (TOP,0)

  //       (BOTTOM,1)

  //       (LEFT,2)

  //       (RIGHT,3)

}

源码: https://github.com/scala/scala/blob/v2.12.0/src/library/scala/Enumeration.scala

Scala 枚举: Enumeration 概述

标签:初始化   xtend   lte   类型   功能   spark   没有   gre   而不是   

原文地址:http://www.cnblogs.com/bonnienote/p/6087195.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!