scala - 如何在不抛出异常的情况下将字符串读入枚举？

Question

我有一些 .csv 格式的数据。它手动输入到 Excel 中，我必须将它加载到 Scala 程序中。

单个记录的大部分字段都是自由文本字符串或数字，但也有一些来自预先指定的可能字符串的短列表。我为这类信息创建了枚举，比如

object ScaleType extends Enumeration {
    val unknown = Value("unknown")
    val nominal = Value("nominal")
    val ordinal = Value("ordinal")
    val interval = Value("interval")
    val ratio = Value("ratio")
}

我将 csv 读入 Iterator[Array[String]]，并为每一行创建一个 scala 类的新实例，根据行中的信息设置其属性。假设在 csv 中，第 (8) 行告诉我们使用的刻度类型。所以，line是一个Array[String]，line(8)是一个字符串，它的内容应该是下面列出的值之一枚举。

由于数据是手工输入的，所以它包含错误。我使用 if 语句来确定 line(8) 是否完全为空，但我完全不知道如何查看我得到的字符串是否是刻度类型。

val scale = if(line(8).length > 0)
            {
                ScaleType.withName(line(8))
            }
        else ScaleType.unknown

我想要发生的事情:如果有人输入了比例类型“rtnl”，我希望上面的代码将 val scale 设置为 ScaleType.unknown 并且记录问题(类似 print("scale reading error on line "+ lineNumber 就足够了)。相反，会抛出异常，我不知道如何在异常发生之前检查问题发生了。

Answer 1

您可以使用 ValueSet.find:

val scale = ScaleType.values.find(_.toString == line(8)).getOrElse(ScaleType.unknown)

如果这样效率太低，您可以创建一个 map :

val lookup = ScaleType.values.map(v => (v.toString, v)).toMap
val scale = lookup.get(line(8)).getOrElse(ScaleType.unknown)