在我目前的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?也许定义另一个写定义?
请给我最好的镜头,并记住我是斯卡拉新手,所以要温柔=)!
答案 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
}