GridFS implicits

ObjectID implicits
DatabaseProvider with default registry

Author: sfxcode

src/main/scala/com/sfxcode/nosql/mongo/database/DatabaseProvider.scala

@@ -3,16 +3,22 @@ package com.sfxcode.nosql.mongo.database
 import com.sfxcode.nosql.mongo.bson.codecs.CustomCodecProvider
 import org.bson.codecs.configuration.CodecRegistries.fromProviders
 import org.bson.codecs.configuration.CodecRegistry
+import org.mongodb.scala.bson.codecs.DEFAULT_CODEC_REGISTRY
 import org.mongodb.scala.{ MongoClient, MongoDatabase }
 
-case class DatabaseProvider(databaseName: String, registry: CodecRegistry, client: MongoClient = MongoClient()) {
+case class DatabaseProvider(databaseName: String, registry: CodecRegistry = DEFAULT_CODEC_REGISTRY, client: MongoClient = MongoClient()) {
 
   import org.bson.codecs.configuration.CodecRegistries.fromRegistries
   import org.mongodb.scala.bson.codecs.DEFAULT_CODEC_REGISTRY
 
   private val CustomRegistry = fromProviders(CustomCodecProvider())
 
-  private val codecRegistry: CodecRegistry = fromRegistries(registry, CustomRegistry, DEFAULT_CODEC_REGISTRY)
+  val codecRegistry: CodecRegistry = {
+    if (registry == DEFAULT_CODEC_REGISTRY)
+      fromRegistries(registry, CustomRegistry)
+    else
+      fromRegistries(registry, CustomRegistry, DEFAULT_CODEC_REGISTRY)
+  }
 
   val database: MongoDatabase = client.getDatabase(databaseName).withCodecRegistry(codecRegistry)
 

src/main/scala/com/sfxcode/nosql/mongo/package.scala

@@ -3,8 +3,11 @@ package com.sfxcode.nosql
 import com.sfxcode.nosql.mongo.bson.BsonConverter
 import com.sfxcode.nosql.mongo.database.DatabaseProvider
 import com.sfxcode.nosql.mongo.operation.ObservableIncludes
+import org.bson.BsonValue
+import org.bson.types.ObjectId
 import org.mongodb.scala.bson.conversions.Bson
-import org.mongodb.scala.{ Document, FindObservable, MongoDatabase, Observable }
+import org.mongodb.scala.gridfs.{GridFSFile, GridFSFindObservable}
+import org.mongodb.scala.{Document, FindObservable, MongoDatabase, Observable}
 
 import scala.collection.JavaConverters._
 import scala.language.implicitConversions
@@ -29,6 +32,7 @@ package object mongo extends ObservableIncludes {
     documentFromScalaMap(map.toMap)
   }
 
+
   implicit def documentFromScalaMap(map: Map[String, Any]): Document = {
     var result = Document()
     map.keys.foreach(key => {
@@ -55,4 +59,17 @@ package object mongo extends ObservableIncludes {
     BsonConverter.asMapList(documents)
   }
 
+  implicit def stringToObjectId(str: String): ObjectId = new ObjectId(str)
+
+  implicit def documentToObjectId(doc: Document): ObjectId = doc.getObjectId("_id")
+
+  // gridfs
+
+  implicit def gridFSindObservableToFiles(observable:GridFSFindObservable):List[GridFSFile] = observable.resultList()
+
+
+  implicit def gridfSFileToObjectId(file: GridFSFile): ObjectId = file.getObjectId
+
+  implicit def gridfSFileToBSonIdValue(file: GridFSFile): BsonValue = file.getId
+
 }