Skip to main content

Databricks (S3 Staging)

Data Warehouse Target

Synopsis

The Databricks S3 target stages telemetry files to Amazon S3, then executes COPY INTO commands on Databricks SQL warehouses to load data into Unity Catalog tables.

Schema

- name: <string>
description: <string>
type: amazondatabricks
properties:
server_hostname: <string>
http_path: <string>
access_token: <string>
catalog: <string>
namespace: <string>
staging_bucket: <string>
staging_prefix: <string>
region: <string>
key: <string>
secret: <string>
session: <string>
table: <string>
schema: <string>
name: <string>
format: <string>
compression: <string>
extension: <string>
tables:
- table: <string>
schema: <string>
name: <string>
format: <string>
compression: <string>
extension: <string>
batch_size: <integer>
max_size: <integer>
timeout: <integer>
part_size: <integer>
drop_unknown_stream_events: <boolean>
field_format: <string>
max_rows_per_rowgroup: <numeric>
buffer_size: <numeric>
data_page_version: <string>
metadata: <key-value>
debug:
status: <boolean>
dont_send_logs: <boolean>

Configuration

The following fields are used to define the target:

Base Target Fields

FieldRequiredDefaultDescription
nameY-Unique identifier for this target
descriptionN-Human-readable description
typeY-Must be amazondatabricks
pipelinesN-Pipeline names to apply before sending
statusNtrueEnable/disable the target

Databricks Connection

FieldRequiredDefaultDescription
server_hostnameY-Databricks workspace URL (e.g., abc123.cloud.databricks.com)
http_pathY-SQL warehouse HTTP path (e.g., /sql/1.0/warehouses/abc123def456)
access_tokenY-Databricks personal access token
catalogY-Unity Catalog name
namespaceNdefaultDatabricks schema name

S3 Staging Configuration

FieldRequiredDefaultDescription
staging_bucketY-S3 bucket name for staging files
staging_prefixNdatabricks-staging/S3 prefix path
regionY-AWS region for S3 bucket
keyN*-AWS access key ID (uses default credentials chain if omitted)
secretN*-AWS secret access key
sessionN-AWS session token for temporary credentials

* = key and secret take effect only when both are set. session applies only alongside them.

Table Configuration

FieldRequiredDefaultDescription
tableY*-Catch-all table name for all events
schemaY*-Avro/Parquet schema: a Library name, built-in name, file path, or inline JSON definition
nameNvmetric.{{.Timestamp}}.{{.Extension}}File naming template
formatNjsonFile format. Use json, avro, or parquet — see the warning below.
compressionNzstdCompression algorithm
extensionN-File extension override. Defaults to the resolved format.
tablesN-Multiple table configurations (see below)
tables.tableY-Target table name
tables.schemaY*-Avro/Parquet schema for this table (Library name, built-in name, file path, or inline JSON)
tables.nameY-File naming template for this table
tables.formatN-File format for this table. Falls back to the catch-all format.
tables.compressionN-Compression algorithm for this table. Falls back to the catch-all compression.
tables.extensionN-File extension override for this table. Falls back to the catch-all extension.

* = At least one of table (catch-all) or tables (multiple) must be configured. For Avro/Parquet formats, schema is required.

warning

Only json, avro, and parquet are written correctly. csv, orc, and text pass configuration validation, but the staged file is written as JSON while the COPY INTO command still declares the requested format, so the load fails.

Parquet Options

These apply only when format is parquet.

FieldRequiredDefaultDescription
max_rows_per_rowgroupN10000Maximum rows per Parquet row group
buffer_sizeN262144Parquet page buffer size in bytes (256KB)
data_page_versionNV2Parquet data page version (V1, V2)
metadataN-Key/value pairs written into the Parquet file metadata

Batch Configuration

FieldRequiredDefaultDescription
batch_sizeN100000Maximum events per file before flush
max_sizeN33554432Maximum file size in bytes before flush (32MB)
timeoutN300COPY INTO command timeout in seconds
part_sizeN5S3 multipart upload part size in MB. Values below 5 are raised to 5.
drop_unknown_stream_eventsNtrueDrop events that do not match any configured table

Normalization

