ElasticSearchScalaFuture.scala

object implicits {
  implicit class ESFuture[Response <: ActionResponse](future: ListenableActionFuture[Response])
    extends Future[Response] {

    override def onComplete[U](f: (Try[Response]) => U)(implicit executor: ExecutionContext): Unit = {
      future.addListener(new ActionListener[Response] {
        override def onFailure(e: Throwable): Unit = throw e

        override def onResponse(response: Response): Unit = f(Try(response))
      })
    }

    override def isCompleted: Boolean = future.isDone

    override def value: Option[Try[Response]] =
      future.isDone match {
        case true => Some(Try(future.get()))
        case false => None
      }

    @scala.throws[InterruptedException](classOf[InterruptedException])
    @scala.throws[TimeoutException](classOf[TimeoutException])
    override def ready(atMost: Duration)(implicit permit: CanAwait): ESFuture.this.type = {
      future.wait(atMost.toMillis)
      this
    }

    @scala.throws[Exception](classOf[Exception])
    override def result(atMost: Duration)(implicit permit: CanAwait): Response =
      future.get(atMost.toMillis, TimeUnit.MILLISECONDS)

  }
}