Scala 模式匹配类型

Question

我写了以下代码：

array(0).getClass match {
  case Int.getClass =>
    byeBuffer = ByteBuffer.allocate(4 * array.length)

  case Long.getClass =>
    ByteBuffer.allocate(8 * array.length)

  case Float.getClass =>
    ByteBuffer.allocate(4 * array.length)

  case Double.getClass =>
    ByteBuffer.allocate(8 * array.length)

  case Boolean.getClass =>
    ByteBuffer.allocate(1 * array.length)

不过，过度使用getClass 让我觉得很笨拙。

有没有更好的写法？

Answer 1

您可以省略 getClass 并使用类型运算符 (:)：

val byteBuffer = array(0) match {
  case _: Int =>
    ByteBuffer.allocate(4 * array.length)
  case _: Long =>
    ByteBuffer.allocate(8 * array.length)
  case _: Float =>
    ByteBuffer.allocate(4 * array.length)
  case _: Double =>
    ByteBuffer.allocate(8 * array.length)
  case _: Boolean =>
    ByteBuffer.allocate(1 * array.length)
}

还要注意，match 是 Scala 中的表达式，因此您可以将 byteBuffer 移到外面并将结果分配给它。这种函数式方法将使其更简洁，并允许我们避免重新分配给var，而是使用val。

如果您想使用与类型匹配的变量，则可以简单地编写例如 l: Long 并使用名称为 Long 的变量 l。