如何将ReactiveMongo与枚举一起使用?

时间:2015-07-11 22:50:08

标签: scala enums reactivemongo

我正在使用Play Framework和ReactiveMongo。我正在尝试为我的班级写一个名为Platforms的读者和作家。我正在尝试使用我创建的类型作为scala枚举,但我不知道应该如何定义读取器/写入器语法。有人可以帮我找出正确的语法吗?

import reactivemongo.bson._

sealed trait PlatformType { def name: String }
case object PROPER extends PlatformType { val name = "PROPER" }
case object TRANSACT extends PlatformType { val name = "TRANSACT" }
case object UPPER extends PlatformType { val name = "UPPPER" }


case class Platforms(
  id: Option[BSONObjectID],
  Platform: PlatformType,
  Active: Boolean,
  SystemIds:List[String],
  creationDate: Option[DateTime],
  updateDate: Option[DateTime])

object Platforms {

 implicit object PlatformsBSONReader extends BSONDocumentReader[Platforms] {
   def read(doc: BSONDocument): Platforms =
     Platforms(
       doc.getAs[BSONObjectID]("_id"), 
       doc.getAs[PlatformType]("Platform").get, 
       doc.getAs[Boolean]("Active").get,
       doc.getAs[List[String]]("SystemIds").get, 
       doc.getAs[BSONDateTime]("creationDate").map(dt => new DateTime(dt.value)),
       doc.getAs[BSONDateTime]("updateDate").map(dt => new DateTime(dt.value)))
 }  

 implicit object PlatformsBSONWriter extends BSONDocumentWriter[Platforms] {
    def write(platforms: Platforms): BSONDocument =
      BSONDocument(
        "_id" -> platforms.id.getOrElse(BSONObjectID.generate),
        "Platform" -> platforms.Platform,
        "Active" -> platforms.Active,
        "SystemIds" -> platforms.SystemIds,
        "creationDate" -> platforms.creationDate.map(date => BSONDateTime(date.getMillis)),
        "updateDate" -> platforms.updateDate.map(date => BSONDateTime(date.getMillis)))
  } 
}

1 个答案:

答案 0 :(得分:1)

PlatformType

implicit object PTW extends BSONWriter[PlatformType, BSONString] {
  def write(t: PlatformType): BSONString = BSONString(n.type)
}
implicit object PTR extends BSONReader[BSONValue, PlatformType] {
  def read(bson: BSONValue): PlatformType = bson match {
    case BSONString("PROPER") => PROPER
    // ...
  }
}
  

有一个关于BSON读者的在线documentation& ReactiveMongo的作家。