com.github.jeroenr.tepkin.protocol.result.CreateIndexesResult.scala Maven / Gradle / Ivy

Go to download
package com.github.jeroenr.tepkin.protocol.result

import com.github.jeroenr.bson.{BsonDocument, BsonValueNumber}
import com.github.jeroenr.tepkin.protocol.exception.WriteException

case class CreateIndexesResult(ok: Boolean,
                               createdCollectionAutomatically: Boolean,
                               numIndexesBefore: Int,
                               numIndexesAfter: Int,
                               note: Option[String] = None,
                               errmsg: Option[String] = None,
                               code: Option[Int] = None) {
  /**
   * Used to explicitly throw an exception when the result is not OK.
   */
  def convertErrorToException(): this.type = {
    if (!ok && code.isEmpty) {
      throw new IllegalStateException("Result is not OK but there are no errors.")
    }

    if (!ok) {
      throw WriteException(List(WriteError(code.get, errmsg.getOrElse(""))))
    }

    this
  }
}

object CreateIndexesResult {
  def apply(document: BsonDocument): CreateIndexesResult = {
    CreateIndexesResult(
      ok = document.get[BsonValueNumber]("ok").map(_.toInt).getOrElse(0) == 1,
      createdCollectionAutomatically = document.getAs[Boolean]("createdCollectionAutomatically").getOrElse(false),
      numIndexesBefore = document.getAs[Int]("numIndexesBefore").getOrElse(0),
      numIndexesAfter = document.getAs[Int]("numIndexesAfter").getOrElse(0),
      note = document.getAs[String]("note"),
      errmsg = document.getAs[String]("errmsg"),
      code = document.getAs[Int]("code")
    )
  }
}