将Model Class转换为ReactiveMongo BSONDocument

时间:2015-10-21 04:41:41

标签: json mongodb reactivemongo

我有一个案例类,它是我的应用程序中的模型对象,如下所示:

case class MyModel(myTypeA: TypeA, str: String)
case class TypeA(intVal: Int, strVal: String)

我使用MongoDB和ReactiveMongo库来满足我的持久性需求。我想将此类型转换为BSONDocument,以便我可以坚持这一点。我不想使用play reactive mongo模块,因为我将离开Play框架并在稍后阶段使用Spray。我知道我应该在BSON和我的模型类型之间编写隐式转换。作为首发,我做了以下几点:

implicit object MyModelBSONHandler extends BSONDocumentReader[MyModel] with BSONDocumentWriter[MyModel] {
  def read(doc: BSONDocument) = {
    MyModel(
      str = doc.getAs[String]("str").get,
      type = doc.getAs[TypeA]("myTypeA").get
    )
  }
  def write(myModel: MyModel) = {
    BSONDocument(
      "str" -> myModel.str,
      "myTypeA" -> myModel.myTypeA // Is this correct?
    )
  }
}

implicit object TypeABSONHandler extends BSONDocumentReader[TypeA] with BSONDocumentWriter[TypeA] {
  def read(doc: BSONDocument) = {
    TypeA(
      doc.getAs[Int]("intVal").get,
      doc.getAs[String]("strVal").get
    )
  }
  def write(typeA: TypeA) = {
    BSONDocument(
      "intVal" -> typeA.intVal,
      "strVal" -> typeA.strVal
    )
  }
}

可以看出,我不确定如何编写复杂的BSON类型?我还没试过我的代码库!

1 个答案:

答案 0 :(得分:3)

尝试通过Macros在对象伴随中使用隐式处理程序值。它更简单和优雅的方式来做到这一点。 There are类docs中的一些复杂处理程序示例。

import reactivemongo.bson.Macros

case class MyModel(myTypeA: TypeA, str: String)

object MyModel{
    implicit val handler = Macros.handler[MyModel]
}

case class TypeA(intVal: Int, strVal: String)

object TypeA{
    implicit val handler = Macros.handler[TypeA]
}

class class Foo(n: Int, model: MyModel, tpe: TypeA, date: Long,
                name: String/* and so on */ )

object Foo{
   //and simple handler in one line of code
   implicit val handler = Macros.handler[TypeA]
}

我在需要时使用自定义处理程序是一种非常特殊的行为,其他情况完全由宏覆盖。