This is an excerpt from the Scala Cookbook (partially modified for the internet). This is Recipe 10.26, “How to Create and Use Scala Enumerations”
Problem
You want to use an enumeration (a set of named values that act as constants) in your application.
Solution
Extend the scala.Enumeration class to create your enumeration:
package com.acme.app {
object Margin extends Enumeration {
type Margin = Value
val TOP, BOTTOM, LEFT, RIGHT = Value
}
}
Then import the enumeration to use it in your application:
object Main extends App {
import com.acme.app.Margin._
// use an enumeration value in a test
var currentMargin = TOP
// later in the code ...
if (currentMargin == TOP) println("working on Top")
// print all the enumeration values
import com.acme.app.Margin
Margin.values foreach println
}
Enumerations are useful tool for creating groups of constants, such as days of the week, weeks of the year, and many other situations where you have a group of related, constant values.
Scala Enumeration sort order
A nice feature of the Enumeration class is that it uses the order in which you define members as the sort order for each element in the enumeration. What I mean by this is that if you define an enumeration like this to define a deck of cards:
object FaceValue extends Enumeration {
type FaceValue = Value
val Ace, Two, Three, Four, Five, Six, Seven, Eight, Nine, Ten,
Jack, Queen, King = Value
}
and then print the elements like this:
object Test extends App {
import FaceValue._
// see that they're printed in order
FaceValue.values foreach println
// test less-than relationships
if (Ace < Two) println("Ace < Two")
if (Two < Three) println("Two < Three")
if (Three < Four) println("Three < Four")
if (Four < Five) println("Four < Five")
if (Five < Six) println("Five < Six")
if (Six < Seven) println("Six < Seven")
if (Seven < Eight) println("Seven < Eight")
if (Eight < Nine) println("Eight < Nine")
if (Nine < Ten) println("Nine < Ten")
if (Ten < Jack) println("Ten < Jack")
if (Jack < Queen) println("Jack < Queen")
if (Queen < King) println("Queen < King")
}
you’ll see this output:
Ace Two Three Four Five Six Seven Eight Nine Ten Jack Queen King Ace < Two Two < Three Three < Four Four < Five Five < Six Six < Seven Seven < Eight Eight < Nine Nine < Ten Ten < Jack Jack < Queen Queen < King
I find that to be a useful (essential?) trait of enumerations.
Using traits and case objects for enumerations
You can also use the following approach, but it generates about four times as much code as an Enumeration, most of which you won’t need if your sole purpose is to use it like an enumeration:
// a "heavier" approach
package com.acme.app {
trait Margin
case object TOP extends Margin
case object RIGHT extends Margin
case object BOTTOM extends Margin
case object LEFT extends Margin
}
See Also
| this post is sponsored by my books: | |||
#1 New Release |
FP Best Seller |
Learn Scala 3 |
Learn FP Fast |