Scala，强制数组/集合参数的长度

Question

假设我有一个类定义为

case class MyClass(myArray: Array[Int])

我只想允许 myArray 参数具有特定长度的实例，比如 3。

1. 我可以在类型/编译级别强制执行 myArray.size == 3 吗？
2. 其他系列会有所不同吗？ （比如说，一个不可变的列表）

我发现的唯一方法是在运行时通过智能构造函数检查 myArray 的大小并失败 - 例如使用 require。

Answer 1

我可以在类型/编译级别强制 myArray.size == 3 吗？

是的，Scala 可以represent numbers at type-level。例如，使用 shapeless 中的Sized

import shapeless._
import syntax.sized._
import nat._

case class MyClass(myArray: Sized[Array[Int], _3])
MyClass(Sized[Array](1,2,3))         // ok
MyClass(Sized[Array](1,2,3,4))       // compile-time error

Array(1,2,3,4).sized(3).map(MyClass) // None

以下是在 Scala 3 中使用 scala.compiletime.ops 设施的方法

import scala.compiletime.ops.int.S

enum MyArray[Size, +A]:
  case Nil extends MyArray[0, Nothing]
  case Cons[N <: Int, B](head: B, tail: MyArray[N, B]) extends MyArray[S[N], B]

import MyArray._

val xs: MyArray[3, Int] = Cons(1, Cons(2, Cons(3, Nil)))
val ys: MyArray[4, Int] = Cons(1, Cons(2, Cons(3, Cons(4, Nil))))

case class MyClass(myArray: MyArray[3, Int])
MyClass(xs) // ok
MyClass(ys) // compile-time error

Answer 2

你可以将 require 用于你想要的一切:)

case class MyClass(myArray: Array[Int]) {
  require(myArray.size == 3, "Only arrays with 3 items are allowed!")
}

如果你传递的数组不符合要求，它会抛出一个IllegalArgumentException。