scala - 如何为 Apache Spark 数据集中的枚举列编写案例类？

Question

story1, 10, small
story2, 20, medium
sotry3, 3, small
story4, 50, xlarge

我想将我的数据转换为数据集。我有一个列名 storyType（小、中、大、xlarge）。所以我不知道在这种情况下如何编写我的案例类

case class Story(name:String, point: Int, storyType: ???)

Answer 1

尝试

sealed trait StoryType
case object small extends StoryType
case object medium extends StoryType
case object large extends StoryType
case object xlarge extends StoryType