AWSGlue中的GoogleBigQueryConnector是否允许编写自定义查询?
创始人
2024-09-25 17:02:56
0

AWS Glue中使用Google BigQuery Connector的过程中,确实能够编写自定义查询。下面是一个基本的编写自定义查询的示例:

import com.amazonaws.services.glue.util.AWSGlueArgParser
import com.amazonaws.services.glue.DynamicFrame

import com.google.cloud.hadoop.io.bigquery.BigQueryConfiguration
import com.google.cloud.hadoop.io.bigquery.BigQueryStrings
import com.google.cloud.hadoop.io.bigquery.BigQueryUtils

import org.apache.hadoop.mapreduce.Job

import org.apache.spark.SparkContext
import org.apache.spark.sql.SparkSession
import org.apache.spark.sql.DataFrame
import org.apache.spark.sql.functions._

object CustomQuery {
  def main(sysArgs: Array[String]) {
  
    // Glue job-specific args.
    val glueArgs = Map("JOB_NAME" -> "custom-query")
    
    // Google BigQuery related args.
    val projectId = "my-big-query-project"
    val bucketName = "my-bucket"
    val datasetName = "my-dataset"
    val tableName = "my-table"
    val customQuery = "SELECT COUNT(*) FROM `my-dataset.my-table`"
  
    // Use AWSGlueArgParser to parse args.
    val args = AWSGlueArgParser.getResolvedOptions(sysArgs, Seq("JOB_NAME").toArray)
  
    // Set Spark context.
    val sparkContext = new SparkContext()
    val sparkSession = SparkSession
      .builder()
      .appName(args("JOB_NAME"))
      .getOrCreate()
  
    // Initialize the BigQuery Configuration.
    val bigQueryConf = BigQueryConfiguration.configureBigQueryOutput(projectId)
  
    // Set the Google Cloud storage bucket to export results to.
    bigQueryConf.set(BigQueryConfiguration.OUTPUT_FORMAT_OPTION, BigQueryStrings.CSV)
    bigQueryConf.set(BigQueryConfiguration.GCS_BUCKET_OPTION, s"gs://$bucketName")
  
    // Set up the Job, OutputFormat and Table to write to.
    val job = Job.getInstance(sparkContext.hadoopConfiguration)
    BigQueryConfiguration.configureBigQueryOutput(job, projectId, customQuery, datasetName, tableName)
  
    // Load DataFrame from customQuery.
    val df = sparkSession.read
      .format("

相关内容

热门资讯

关于!微信微乐自建房辅助工具,... 关于!微信微乐自建房辅助工具,微乐家乡麻将自建房辅助app(开挂自建房)方法1)微乐家乡麻将自建房辅...
大家学习交流!微信卡五星辅助器... 大家学习交流!微信卡五星辅助器,微乐小程序免费黑科技(开挂自建房)方法1、很好的工具软件,可以解锁游...
三分钟了解!微乐微信自建房辅助... 三分钟了解!微乐微信自建房辅助工具插件,微乐家乡麻将自建房辅助app(开挂自建房)工具1、三分钟了解...
一分钟教会你!微乐家乡麻将自建... 一分钟教会你!微乐家乡麻将自建房辅助app,微乐自建房脚本免费入口(开挂自建房)方法1、用户打开应用...
分享实测!微信微乐自建房辅助工... 分享实测!微信微乐自建房辅助工具,微乐自建房脚本免费入口(开挂自建房)器1、起透看视 微信微乐自建房...
重大通报!微乐自建房辅助工具脚... 重大通报!微乐自建房辅助工具脚本,微乐小程序免费黑科技(开挂自建房)app微乐自建房辅助工具脚本破解...
如何分辨真伪!微乐微信自建房辅... 如何分辨真伪!微乐微信自建房辅助工具插件,微乐小程序免费黑科技(开挂自建房)技巧1、微乐微信自建房辅...
我来教教你!微乐自建房辅助工具... 我来教教你!微乐自建房辅助工具挂,微乐家乡麻将自建房辅助app(开挂自建房)工具1、玩家可以在微乐家...
重大来袭!微乐自建房辅助工具神... 重大来袭!微乐自建房辅助工具神器,微乐小程序免费黑科技(开挂自建房)器1、游戏颠覆性的策略玩法,独创...
带你了解!微信卡五星辅助器,微... 带你了解!微信卡五星辅助器,微乐自建房脚本免费入口(开挂自建房)方法1、带你了解!微信卡五星辅助器,...