added index handling

Author: sfxcode

src/main/scala/com/sfxcode/nosql/mongo/operation/Base.scala

@@ -1,12 +1,15 @@
 package com.sfxcode.nosql.mongo.operation
 
 import com.sfxcode.nosql.mongo._
-import com.sfxcode.nosql.mongo.bson.BsonConverter._
 
+import org.mongodb.scala.model.Sorts._
+
+import com.sfxcode.nosql.mongo.bson.BsonConverter._
 import com.typesafe.scalalogging.LazyLogging
 import org.bson.BsonValue
 import org.mongodb.scala.bson.conversions.Bson
-import org.mongodb.scala.{ Completed, DistinctObservable, Document, MongoCollection, Observable, Observer }
+import org.mongodb.scala.model.IndexOptions
+import org.mongodb.scala.{ Completed, DistinctObservable, Document, MongoCollection, Observable, Observer, SingleObservable }
 
 import scala.reflect.ClassTag
 
@@ -30,6 +33,27 @@ abstract class Base[A]()(implicit ct: ClassTag[A]) extends LazyLogging {
 
   def dropResult(): Completed = drop().headResult()
 
+  def createIndexForField(field: String, sortAscending: Boolean = true): SingleObservable[String] = {
+    if (sortAscending)
+      createIndex(ascending(field))
+    else
+      createIndex(descending(field))
+  }
+
+  def createIndexForFieldResult(field: String, sortAscending: Boolean = true): String = createIndexForField(field, sortAscending)
+
+  def createIndex(key: Bson, options: IndexOptions = IndexOptions()): SingleObservable[String] = coll.createIndex(key, options)
+
+  def createsIndexResult(key: Bson, options: IndexOptions = IndexOptions()): String = createIndex(key, options)
+
+  def dropIndexForName(name: String): SingleObservable[Completed] = coll.dropIndex(name)
+
+  def dropIndexForNameResult(name: String): Completed = dropIndexForName(name)
+
+  def dropIndex(keys: Bson): SingleObservable[Completed] = coll.dropIndex(keys)
+
+  def dropIndexResult(keys: Bson): Completed = dropIndex(keys)
+
 }
 
 class SimpleCompletedObserver[T] extends Observer[T] with LazyLogging {

src/test/scala/com/sfxcode/nosql/mongo/model/Person.scala

@@ -27,8 +27,7 @@ case class Person(
   friends: List[Friend],
   greeting: String,
   favoriteFruit: String,
-  _id: ObjectId = new ObjectId()
-)
+  _id: ObjectId = new ObjectId())
 
 case class Friend(id: Long, name: String)
 

src/test/scala/com/sfxcode/nosql/mongo/operation/BaseSpec.scala

@@ -1,8 +1,8 @@
 package com.sfxcode.nosql.mongo.operation
 
-import com.sfxcode.nosql.mongo.tour.Database._
 import com.sfxcode.nosql.mongo._
 import com.sfxcode.nosql.mongo.model.{ Author, Book }
+import com.sfxcode.nosql.mongo.tour.Database._
 import org.specs2.mutable.Specification
 import org.specs2.specification.BeforeAll
 
@@ -39,6 +39,18 @@ class BaseSpec extends Specification with BeforeAll {
       genderList must have size 1
     }
 
+    "must create / drop indexes for key" in {
+
+      val createIndexResult = PersonDAO.createIndexForFieldResult("name")
+
+      createIndexResult must be equalTo "name_1"
+
+      PersonDAO.createIndexForFieldResult("name") must be equalTo "name_1"
+
+      PersonDAO.dropIndexForNameResult("name_1") must not beNull
+
+    }
+
   }
 
   override def beforeAll: Unit = BookDAO.dropResult()