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("

相关内容

热门资讯

6分钟辅助!hhpoker是真... 6分钟辅助!hhpoker是真的假的,hhpoker真的有透视吗,演示教程(真是有挂)1、每一步都需...
第七分钟辅助!we poker... 第七分钟辅助!we poker插件,we poker免费辅助器,手筋教程(有挂方式)暗藏猫腻,小编详...
七分钟辅助!aa poker辅... 七分钟辅助!aa poker辅助包,pokemmo脚本辅助器下载,讲义教程(有挂秘籍)1、pokem...
第四分钟辅助!wepoker辅... 第四分钟辅助!wepoker辅助器安装包定制,aapoker透视脚本,大纲教程(有挂总结)所有人都在...
第四分钟辅助!wpk透视辅助靠... 第四分钟辅助!wpk透视辅助靠谱吗,wepoker透视苹果系统,诀窍教程(有挂猫腻);运wepoke...
七分钟辅助!aapoker透视... 七分钟辅助!aapoker透视脚本入口,wejoker开挂,学习教程(有挂教程)1、打开软件启动之后...
第四分钟辅助!淘宝买wepok... 第四分钟辅助!淘宝买wepoker透视有用吗,wpk俱乐部怎么作弊,积累教程(有挂规律)1、下载好淘...
6分钟辅助!we-poker软... 6分钟辅助!we-poker软件,购买的wpk辅助在哪里下载,积累教程(有挂技巧)购买的wpk辅助在...
第2分钟辅助!如何下载wepo... 第2分钟辅助!如何下载wepoker安装包,newpoker怎么安装脚本,教程书教程(今日头条)1、...
4分钟辅助!wepoker钻石... 4分钟辅助!wepoker钻石怎么看底牌,wepoker辅助器有哪些功能,法门教程(有挂详细)该软件...