commons/src/main/scala/io/prediction/commons/settings/mongodb/MongoSequences.scala

package io.prediction.commons.settings.mongodb

import com.mongodb.casbah.Imports.MongoDB
import com.mongodb.casbah.query.Imports._

/** Provides incremental sequence number generation. */
class MongoSequences(db: MongoDB) {
  private val seqColl = db("seq")

  /** Get the next sequence number from the given sequence name. */
  def genNext(name: String): Int = {
    val qFind = MongoDBObject("_id" -> name)
    val qField = MongoDBObject("next" -> 1)
    val qSort = MongoDBObject()
    val qRemove = false
    val qModify = $inc("next" -> 1)
    val qReturnNew = true
    val qUpsert = true
    seqColl.findAndModify(qFind, qField, qSort, qRemove, qModify, qReturnNew, qUpsert).get.getAsOrElse[Number]("next", 0).intValue
  }
}