mongodb

build.sbt

@@ -1,3 +1,9 @@
+import _root_.sbt.Keys._
+
+name := "lyrics-engine"
+
+version := "1.0-SNAPSHOT"
+
 name := "lyrics-engine"
 
 lazy val commonSettings = Seq(
@@ -16,10 +22,17 @@ lazy val parser = (project in file("parser"))
   .dependsOn(domain)
 lazy val api = (project in file("api"))
   .settings(commonSettings: _*)
+
+lazy val persistence = (project in file("persistence"))
+  .settings(commonSettings: _*)
+  .dependsOn(domain)  
 
 lazy val root = (project in file("."))
   .settings(commonSettings: _*)
   .dependsOn(api)
-  .aggregate(domain, parser, api)
+  .aggregate(domain, parser, api, persistence)
   .enablePlugins(UniversalPlugin, JavaAppPackaging)
 
+
+
+

persistence/build.sbt

@@ -0,0 +1,24 @@
+import sbt.Keys._
+
+name          := "persistence"
+
+organization  := "scalalab3.lyricsengine"
+
+version       := "0.1.0-SNAPSHOT"
+
+scalaVersion  := "2.11.8"
+
+resolvers += "releases"  at "https://oss.sonatype.org/content/groups/scala-tools"
+
+val specsV = "3.7.2"
+val scalaTestV = "2.2.6"
+
+libraryDependencies ++=
+  Seq(
+    "org.scalatest" %% "scalatest" % scalaTestV,
+    "org.specs2" %% "specs2-core" % specsV,
+    "org.specs2" %% "specs2-matcher-extra" % specsV,
+    "org.mongodb" %% "casbah" % "3.1.1",
+    "com.typesafe" % "config" % "1.3.0"
+  )
+

persistence/src/main/resources/application.conf

@@ -0,0 +1,10 @@
+mongo {
+  host= "localhost"
+  port= "8080"
+  user=""
+  password=""
+  dbname= "lyrics"
+  songscollection= "songs"
+  wordsdefinitions= "wordsdefinitions"
+}
+

persistence/src/main/scala/scalalab3/lyricsengine/persistence/StorageComponent.scala

@@ -0,0 +1,29 @@
+package scalalab3.lyricsengine.persistence
+
+import scalalab3.lyricsengine.domain._
+
+/**
+ * Created by annie on 5/17/16.
+ */
+trait StorageComponent {
+
+  val storage: Storage
+
+  trait Storage {
+
+    def addSongs(songsToAdd: Seq[Song], version: Option[Int] = None)
+
+    def addWordsDefinition(wd: WordsDefinition, version: Option[Int] = None)
+
+    def findSongs(version: Option[Int] = None): Seq[Song]
+
+    def findWordsDefinitions(version: Option[Int] = None): Seq[Map[Int, String]]
+
+    def countSongs(): Int
+
+    def countWD(): Int
+
+    def getLastVersion():Option[Int]
+  }
+
+}

persistence/src/main/scala/scalalab3/lyricsengine/persistence/StorageComponentImpl.scala

@@ -0,0 +1,129 @@
+package scalalab3.lyricsengine.persistence
+
+import com.mongodb.casbah.Imports._
+
+import scalalab3.lyricsengine.domain._
+import scalalab3.lyricsengine.persistence.mongo.MongoContext
+
+trait StorageComponentImpl extends StorageComponent {
+
+  override val storage: Storage
+
+
+  class StorageImpl(implicit context: MongoContext) extends Storage {
+
+    def addSongs(songsToAdd: Seq[Song], version: Option[Int] = None) = {
+      val builder = context.songsCollection.initializeOrderedBulkOperation //will automatically split the operation into batches
+      for {
+        song <- songsToAdd
+      }  builder.insert(songToMongoDBObj(song, version))
+      val result = builder.execute()
+    }
+
+    def addWordsDefinition(wd: WordsDefinition, version: Option[Int] = None) = {
+      val builder = context.wdCollection.initializeOrderedBulkOperation
+      builder.insert(wdToMongoDbObject(wd, version))
+
+      val result = builder.execute()
+    }
+
+    def findSongs(version: Option[Int] = None): Seq[Song] = {
+
+      val query = version match {
+        case Some(v) => MongoDBObject("version" -> v)
+        case None => MongoDBObject("version" -> null)
+      }
+      val result = context.songsCollection.find(query)
+
+      val songsSet = for {
+        song <- result
+      } yield songFromMongoDBObj(song)
+
+      songsSet.toSeq
+    }
+
+    def findWordsDefinitions(version: Option[Int] = None): Seq[Map[Int, String]] = {
+      val query = version match {
+        case Some(v) => MongoDBObject("version" -> version.get)
+        case None => MongoDBObject("version" -> null)
+      }
+      val result = context.wdCollection.find(query)
+
+      val wdSet = for {
+        wd <- result
+      } yield wdFromMongoDBObj(wd)
+
+      wdSet.toSeq
+    }
+
+    def countSongs(): Int = context.songsCollection.find().count()
+
+    def countWD(): Int = context.wdCollection.find().count()
+
+    private def songFromMongoDBObj(obj: MongoDBObject): Song = {
+      val _msdTrackId = obj.getAs[String]("msdTrackId").get
+      val _mxmTrackId = obj.getAs[String]("mxmTrackId").get
+
+      val _words = obj.getAs[Map[Int, Int]]("words").get
+
+      Song(msdTrackId = _msdTrackId, mxmTrackId = _mxmTrackId, words = _words)
+
+    }
+
+    private def wdFromMongoDBObj(obj: MongoDBObject): Map[Int, String] = {
+
+      val version = obj.getAs[Int]("version").getOrElse(null)
+
+      val words = obj.getAs[Map[String, String]]("wordsDefinitions").get.map { case (k, v) => (k.toInt, v.toString) }
+      words
+    }
+
+    private def songToMongoDBObj(song: Song, version: Option[Int]): MongoDBObject = {
+      val songBuilder = MongoDBObject.newBuilder
+      songBuilder +=("msdTrackId" -> song.msdTrackId,
+        "mxmTrackId" -> song.mxmTrackId,
+        "words" -> getWDefinitions(song),
+        "version" -> version.getOrElse(getLastVersion)
+        )
+      songBuilder.result()
+    }
+
+    private def wdToMongoDbObject(wd: WordsDefinition, version: Option[Int]): MongoDBObject = {
+      val wdBuilder = MongoDBObject.newBuilder
+      wdBuilder +=(
+        "wordsDefinitions" -> transformWordsDef(wd),
+        "version" -> version.getOrElse(getLastVersion.getOrElse(null))
+        )
+      wdBuilder.result()
+    }
+
+    private def transformWordsDef(w: WordsDefinition): MongoDBObject = {
+      val collection = MongoDBObject.newBuilder
+      val transformed = w.map { case (k, v) => (k.toString, v) }
+      collection ++= transformed
+      collection.result()
+    }
+
+    private def getWDefinitions(song: Song): MongoDBObject = {
+      val words = MongoDBObject.newBuilder
+      val songWords = song.words.map { case (k, v) => (k.toString, v) }
+      words ++= songWords
+      words.result()
+    }
+
+      def getLastVersion: Option[Int] = {
+      val query = MongoDBObject() // All documents
+      val fields = MongoDBObject("version" -> 1) // Only return `version`
+      val orderBy = MongoDBObject("version" -> -1) // Order by version descending
+
+      val result = context.wdCollection.findOne(query, fields, orderBy)
+
+      result match {
+        case Some(res) => res.getAs[Number]("version").map(_.intValue())
+        case None => None
+      }
+    }
+
+  }
+
+}

persistence/src/main/scala/scalalab3/lyricsengine/persistence/mongo/MongoConfig.scala

@@ -0,0 +1,32 @@
+package scalalab3.lyricsengine.persistence.mongo
+
+import com.typesafe.config.ConfigFactory
+
+import constants.DefaultConfigValues._
+import constants.DefaultConfigKeys._
+import scala.util.Try
+
+case class MongoConfig(host: String = defaultHost,
+                       port: Int = defaultPort,
+                       user: String = defaultUser,
+                       password: String = defaultPassword,
+                       dbName: String = defaultDbName,
+                       songsCollection: String = defaultSongsCollection,
+                       wdCollection: String  = defaultWDCollection)
+
+object MongoConfig {
+  private val config = ConfigFactory.load()
+
+  def load(): MongoConfig =
+    MongoConfig(
+      getString(host, defaultHost),
+      getInt(port, defaultPort),
+      getString(user, defaultUser),
+      getString(password, defaultPassword),
+      getString(dbname, defaultDbName),
+      getString(songsCollection, defaultSongsCollection),
+      getString(wordsDefinitions, defaultWDCollection)
+    )
+  private def getString(key: String, defaultValue: String) = Try(config.getString(key)).getOrElse(defaultValue)
+  private def getInt(key: String, defaultValue: Int) = Try(config.getInt(key)).getOrElse(defaultValue)
+}

persistence/src/main/scala/scalalab3/lyricsengine/persistence/mongo/MongoContext.scala

@@ -0,0 +1,37 @@
+package scalalab3.lyricsengine.persistence.mongo
+
+import com.mongodb.BasicDBObjectBuilder
+import com.mongodb.casbah.Imports._
+import com.mongodb.casbah.MongoClient
+
+class MongoContext(val config: MongoConfig) {
+
+  val mongoClient = MongoClient()
+  val mongoDB = mongoClient(config.dbName)
+
+  implicit val options: DBObject = BasicDBObjectBuilder.start().add("capped", true).add("size", 2000000000l).get()
+
+  def songsCollection =
+    if (mongoDB.collectionExists(config.songsCollection)) {
+      mongoDB(config.songsCollection)
+    } else {
+      mongoDB.createCollection(config.songsCollection, options)
+      mongoDB(config.songsCollection)
+    }
+
+  def wdCollection = {
+    if (mongoDB.collectionExists(config.wdCollection)) {
+      mongoDB(config.wdCollection)
+    } else {
+      mongoDB.createCollection(config.wdCollection, options)
+      mongoDB(config.wdCollection)
+    }
+  }
+
+  def drop = {
+    mongoDB.dropDatabase()
+  }
+
+
+
+}

persistence/src/main/scala/scalalab3/lyricsengine/persistence/mongo/constants/DefaultConfigKeys.scala

@@ -0,0 +1,13 @@
+package scalalab3.lyricsengine.persistence.mongo.constants
+
+object DefaultConfigKeys {
+  val mongo = "mongo"
+  val host = s"$mongo.host"
+  val port = s"$mongo.port"
+  val user = s"$mongo.user"
+  val dbname = s"$mongo.dbname"
+  val password = s"$mongo.password"
+  val songsCollection = s"$mongo.songscollection"
+  val wordsDefinitions = s"$mongo.wordsdefinitions"
+
+}

persistence/src/main/scala/scalalab3/lyricsengine/persistence/mongo/constants/DefaultConfigValues.scala

@@ -0,0 +1,11 @@
+package scalalab3.lyricsengine.persistence.mongo.constants
+
+object DefaultConfigValues {
+  val defaultHost = "localhost"
+  val defaultPort = 27017
+  val defaultUser = "admin"
+  val defaultPassword = ""
+  val defaultDbName = "lyrics"
+  val defaultSongsCollection = "songs"
+  val defaultWDCollection = "songs"
+}