FieldRequiredDefaultDescription
field_formatN-Apply format normalization (ECS, ASIM, UDM)

Debug Options

FieldRequiredDefaultDescription
debug.statusNfalseEnable debug logging for this target
debug.dont_send_logsNfalseLog events without sending to Databricks

Details

Architecture Overview

The Databricks S3 target implements a two-stage loading pattern:

  1. Stage Files to S3: Events are written to files in S3 using the configured format
  2. Execute COPY INTO: SQL commands load data from S3 into Databricks Unity Catalog tables

Unity Catalog Integration

Catalog Structure:

  • Tables are organized within Unity Catalog using three-level namespace: catalog.namespace.table
  • The catalog field specifies the Unity Catalog name
  • The namespace field specifies the schema (defaults to default)
  • Table names are validated to ensure they are valid SQL identifiers

Warehouse ID Extraction:

  • The target automatically extracts the warehouse ID from the http_path
  • Example: /sql/1.0/warehouses/abc123def456abc123def456
  • This warehouse ID is used for all COPY INTO operations
Unity Catalog Permissions

The Databricks access token requires permissions to:

  • Execute SQL statements on the specified warehouse
  • Write data to the target catalog and schema
  • Access the S3 staging location (configured separately in Databricks)

S3 Staging Operations

File Upload:

  • Files are staged to s3://bucket/prefix/table/filename structure
  • Uses AWS SDK multipart upload for large files
  • Supports AWS credentials chain (access key, IAM role, instance profile)

Cleanup:

  • Staged files are automatically deleted after successful COPY INTO execution
  • Failed uploads remain in S3 for troubleshooting

File Format Support

Valid Formats:

  • CSV: Comma-separated values with optional headers
  • JSON: Newline-delimited JSON objects
  • AVRO: Schema-based binary format (requires schema)
  • ORC: Optimized row columnar format
  • PARQUET: Columnar storage format (requires schema)
  • TEXT: Plain text with delimiters

Schema Requirements:

  • Avro and Parquet formats require schema field with valid schema definition
  • Schema must match the expected table structure in Databricks
  • Other formats use schema inference from data

Multi-Table Routing

Catch-All Table:

  • Use table field to send all events to a single table
  • Simplest configuration for single-destination scenarios

Multiple Tables:

  • Use tables array to route different event types to different tables
  • Each table entry specifies table, schema, name, format fields
  • Events routed based on SystemS3 field in pipeline

Example Configuration:

tables:
- table: security_events
schema: security_schema.avsc
name: security.{{.Timestamp}}.parquet
format: parquet
- table: access_logs
schema: access_schema.avsc
name: access.{{.Timestamp}}.parquet
format: parquet

Performance Considerations

Batch Processing:

  • Events are buffered until batch_size or max_size limits are reached
  • Larger batches reduce S3 API calls and COPY INTO operations
  • Balance batch size against latency requirements

Upload Optimization:

  • Multipart uploads automatically handle large files
  • Configure part_size for optimal network performance
  • Default part size is AWS SDK default (5 MB)

COPY INTO Performance:

  • COPY INTO commands are executed with configurable timeout
  • Failed COPY operations return errors for retry logic
  • Warehouse must be running for COPY INTO to succeed
Warehouse State

Ensure the SQL warehouse is running before sending data. COPY INTO commands will fail if the warehouse is stopped. Configure warehouse auto-start or manual start procedures.

Error Handling

Upload Failures:

  • Failed S3 uploads are retried based on sender configuration
  • Permanent failures prevent COPY INTO execution
  • Check S3 bucket permissions and network connectivity

COPY INTO Failures:

  • Schema mismatches between files and tables cause failures
  • Invalid SQL identifiers (catalog, schema, table names) are rejected at validation
  • Check Databricks query history for detailed error messages

Examples

Basic Configuration

Sending telemetry to Databricks using S3 staging with Parquet format...

targets:
- name: databricks-warehouse
type: amazondatabricks
properties:
server_hostname: abc123.cloud.databricks.com
http_path: /sql/1.0/warehouses/abc123def456
access_token: "${DATABRICKS_TOKEN}"
catalog: production_data
namespace: telemetry
staging_bucket: datastream-staging
region: us-east-1
table: events
schema: event_schema.avsc
name: events.{{.Timestamp}}.parquet
format: parquet

