重复提示直至输入正确

Question

我最近正在学习 Scala。我以前已经习惯了C和Java。我想知道是否有一种更优雅的方式来重复要求输入，直到给出正确的输入。

val choiceType = {
      var in = ""
      var pass = false
      do {
    in = readLine()
    pass = in match {
        case "1" => println("Not implemented"); true
        case "2" => println("Not implemented"); true
        case "3" => println("Not implemented"); true
        case "4" => println("Not implemented"); true
        case "5" => println("Thanks for using."); true
        case _ => println("Error input. Please enter again. (Possible value: 1 - 5)"); false
    }
      } while (!pass)
      in.toInt
    }
    if (choiceType == 5) System.exit(0)

我想知道在 Scala 中是否有更好的方法来做到这一点？

Answer 1

您可以使用 Iterate.continually 一遍又一遍地执行相同的操作，直到施加一些停止条件（使用 dropWhile），或者您也可以使用 Iterator。 iterate 为您提供上一行，以防您想在错误消息中使用它：

val choiceType = Iterator.iterate(readLine())(line => {
  println("Error input: "+line+".  Please enter again (from 1-5).)")
  readLine()
}).collect(line => line match {
  case "1" => println("Not implemented"); line
  case "2" => println("Not implemented"); line
  case "3" => println("Not implemented"); line
  case "4" => println("Not implemented"); line
  case "5" => println("Thanks for using."); line
}).next.toInt

其工作方式是从 readLine 开始，然后如果需要另一行，它会根据上一行宣布错误消息行（显然这是错误的）并读取另一行。然后，您使用collect块来挑选出正确的输入；错误的输入只会被丢弃而不会被收集。在本例中，由于您想将其转换为整数，所以我只是传递该行。现在，我们只想要一个好的条目，因此我们获取下一个条目并将其转换为 int。

您还可以使用递归函数来做类似的事情：

def getChoice: String = {
  val line = readLine()
  line match {
    case "1" => println("Not implemented"); line
    case "2" => println("Not implemented"); line
    case "3" => println("Not implemented"); line
    case "4" => println("Not implemented"); line
    case "5" => println("Thanks for using."); line
    case _ => println("Error, blah blah."); getChoice
  }
}
val choiceType = getChoice.toInt

这里的技巧是，如果输入错误，您只需再次调用该函数即可。由于这是函数中发生的最后一件事，Scala 将避免真正的函数调用，而只是再次跳转到开头（尾递归），因此不会溢出堆栈。

Answer 2

递归 FTW，恕我直言。不管怎样，我会建议对 Rex Kerr 的递归解决方案进行一些修改。

def getChoice: String = {
  val line = readLine()
  line match {
    case "1" | "2" | "3" | "4" => println("Not implemented"); line
    case "5" => println("Thanks for using."); line
    case _   => println("Error, blah blah."); getChoice
  }
}

Answer 3

import io.Source.stdin

val choices = stdin.getLines.collect {
  case "1" => println("Not implemented")
  case "2" => println("Not implemented")
  case "3" => println("Not implemented")
  case "4" => println("Not implemented")
  case "5" => println("Thanks for using.")
              System.exit(0)
}

choices.next

Answer 4

Scala 允许您将 { case } 块视为 PartialFunction 特征的实例。 PartialFunction 为您提供了一种测试该函数是否是为特定输入定义的方法。因此，您可以这样重写：

val operation: PartialFunction[String, Unit] = {
  case "1" => println("Not implemented")
  case "2" => println("Not implemented")
  case "3" => println("Not implemented")
  case "4" => println("Not implemented")
  case "5" => println("Thanks for using."); System.exit(0)
}

var input: String = ""

do {
  input = readLine()
} while(!operation.isDefinedAt(input))

operation(input)

如果您想避免使用可变的 input 变量，您还可以使用 Iterator.continually() （将表达式转换为无限迭代器，该迭代器重复计算表达式）。

val operation = ... // as above
val input = Iterator.continually(readLine()).dropWhile(!operation.isDefinedAt(_)).next
operation(input)

通过使用 Iterator 的 collect 方法，您可以避免为 operation 指定名称。

Iterator.continually(readLine()).collect {
  case "1" => println("one")
}.next

Answer 5

我会使用递归函数。像这样的东西：

def getChoice: Int = readLine match {
  case x if x < "6" && x > "0" && x.length == 1 => x.toInt
  case _ => println("Error, Possible values: (1 - 5)")
            getChoice
}