How to use ReactiveMongo with an enum?

2019-08-14 17:56发布

问题:

I am working with the Play Framework and ReactiveMongo. I am trying to write a reader and a writer for my class called Platforms. I am trying to use a type that I created as scala enum, but I don't know how the reader/writer syntax should be defined. Can someone help me figure out the correct syntax?

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:

For 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
    // ...
  }
}

There is an online documentation about the BSON readers & writers in ReactiveMongo.