ongo.reactivemongo-play-json_2.11.0.11.12-play24.source-code.findandmodify.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2012-2013 Stephane Godbillon (@sgodbillon)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package reactivemongo.play.json.commands

import play.api.libs.json.{ Json, JsObject, OWrites }, Json.JsValueWrapper

import reactivemongo.api.commands.{
  FindAndModifyCommand,
  ResolvedCollectionCommand
}
import reactivemongo.play.json.JSONSerializationPack

object JSONFindAndModifyCommand
    extends FindAndModifyCommand[JSONSerializationPack.type] {
  val pack: JSONSerializationPack.type = JSONSerializationPack
}

object JSONFindAndModifyImplicits {
  import JSONFindAndModifyCommand._
  import reactivemongo.utils.option

  implicit object FindAndModifyResultReader
      extends DealingWithGenericCommandErrorsReader[FindAndModifyResult] {

    def readResult(result: JsObject): FindAndModifyResult = try {
      FindAndModifyResult(
        (result \ "lastErrorObject").asOpt[JsObject].map { doc =>
          UpdateLastError(
            updatedExisting = (doc \ "updatedExisting").
              asOpt[Boolean].getOrElse(false),
            n = (doc \ "n").asOpt[Int].getOrElse(0),
            err = (doc \ "err").asOpt[String],
            upsertedId = (doc \ "upserted").toOption
          )
        },
        (result \ "value").asOpt[JsObject]
      )
    } catch {
      case e: Throwable =>
        e.printStackTrace()
        sys.error("Error")
    }
  }

  implicit object FindAndModifyWriter
      extends OWrites[ResolvedCollectionCommand[FindAndModify]] {

    def writes(command: ResolvedCollectionCommand[FindAndModify]): JsObject = {
      val optionalFields = List[Option[(String, JsValueWrapper)]](
        command.command.sort.map("sort" -> _),
        command.command.fields.map("fields" -> _)
      ).flatten

      Json.obj(
        "findAndModify" -> command.collection,
        "query" -> command.command.query
      ) ++
        Json.obj(optionalFields: _*) ++
        (command.command.modify match {
          case Update(document, fetchNewObject, upsert) => Json.obj(
            "update" -> document,
            "new" -> fetchNewObject,
            "upsert" -> upsert
          )

          case Remove => Json.obj("remove" -> true)
        })
    }
  }
}