dev.mongocamp.driver.mongodb.Field.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongodb-driver_2.12 Show documentation

mongodb-driver

There is a newer version: 2.8.1

package dev.mongocamp.driver.mongodb

import dev.mongocamp.driver.mongodb.bson.BsonConverter
import dev.mongocamp.driver.mongodb.database.DatabaseProvider._
import org.mongodb.scala.bson.BsonValue
import org.mongodb.scala.model.Accumulators._
import org.mongodb.scala.model.BsonField

object Field extends Field

trait Field {

  def firstField(fieldName: String): BsonField =
    first(fieldName, "$" + fieldName)

  def minField(fieldName: String): BsonField = min(fieldName, "$" + fieldName)

  def maxField(fieldName: String): BsonField = max(fieldName, "$" + fieldName)

  def lastField(fieldName: String): BsonField = last(fieldName, "$" + fieldName)

  def sumField(fieldName: String): BsonField = sum(fieldName, "$" + fieldName)

  def avgField(fieldName: String): BsonField = avg(fieldName, "$" + fieldName)

  def groupField(fieldName: String): BsonValue = groupFields(List(fieldName))

  def firstFields(fieldnames: Iterable[String]): Set[BsonField] =
    fieldnames.map(fieldname => firstField(fieldname)).toSet

  def minFields(fieldnames: Iterable[String]): Set[BsonField] =
    fieldnames.map(fieldname => minField(fieldname)).toSet

  def maxFields(fieldnames: Iterable[String]): Set[BsonField] =
    fieldnames.map(fieldname => maxField(fieldname)).toSet

  def lastFields(fieldnames: Iterable[String]): Set[BsonField] =
    fieldnames.map(fieldname => lastField(fieldname)).toSet

  def sumFields(fieldnames: Iterable[String]): Set[BsonField] =
    fieldnames.map(fieldname => sumField(fieldname)).toSet

  def avgFields(fieldnames: Iterable[String]): Set[BsonField] =
    fieldnames.map(fieldname => avgField(fieldname)).toSet

  def groupFields(fieldnames: Iterable[String]): BsonValue = {
    val list = fieldnames.map { name =>
      if (name.startsWith("$"))
        name
      else
        "$" + name
    }.toList
    BsonConverter.toBson(Map(ObjectIdKey -> list))
  }

}