gcp4zio.bq.BQClient.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gcp4zio-bq_2.13 Show documentation

Scala interface to Google Cloud API based on ZIO

The newest version!

package gcp4zio.bq

import com.google.auth.oauth2.{GoogleCredentials, ServiceAccountCredentials}
import com.google.cloud.bigquery.{BigQuery, BigQueryOptions}
import zio.{Task, ZIO}
import java.io.FileInputStream

object BQClient {

  private def getBQ(path: String): BigQuery = {
    val credentials: GoogleCredentials = ServiceAccountCredentials.fromStream(new FileInputStream(path))
    BigQueryOptions.newBuilder().setCredentials(credentials).build().getService
  }

  /** Returns BigQuery object wrapped in ZIO
    * @param path
    *   Optional path to Service Account Credentials file
    * @return
    *   RIO[Scope, BigQuery]
    */
  def apply(path: Option[String] = None): Task[BigQuery] = ZIO.attempt {
    val envPath: String = sys.env.getOrElse("GOOGLE_APPLICATION_CREDENTIALS", "NOT_SET_IN_ENV")
    path match {
      case Some(p) =>
        logger.info("Using credentials from values passed in function for BigQuery Client")
        getBQ(p)
      case None =>
        if (envPath == "NOT_SET_IN_ENV") {
          logger.info("Using credentials from local sdk for BigQuery Client")
          BigQueryOptions.getDefaultInstance.getService
        } else {
          logger.info("Using credentials from GOOGLE_APPLICATION_CREDENTIALS for BigQuery Client")
          getBQ(envPath)
        }
    }
  }
}