如何处理我的json写入以更新ReactiveMongo中的现有对象?

时间:2013-08-15 19:48:39

标签: json scala playframework-2.1 securesocial reactivemongo

在我目前的Play2项目中,我已经实现了ReactiveMongo来存储我的用户对象。

用户:

case class User(override var _id: Option[BSONObjectID] = None,
                override var created: Option[DateTime] = None,
                override var updated: Option[DateTime] = None,
                firstName: String,
                lastName: String,
                email: String,
                emailValidated: Boolean,
                phoneNumber: String,
                lastLogin: DateTime,
                linkedProviders: Seq[LinkedProvider],
                userRoles: Seq[UserRole.UserRole]) extends TemporalModel {
}

import EnumUtils.enumReads
implicit val userRoleReads = enumReads(UserRole)
import mongo_models.LinkedProvider.linkedProviderReads
implicit val userReads: Reads[User] = (
    (__ \ "_id").read[Option[BSONObjectID]] and
      (__ \ "created").read[Option[DateTime]] and
      (__ \ "updated").read[Option[DateTime]] and
      (__ \ "firstName").read[String] and
      (__ \ "lastName").read[String] and
      (__ \ "email").read[String] and
      (__ \ "emailValidated").read[Boolean] and
      (__ \ "phoneNumber").read[String] and
      (__ \ "lastLogin").read[DateTime] and
      (__ \ "linkedProviders").read(list[LinkedProvider](linkedProviderReads)) and
      (__ \ "userRoles").read(list[UserRole.UserRole])
    )(User.apply _)


  import EnumUtils.enumWrites
  import mongo_models.LinkedProvider.linkedProviderWrites
  import play.api.libs.json.Writes._

  implicit val userWrites: Writes[User] = (
    (__ \ "_id").write[Option[BSONObjectID]] and
      (__ \ "created").write[Option[DateTime]] and
      (__ \ "updated").write[Option[DateTime]] and
      (__ \ "firstName").write[String] and
      (__ \ "lastName").write[String] and
      (__ \ "email").write[String] and
      (__ \ "emailValidated").write[Boolean] and
      (__ \ "phoneNumber").write[String] and
      (__ \ "lastLogin").write[DateTime] and
      (__ \ "linkedProviders").write(Writes.traversableWrites[LinkedProvider](linkedProviderWrites)) and
      (__ \ "userRoles").write(Writes.traversableWrites[UserRole.UserRole])
    )(unlift(User.unapply))

LinkedProvider:

case class LinkedProvider(userId: String,
                          providerId: String,
                          authMethod: String,
                          avatarUrl: String,
                          oAuth1Info: Option[OAuth1Info] = None,
                          oAuth2Info: Option[OAuth2Info] = None,
                          passwordInfo: Option[PasswordInfo] = None) {

}

object LinkedProvider {

  implicit val o1 = Json.format[OAuth1Info]
  implicit val o2 = Json.format[OAuth2Info]
  implicit val p = Json.format[PasswordInfo]

  implicit val linkedProviderReads: Reads[LinkedProvider] = (
    (__ \ "userId").read[String] and
      (__ \ "providerId").read[String] and
      (__ \ "authMethod").read[String] and
      (__ \ "avatarUrl").read[String] and
        (__ \ "oAuth1Info").read[Option[OAuth1Info]] and
        (__ \ "oAuth2Info").read[Option[OAuth2Info]] and
          (__ \ "passwordInfo").read[Option[PasswordInfo]]
    )(LinkedProvider.apply _)

  implicit val linkedProviderWrites: Writes[LinkedProvider] = (
    (__ \ "userId").write[String] and
      (__ \ "providerId").write[String] and
      (__ \ "authMethod").write[String] and
      (__ \ "avatarUrl").write[String] and
      (__ \ "oAuth1Info").write[Option[OAuth1Info]] and
      (__ \ "oAuth2Info").write[Option[OAuth2Info]] and
      (__ \ "passwordInfo").write[Option[PasswordInfo]]
    )(unlift(LinkedProvider.unapply))
}

如您所见,我已实现读取和写入,以便正确存储对象 在mongoDB实例中。 当使用新对象进行插入时,所有内容都像魅力一样工作,并且从mongoDB正确保存和检索对象结构。

我遇到的问题是如何处理更新。 所以,我想在我的用户对象上更新一些值,如下所示:

val userForUpdate = user.copy(
       firstName = identity.firstName,
       lastName = identity.lastName,
       email = identity.email.getOrElse(""),
       lastLogin = DateTime.now()
       )

然后调用我的更新方法:

UserDAO.update(user._id.get.stringify, userForUpdate, false)

updateMethod:

def update(id: String, document: T, upsert: Boolean)(implicit writer: Writes[T]): Future[Either[ServiceException, T]] = {
    document.updated = Some(new DateTime())
    Logger.debug(s"Updating document: [collection=$collectionName, id=$id, document=$document]")
    Recover(collection.update(DBQueryBuilder.id(id), DBQueryBuilder.set(document))) {
      document
    }
  }

DBQueryBuilder.set()方法:

def set[T](data: T)(implicit writer: Writes[T]): JsObject = Json.obj("$set" -> data)

这将导致此错误:

[error] application - DB operation failed: [message=DatabaseException['Mod on _id not allowed' (code = 10148)]]
[error] application - DatabaseException: [code=10148, isNotAPrimaryError=false]

因为写入(__ \ "_id").write[Option[BSONObjectID]]指出在调用 DBQueryBuilder.set()方法时也应该序列化 _id 字段。我们知道不允许更新 _id ,在这种情况下绝对不应该这样做。

所以我的问题是:我该如何处理?我想有一些聪明的Scala方式不会让我写出整个“$ set” - >查询?也许创建一个更好的DBQueryBuilder?也许定义另一个写定义?
请给我最好的镜头,并记住我是斯卡拉新手,所以要温柔=)!

2 个答案:

答案 0 :(得分:0)

而不是

def set[T](data: T)(implicit writer: Writes[T]): JsObject = Json.obj("$set" -> data)

你可以使用类似的东西

def set[T](data: T)(implicit writer: Writes[T]): JsObject = {
    val data = Json.obj(data)
    val dataWithoutId = data.remove("_id")
    Json.obj("$set" -> dataWithoutId)
}

Play Json库中应该有一些内容可以移除_id字段(但可能不是“remove(...)”...)


由于你使用选项,你可能会写一些类似的东西:

val userForUpdate = user.copy(
       id = None,
       firstName = identity.firstName,
       lastName = identity.lastName,
       email = identity.email.getOrElse(""),
       lastLogin = DateTime.now()
       )

最好的方法可能是对所有代表Mongo文档的类使用一个共同特征(也许是你使用的T?)并在特征上添加删除id的可能性(将其设置为none)。

这样你可以打电话:

def set[T](data: T)(implicit writer: Writes[T]): JsObject = Json.obj("$set" -> data.copy(_id = None))

答案 1 :(得分:0)

这个怎么样?

def set[T](data: T)(implicit writer: Writes[T]): JsObject = {
    val jsonTransformer = (__ \ '$set \ '_id).json.prune
    val dataWithId = Json.obj("$set" -> data)
    val withoutId = dataWithId.transform(jsonTransformer).get
    withoutId
}