Skip to content

zentooo/embulk-output-bigquery

 
 

Repository files navigation

embulk-output-bigquery

Embulk output plugin to load/insert data into Google BigQuery using direct insert

Overview

load data into Google BigQuery as batch jobs for big amount of data https://developers.google.com/bigquery/loading-data-into-bigquery

  • Plugin type: output
  • Resume supported: no
  • Cleanup supported: no
  • Dynamic table creating: yes

NOT IMPLEMENTED

Current version of this plugin supports Google API with Service Account Authentication, but does not support OAuth flow for installed applications.

Configuration

  • auth_method: (private_key or compute_engine) (string, optional, default is private_key)
  • service_account_email: your Google service account email (string, required when auth_method is private_key)
  • p12_keyfile_path: fullpath of private key in P12(PKCS12) format (string, required when auth_method is private_key)
  • path_prefix: (string, required)
  • sequence_format: (string, optional, default is %03d.%02d)
  • file_ext: (string, required)
  • source_format: file type (NEWLINE_DELIMITED_JSON or CSV) (string, required, default is CSV)
  • project: project_id (string, required)
  • dataset: dataset (string, required)
  • table: table name (string, required)
  • auto_create_table: (boolean, optional default is 0)
  • schema_path: (string, optional)
  • prevent_duplicate_insert: (boolean, optional default is 0)
  • application_name: application name anything you like (string, optional)
  • delete_from_local_when_job_end: (boolean, optional, default is 0)
  • job_status_max_polling_time: max job status polling time. (int, optional, default is 3600 sec)
  • job_status_polling_interval: job status polling interval. (int, optional, default is 10 sec)
  • is_skip_job_result_check: (boolean, optional, default is 0)
  • field_delimiter: (string, optional, default is ",")
  • max_bad_records: (int, optional, default is 0)
  • encoding: (UTF-8 or ISO-8859-1) (string, optional, default is UTF-8)
  • ignore_unknown_values: (boolean, optional, default is 0)

Example

out:
  type: bigquery
  auth_method: private_key   # default
  service_account_email: ABCXYZ123ABCXYZ123.gserviceaccount.com
  p12_keyfile_path: /path/to/p12_keyfile.p12
  path_prefix: /path/to/output
  file_ext: csv.gz
  source_format: CSV
  project: your-project-000
  dataset: your_dataset_name
  table: your_table_name
  formatter:
    type: csv
    header_line: false
  encoders:
  - {type: gzip}

Authentication

There are two methods supported to fetch access token for the service account.

  1. Public-Private key pair
  2. Predefined access token (Compute Engine only)

The examples above use the first one. You first need to create a service account (client ID), download its private key and deploy the key with embulk.

On the other hand, you don't need to explicitly create a service account for embulk when you run embulk in Google Compute Engine. In this second authentication method, you need to add the API scope "https://www.googleapis.com/auth/bigquery" to the scope list of your Compute Engine instance, then you can configure embulk like this.

out:
  type: bigquery
  auth_method: compute_engine

Table id formatting

table and option accept Time#strftime format to construct table ids. Table ids are formatted at runtime using the local time of the embulk server.

For example, with the configuration below, data is inserted into tables table_2015_04, table_2015_05 and so on.

out:
  type: bigquery
  table: table_%Y_%m

Dynamic table creating

When auto_create_table is set to true, try to create the table using BigQuery API.

If table already exists, insert into it.

To describe the schema of the target table, please write schema path.

out:
  type: bigquery
  auto_create_table: true
  table: table_%Y_%m
  schema_path: /path/to/schema.json

Data Consistency

When prevent_duplicate_insert is set to true, embulk-output-bigquery generate job ID from md5 hash of file and other options to prevent duplicate data insertion.

job ID = md5(md5(file) + dataset + table + schema + source_format + file_delimiter + max_bad_records + encoding + ignore_unknown_values)

job ID must be unique(including failures). So same data can't insert with same settings.

In other words, you can retry as many times as you like, in case something bad error(like network error) happens before job insertion.

out:
  type: bigquery
  prevent_duplicate_insert: true

Build

$ ./gradlew gem

About

Embulk output plugin to load/insert data into Google BigQuery

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Java 99.5%
  • Ruby 0.5%