• Home
• dev.mongocamp
• mongodb-driver_2.12
• 2.7.0
• source code
• Base.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.

dev.mongocamp.driver.mongodb.operation.Base.scala Maven / Gradle / Ivy

package dev.mongocamp.driver.mongodb.operation

import com.typesafe.scalalogging.LazyLogging
import dev.mongocamp.driver.mongodb.database.MongoIndex
import org.mongodb.scala.bson.conversions.Bson
import org.mongodb.scala.model.Sorts._
import org.mongodb.scala.model.{ CountOptions, DropIndexOptions, IndexOptions, Indexes }
import org.mongodb.scala.{ Document, ListIndexesObservable, MongoCollection, Observable, SingleObservable }

import scala.concurrent.duration.Duration
import scala.reflect.ClassTag

abstract class Base[A]()(implicit ct: ClassTag[A]) extends LazyLogging {

  protected def coll: MongoCollection[A]

  def count(filter: Bson = Document(), options: CountOptions = CountOptions()): Observable[Long] = {
    coll.countDocuments(filter, options)
  }

  def drop(): Observable[Void] = coll.drop()

  def createIndexForField(
      fieldName: String,
      sortAscending: Boolean = true,
      options: IndexOptions = IndexOptions()
  ): SingleObservable[String] = {
    if (sortAscending) {
      createIndex(ascending(fieldName), options)
    }
    else {
      createIndex(descending(fieldName), options)
    }
  }

  def createIndexForFieldWithName(
      fieldName: String,
      sortAscending: Boolean = true,
      name: String
  ): SingleObservable[String] =
    createIndexForField(fieldName, sortAscending, MongoIndex.indexOptionsWithName(Some(name)))

  def createUniqueIndexForField(
      fieldName: String,
      sortAscending: Boolean = true,
      name: Option[String] = None
  ): SingleObservable[String] =
    createIndexForField(fieldName, sortAscending, MongoIndex.indexOptionsWithName(name).unique(true))

  def createHashedIndexForField(fieldName: String, options: IndexOptions = IndexOptions()): SingleObservable[String] =
    createIndex(Indexes.hashed(fieldName), options)

  def createTextIndexForField(fieldName: String, options: IndexOptions = IndexOptions()): SingleObservable[String] =
    createIndex(Indexes.text(fieldName), options)

  def createExpiringIndexForField(
      fieldName: String,
      duration: Duration,
      sortAscending: Boolean = true,
      name: Option[String] = None
  ): SingleObservable[String] =
    createIndexForField(
      fieldName,
      sortAscending,
      MongoIndex.indexOptionsWithName(name).expireAfter(duration._1, duration._2)
    )

  def createIndex(key: Bson, options: IndexOptions = IndexOptions()): SingleObservable[String] =
    coll.createIndex(key, options)

  def dropIndexForName(name: String, options: DropIndexOptions = new DropIndexOptions()): SingleObservable[Void] =
    coll.dropIndex(name, options)

  def dropIndex(keys: Bson, options: DropIndexOptions = new DropIndexOptions()): SingleObservable[Void] =
    coll.dropIndex(keys, options)

  def listIndexes: ListIndexesObservable[Map[String, Any]] = coll.listIndexes[Map[String, Any]]()

  def indexList(): List[MongoIndex] = MongoIndex.convertIndexDocumentsToMongoIndexList(listIndexes)

  def indexForName(name: String): Option[MongoIndex] = indexList().find(_.name.equals(name))

  def hasIndexForField(fieldName: String): Boolean = indexList().exists(index => index.fields.contains(fieldName))

}