• Home
• com.avsystem.commons
• commons-mongo_2.13
• 2.22.0
• source code
• Doc.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.avsystem.commons.mongo.Doc.scala Maven / Gradle / Ivy

package com.avsystem.commons
package mongo

import org.bson.{BsonDocument, BsonValue}

/**
  * @author MKej
  */
class Doc(private val doc: BsonDocument) extends AnyVal {
  def get[A, BSON <: BsonValue](key: DocKey[A, BSON]): Option[A] =
    Option(doc.get(key.key).asInstanceOf[BSON]).map(key.codec.fromBson)

  def getOpt[A, BSON <: BsonValue](key: DocKey[A, BSON]): Opt[A] =
    Opt(doc.get(key.key).asInstanceOf[BSON]).map(key.codec.fromBson)

  def require[A](key: DocKey[A, _ <: BsonValue]): A = getOpt(key).get

  def put[A](key: DocKey[A, _ <: BsonValue], value: A): Doc = {
    doc.put(key.key, key.codec.toBson(value))
    this
  }

  def putOpt[A](key: DocKey[A, _ <: BsonValue], optValue: Option[A]): Doc = optValue.fold(this)(put(key, _))

  def putOpt[A](key: DocKey[A, _ <: BsonValue], optValue: Opt[A]): Doc = optValue.fold(this)(put(key, _))

  def toBson: BsonDocument = doc
}

object Doc {
  def apply(): Doc = new Doc(new BsonDocument())
  def apply[A](key: DocKey[A, _ <: BsonValue], value: A): Doc = apply().put(key, value)
}