Cloud Composer 1 | Cloud Composer 2
Nesta página, descrevemos como usar o DataflowTemplateOperator
para iniciar
pipelines do Dataflow no
Cloud Composer.
O pipeline Cloud Storage Text para BigQuery é um pipeline em lote que permite fazer upload de arquivos de texto armazenados no Cloud Storage, transformá-los usando uma função definida pelo usuário (UDF) em JavaScript fornecida por você e gerar os resultados no BigQuery.
Visão geral
Antes de iniciar o fluxo de trabalho, você criará as seguintes entidades:
Uma tabela vazia do BigQuery de um conjunto de dados vazio que conterá as seguintes colunas de informações:
location
,average_temperature
,month
e, opcionalmente,inches_of_rain
,is_current
elatest_measurement
.Um arquivo JSON que normalizará os dados do arquivo
.txt
para o formato correto para o esquema da tabela do BigQuery. O objeto JSON terá uma matriz deBigQuery Schema
, em que cada objeto conterá um nome de coluna, tipo de entrada e se é ou não um campo obrigatório.Um arquivo de entrada
.txt
que conterá os dados que serão enviados por upload em lote para a tabela do BigQuery.Uma função definida pelo usuário escrita em JavaScript que transformará cada linha do arquivo
.txt
nas variáveis relevantes para a tabela.Um arquivo DAG do Airflow que apontará para o local desses arquivos.
Em seguida, faça upload dos arquivos
.txt
,.js
e de esquema.json
para um bucket do Cloud Storage. Você também fará upload do DAG para o ambiente do Cloud Composer.Depois que o DAG for carregado, o Airflow executará uma tarefa a partir dele. Essa tarefa iniciará um pipeline do Dataflow que aplicará a função definida pelo usuário ao arquivo
.txt
e a formatará de acordo com o esquema JSON.Por fim, os dados serão enviados para a tabela do BigQuery que você criou anteriormente.
Antes de começar
- Este guia requer familiaridade com JavaScript para criar a função definida pelo usuário.
- Para seguir este guia, você precisa ter um ambiente do Cloud Composer. Consulte Criar ambiente para criar um. É possível usar qualquer versão do Cloud Composer com este guia.
-
Enable the Cloud Composer, Dataflow, Cloud Storage, BigQuery APIs.
Criar uma tabela vazia do BigQuery com definição de esquema
Criar uma tabela do BigQuery com uma definição de esquema Você vai usar essa definição de esquema posteriormente neste guia. Essa tabela do BigQuery manterá os resultados do upload em lote.
Para criar uma tabela vazia com definição de esquema, faça o seguinte:
Console
No console do Google Cloud, acesse a página do BigQuery:
No painel de navegação, na seção Recursos, expanda o projeto.
No painel de detalhes, clique em Criar conjunto de dados.
Na página "Criar conjunto de dados", na seção ID do conjunto de dados, nomeie o conjunto de dados como
average_weather
. Deixe todos os outros campos no estado padrão.Clique em Criar conjunto de dados.
Volte ao painel de navegação, na seção Recursos, expanda o projeto. Em seguida, clique no conjunto de dados
average_weather
.No painel de detalhes, clique em Criar tabela.
Na página Criar tabela, na seção Origem, selecione Tabela vazia.
Na página Criar tabela, na seção Destino:
Em Nome do conjunto de dados, escolha o conjunto de dados
average_weather
.No campo Nome da tabela, digite o nome
average_weather
.Verifique se Table type está definido como Native table.
Na seção Esquema, insira a definição do esquema. Use uma das seguintes abordagens:
Para inserir as informações do esquema manualmente, ative a opção Editar como texto e insira o esquema da tabela como uma matriz JSON. Digite os seguintes campos:
[ { "name": "location", "type": "GEOGRAPHY", "mode": "REQUIRED" }, { "name": "average_temperature", "type": "INTEGER", "mode": "REQUIRED" }, { "name": "month", "type": "STRING", "mode": "REQUIRED" }, { "name": "inches_of_rain", "type": "NUMERIC" }, { "name": "is_current", "type": "BOOLEAN" }, { "name": "latest_measurement", "type": "DATE" } ]
Use Adicionar campo para inserir manualmente o esquema:
Em Configurações de partição e cluster, use o valor padrão
No partitioning
.Na seção Opções avançadas, para Criptografia, use o valor padrão
Google-managed key
.Clique em Criar tabela.
bq
Use o comando bq mk
para criar um conjunto de dados vazios e uma tabela nesse conjunto de dados.
Execute o seguinte comando para criar um conjunto de dados de clima global médio:
bq --location=LOCATION mk \
--dataset PROJECT_ID:average_weather
Substitua:
LOCATION
: a região em que o ambiente está localizado.PROJECT_ID
: o ID do projeto.
Execute o seguinte comando para criar uma tabela vazia nesse conjunto de dados com a definição do esquema:
bq mk --table \
PROJECT_ID:average_weather.average_weather \
location:GEOGRAPHY,average_temperature:INTEGER,month:STRING,inches_of_rain:NUMERIC,is_current:BOOLEAN,latest_measurement:DATE
Após a criação da tabela, é possível atualizar a expiração, a descrição e os rótulos da tabela. É possível também modificar a definição do esquema.
Python
Salve este código como
dataflowtemplateoperator_create_dataset_and_table_helper.py
e atualize as variáveis nele para refletir seu projeto e local. Em seguida, execute-o com o seguinte comando:
python dataflowtemplateoperator_create_dataset_and_table_helper.py
Python
Para autenticar no Cloud Composer, configure o Application Default Credentials. Para mais informações, consulte Configurar a autenticação para um ambiente de desenvolvimento local.
crie um bucket do Cloud Storage
Crie um bucket para guardar todos os arquivos necessários para o fluxo de trabalho. O DAG que você criar mais adiante neste guia fará referência aos arquivos que você envia para esse bucket de armazenamento. Para criar um novo bucket de armazenamento:
Console
Abra o Cloud Storage no console do Google Cloud.
Clique em Criar bucket para abrir o formulário de criação de bucket.
Insira as informações do bucket e clique em Continuar para concluir cada etapa:
Especifique um Nome globalmente exclusivo para o bucket. Este guia usa
bucketName
como exemplo.Selecione Região para o tipo de local. Em seguida, selecione um Local em que os dados do bucket serão armazenados.
Selecione Padrão como a classe de armazenamento padrão dos dados.
Selecione o controle de acesso Uniforme para acessar os objetos.
Clique em Concluído.
gsutil
Use o comando gsutil mb
:
gsutil mb gs://bucketName/
Substitua:
bucketName
: o nome do bucket que você criou anteriormente neste guia.
Amostras de código
C#
Para autenticar no Cloud Composer, configure o Application Default Credentials. Para mais informações, consulte Configurar a autenticação para um ambiente de desenvolvimento local.
Go
Para autenticar no Cloud Composer, configure o Application Default Credentials. Para mais informações, consulte Configurar a autenticação para um ambiente de desenvolvimento local.
Java
Para autenticar no Cloud Composer, configure o Application Default Credentials. Para mais informações, consulte Configurar a autenticação para um ambiente de desenvolvimento local.
Python
Para autenticar no Cloud Composer, configure o Application Default Credentials. Para mais informações, consulte Configurar a autenticação para um ambiente de desenvolvimento local.
Ruby
Para autenticar no Cloud Composer, configure o Application Default Credentials. Para mais informações, consulte Configurar a autenticação para um ambiente de desenvolvimento local.
Criar um esquema do BigQuery formatado em JSON para a tabela de saída
Crie um arquivo de esquema do BigQuery formatado em JSON que corresponda à
tabela de saída criada anteriormente. Os nomes, tipos e modos de campos precisam corresponder aos definidos anteriormente no esquema da tabela do BigQuery. Esse arquivo normalizará os dados do arquivo .txt
para um formato compatível com o esquema do BigQuery. Nomeie esse arquivo como
jsonSchema.json
.
{
"BigQuery Schema": [
{
"name": "location",
"type": "GEOGRAPHY",
"mode": "REQUIRED"
},
{
"name": "average_temperature",
"type": "INTEGER",
"mode": "REQUIRED"
},
{
"name": "month",
"type": "STRING",
"mode": "REQUIRED"
},
{
"name": "inches_of_rain",
"type": "NUMERIC"
},
{
"name": "is_current",
"type": "BOOLEAN"
},
{
"name": "latest_measurement",
"type": "DATE"
}]
}
Criar um arquivo JavaScript para formatar os dados
Nesse arquivo, você definirá a função definida pelo usuário (UDF, na sigla em inglês) que fornece a lógica para transformar as linhas de texto no arquivo de entrada. Essa função usa cada linha de texto do arquivo de entrada como o próprio argumento, portanto, a função será executada uma vez para cada linha do arquivo de entrada. Nomeie esse arquivo como
transformCSVtoJSON.js
.
Criar o arquivo de entrada
Esse arquivo conterá as informações que você quer enviar para sua tabela do BigQuery. Copie esse arquivo localmente e nomeie-o como inputFile.txt
.
POINT(40.7128 74.006),45,'July',null,true,2020-02-16
POINT(41.8781 87.6298),23,'October',13,false,2015-02-13
POINT(48.8566 2.3522),80,'December',null,true,null
POINT(6.5244 3.3792),15,'March',14,true,null
Faça upload dos arquivos no seu bucket
Faça o upload dos seguintes arquivos para o bucket do Cloud Storage criado anteriormente:
- O esquema do BigQuery formatado em JSON (
.json
) - A função JavaScript definida pelo usuário (
transformCSVtoJSON.js
) O arquivo de entrada do texto que você quer processar (
.txt
)
Console
- No console do Google Cloud, acesse a página Buckets do Cloud Storage.
Na lista de buckets, clique no seu bucket.
Na guia Objetos do bucket, siga um destes procedimentos:
Arraste e solte os arquivos que você quer enviar da área de trabalho ou do gerenciador de arquivos para o painel principal no console do Google Cloud.
Clique no botão Fazer upload de arquivos, selecione os arquivos que você quer enviar na caixa de diálogo exibida e clique em Abrir.
gsutil
Execute o comando gsutil cp
:
gsutil cp OBJECT_LOCATION gs://bucketName
Substitua:
bucketName
: o nome do bucket que você criou anteriormente neste guia.OBJECT_LOCATION
: o caminho local do objeto. Por exemplo,Desktop/transformCSVtoJSON.js
.
Amostras de código
Python
Para autenticar no Cloud Composer, configure o Application Default Credentials. Para mais informações, consulte Configurar a autenticação para um ambiente de desenvolvimento local.
Ruby
Para autenticar no Cloud Composer, configure o Application Default Credentials. Para mais informações, consulte Configurar a autenticação para um ambiente de desenvolvimento local.
Configurar DataflowTemplateOperator
Antes de executar o DAG, defina as seguintes variáveis do Airflow.
Variável do Airflow | Valor |
---|---|
project_id
|
O ID do projeto |
gce_zone
|
Zona do Compute Engine em que o cluster do Dataflow precisa ser criado |
bucket_path
|
o local do bucket do Cloud Storage criado anteriormente |
Agora você fará referência aos arquivos criados anteriormente para criar um DAG que inicie
o fluxo de trabalho do Dataflow. Copie esse DAG e salve-o localmente como composer-dataflow-dag.py
.
Airflow 2
Airflow 1
Fazer upload do DAG para o Cloud Storage
Faça upload do DAG para a pasta /dags
no bucket do ambiente. Quando o upload for concluído, clique no link Pasta de DAGs na página de ambientes do Cloud Composer para conferi-lo.
Visualizar o status da tarefa
- Acesse a interface da Web do Airflow.
- Na página DAGs, clique no nome do DAG, como
composerDataflowDAG
. - Na página "Detalhes dos DAGs", clique em Visualizar gráfico.
Verificar status:
Failed
: a tarefa tem uma caixa vermelha ao redor. Também é possível manter o ponteiro sobre a tarefa e procurar por State: failed.Success
: a tarefa tem uma caixa verde ao redor. Também é possível manter o ponteiro sobre a tarefa e verificar se há State: Success.
Depois de alguns minutos, é possível verificar os resultados no Dataflow e no BigQuery.
Ver o job no Dataflow
No Console do Google Cloud, abra a página Dataflow.
O job é nomeado como
dataflow_operator_transform_csv_to_bq
com um ID exclusivo anexado ao final do nome com um hífen, da seguinte forma:Clique no nome para ver os detalhes do job.
Ver os resultados no BigQuery
No Console do Google Cloud, acesse a página BigQuery.
Você pode enviar consultas usando o SQL padrão. Use a consulta a seguir para ver as linhas que foram adicionadas à tabela:
SELECT * FROM projectId.average_weather.average_weather