GridFS Database Sample App

Author: sfxcode

src/test/scala/com/sfxcode/nosql/mongo/gridfs/GridfsDatabase.scala

@@ -0,0 +1,16 @@
+package com.sfxcode.nosql.mongo.gridfs
+
+import com.sfxcode.nosql.mongo.GridFSDAO
+import com.sfxcode.nosql.mongo.database.DatabaseProvider
+
+/**
+ * GridFS Database Sample
+ */
+object GridfsDatabase {
+
+  val database = DatabaseProvider("test")
+
+  object ImageFilesDAO extends GridFSDAO(database, "images")
+
+}
+

src/test/scala/com/sfxcode/nosql/mongo/gridfs/GridfsDatabaseApp.scala

@@ -0,0 +1,37 @@
+package com.sfxcode.nosql.mongo.gridfs
+
+import com.sfxcode.nosql.mongo._
+import org.bson.types.ObjectId
+
+object GridfsDatabaseApp extends App with GridfsDatabaseFunctions {
+
+  val SourcePath = "src/test/resources/images/"
+  val TargetPath = "/tmp/"
+
+  case class ImageMetadata(name:String, group:String="logos", version:Int =1, indexSet:Set[Int] = Set(1,2,3))
+
+  dropImages
+
+  println(imagesCount)
+  createIndexOnImages("name")
+
+  val id: ObjectId = insertImage(SourcePath + "scala-logo.png", ImageMetadata("logo1"))
+
+  println(id)
+
+  val file = findImage(id.toString)
+  println(file)
+  val file2 = findImage(file)
+  println(file2)
+
+  downloadImage(file2,TargetPath + file.getFilename)
+
+  insertImage(SourcePath + "scala-logo.jpg", ImageMetadata("logo1", indexSet = Set(5,6,7)))
+  println(imagesCount)
+
+  val logos = findImages("group", "logos")
+
+  println(logos)
+
+
+}

src/test/scala/com/sfxcode/nosql/mongo/gridfs/GridfsDatabaseFunctions.scala

@@ -0,0 +1,35 @@
+package com.sfxcode.nosql.mongo.gridfs
+
+import better.files.File
+import com.sfxcode.nosql.mongo._
+import com.sfxcode.nosql.mongo.gridfs.GridfsDatabase._
+import org.bson.types.ObjectId
+import org.mongodb.scala.Completed
+import org.mongodb.scala.gridfs.GridFSFile
+
+trait GridfsDatabaseFunctions {
+
+  def createIndexOnImages(key:String): String =
+    ImageFilesDAO.createrMetadataIndex(key)
+
+  def deleteImage(id: ObjectId): Completed =
+    ImageFilesDAO.deleteOne(id)
+
+  def dropImages: Completed = ImageFilesDAO.drop()
+
+  def imagesCount: Long = ImageFilesDAO.count()
+
+  def insertImage(path: String, metadata: AnyRef): ObjectId = {
+    val file = File(path)
+    val stream = file.newInputStream
+    ImageFilesDAO.insertOne(file.name, stream, metadata)
+  }
+
+  def downloadImage(id: ObjectId, path: String): Long =
+    ImageFilesDAO.downloadToStream(id, File(path).path)
+
+  def findImage(id: ObjectId): GridFSFile = ImageFilesDAO.findById(id)
+
+  def findImages(key:String, value:String): List[GridFSFile] = ImageFilesDAO.findByMetadataValue(key, value)
+
+}