With AWS Credentials

Using explicit AWS credentials for S3 staging access...

targets:
- name: databricks-secure
type: amazondatabricks
properties:
server_hostname: xyz789.cloud.databricks.com
http_path: /sql/1.0/warehouses/xyz789def123
access_token: "${DATABRICKS_TOKEN}"
catalog: security_analytics
namespace: logs
staging_bucket: security-logs-staging
staging_prefix: databricks/
region: us-west-2
key: "${AWS_ACCESS_KEY}"
secret: "${AWS_SECRET_KEY}"
table: security_events
schema: security_schema.avsc
name: security.{{.Timestamp}}.parquet
format: parquet

Multi-Table Configuration

Routing different event types to separate Databricks tables...

targets:
- name: databricks-multi-table
type: amazondatabricks
properties:
server_hostname: abc123.cloud.databricks.com
http_path: /sql/1.0/warehouses/abc123def456
access_token: "${DATABRICKS_TOKEN}"
catalog: analytics
namespace: production
staging_bucket: analytics-staging
region: us-east-1
tables:
- table: authentication_events
schema: auth_schema.avsc
name: auth.{{.Timestamp}}.parquet
format: parquet
- table: network_events
schema: network_schema.avsc
name: network.{{.Timestamp}}.parquet
format: parquet
- table: application_logs
schema: app_schema.avsc
name: app.{{.Timestamp}}.parquet
format: parquet

High-Volume Configuration

Optimizing for high-volume ingestion with batch limits and compression...

targets:
- name: databricks-high-volume
type: amazondatabricks
properties:
server_hostname: abc123.cloud.databricks.com
http_path: /sql/1.0/warehouses/abc123def456
access_token: "${DATABRICKS_TOKEN}"
catalog: high_volume_data
namespace: streaming
staging_bucket: streaming-staging
region: us-east-1
batch_size: 100000
max_size: 134217728
part_size: 16
timeout: 600
table: streaming_events
schema: streaming_schema.avsc
name: stream.{{.Timestamp}}.parquet
format: parquet
compression: snappy

JSON Format

Using JSON format for flexible schema evolution and debugging...

targets:
- name: databricks-json
type: amazondatabricks
properties:
server_hostname: abc123.cloud.databricks.com
http_path: /sql/1.0/warehouses/abc123def456
access_token: "${DATABRICKS_TOKEN}"
catalog: development
namespace: test_data
staging_bucket: dev-staging
region: us-east-1
table: test_events
name: test.{{.Timestamp}}.json
format: json

With Normalization

Applying ECS normalization before loading to Databricks...

targets:
- name: databricks-normalized
type: amazondatabricks
properties:
server_hostname: abc123.cloud.databricks.com
http_path: /sql/1.0/warehouses/abc123def456
access_token: "${DATABRICKS_TOKEN}"
catalog: security_data
namespace: normalized
staging_bucket: security-staging
region: us-east-1
field_format: ECS
table: ecs_events
schema: ecs_schema.avsc
name: ecs.{{.Timestamp}}.parquet
format: parquet

Production Configuration

Production-ready configuration with performance tuning, AWS credentials, and multi-table routing...

targets:
- name: databricks-production
type: amazondatabricks
properties:
server_hostname: production.cloud.databricks.com
http_path: /sql/1.0/warehouses/prod123abc456
access_token: "${DATABRICKS_TOKEN}"
catalog: production_analytics
namespace: telemetry
staging_bucket: production-staging-bucket
staging_prefix: datastream/databricks/
region: us-east-1
key: "${AWS_ACCESS_KEY}"
secret: "${AWS_SECRET_KEY}"
batch_size: 50000
max_size: 67108864
part_size: 10
timeout: 300
field_format: ASIM
tables:
- table: security_events
schema: security_schema.avsc
name: security.{{.Timestamp}}.parquet
format: parquet
compression: snappy
- table: audit_logs
schema: audit_schema.avsc
name: audit.{{.Timestamp}}.parquet
format: parquet
compression: snappy
- table: network_flows
schema: network_schema.avsc
name: network.{{.Timestamp}}.parquet
format: parquet
compression: snappy