no.nrk.bigquery.client.http4s.internal.Http4sBQPollImpl.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bigquery-http4s-client_3 Show documentation

bigquery-http4s-client

There is a newer version: 0.19.2

Show newest version

/*
 * Copyright 2020 NRK
 *
 * SPDX-License-Identifier: MIT
 */

package no.nrk.bigquery.client.http4s.internal

import googleapis.bigquery.{Job, JobStatus}
import no.nrk.bigquery.*

private[client] object Http4sBQPollImpl {
  implicit val instance: BQPoll.FromJob[Job] = new BQPoll.FromJob[Job] {
    override def reference(job: Job): BQJobId =
      JobHelper.jobId(job).getOrElse(throw new IllegalStateException("Unable to get id from job"))

    override def toPoll(job: Job): BQPoll = {
      val maybeStatus: Option[JobStatus] = job.status

      val maybeFailed: Option[BQPoll.Failed] =
        maybeStatus.flatMap { (s: JobStatus) =>
          val primary = s.errorResult.map(proto => BQError(proto.location, proto.message, proto.reason))

          val details = s.errors.getOrElse(Nil).map(proto => BQError(proto.location, proto.message, proto.reason))

          if (primary.isEmpty && details.isEmpty) None
          else Some(BQPoll.Failed(BQExecutionException(reference(job), primary, details)))
        }

      maybeFailed.getOrElse {
        maybeStatus match {
          case Some(status) =>
            status.state match {
              case Some("DONE") => BQPoll.Success(job)
              case Some("PENDING") => BQPoll.Pending
              case Some("RUNNING") => BQPoll.Running
              case _ => BQPoll.Unknown
            }
          case None => BQPoll.Unknown
        }
      }
    }
  }
}