如何從scala case類生成java數組

Question

我正在編寫一個小型數據訪問庫，以幫助我在Scala程序中使用Cassandra准備的語句（它不是開源的，但可能有一天）。 我想做的是為case類的bind語句自動生成一個Java Array。

com.datastax.driver.core
PreparedStatement...
  public BoundStatement bind(Object... values);

所以目前我有

case class Entity(foo:String, optionalBar:Option[String]) 

object Entity { 
  def toJArray(e:Entity) = { Array(e.foo, e.optionalBar.getOrElse(null)) } 
}

val e1 = Entity("fred", Option("bill"))
val e2 = Entity("fred", None)

Entity.toJArray(e1)
res5: Array[String] = Array(fred, bill)
Entity.toJArray(e2)
res6: Array[String] = Array(fred, null)

toJArray返回一個我可以在bind語句中使用的Array。 如果有日期，雙精度數或Java枚舉，則樣板代碼會變得更糟

new java.util.Date(createdOn)
scala.Double.box(price)
priceType.name

假設綁定參數與case類字段具有相同的順序，是否可以在Scala中自動生成Array？

編輯感謝@ srgfed01這是我想出的（不完整），但允許我做類似的事情

val customer1 = Customer( "email", "name", None, Option(new Date), OrdStatus.New)
session.execute(populate(customer1, insert))
val customer2 = Customer( "email2", "name2", Option(22), Option(new Date), OrdStatus.Rejected)
session.execute(populate(customer2, insert))

使用此功能

def populate(state:Product, statement:PreparedStatement): BoundStatement = {
  def set(bnd:BoundStatement, i:Int, aval:Any): Unit = {
    aval match {
      case v:Date => bnd.setDate(i, v)
      case v:Int => bnd.setInt(i, v)
      case v:Long => bnd.setLong(i, v)
      case v:Double => bnd.setDouble(i, v)
      case v:String => bnd.setString(i, v)
      case null => bnd.setToNull(i)
      case _ => bnd.setString(i, aval.toString)
    }
  }

  val bnd = statement.bind
  for(i <- 0 until state.productArity) {
    state.productElement(i) match {
      case op: Option[_]  => set(bnd, i, op.getOrElse(null))
      case v => set(bnd, i, v)
    }
  }
  bnd
}

Answer 1

您可以將productIterator調用用於您的case類對象：

case class Entity(foo: String, optionalBar: Option[String])

val e1 = Entity("fred", Option("bill"))
val e2 = Entity("fred", None)

def run(e: Entity): Array[Any] = e.productIterator
  .map {
    case op: Option[_] => op.getOrElse(null)
    case v             => v
  }
  .toArray

println(run(e1).mkString(" ")) // fred bill
println(run(e2).mkString(" ")) // fred null