Skip to content

Commit 0d1f3ac

Browse files
committed
Add mongodb and dynamodb
1 parent 714252e commit 0d1f3ac

1 file changed

Lines changed: 32 additions & 0 deletions

File tree

.github/workflows/run_spicebench.yml

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,10 @@ on:
2424
- databricks-lakebase
2525
- postgres-cdc-cayenne
2626
- postgres-cdc-duckdb
27+
- mongodb-cdc-cayenne
28+
- mongodb-cdc-duckdb
29+
- dynamodb-cdc-cayenne
30+
- dynamodb-cdc-duckdb
2731
etl_type:
2832
description: 'ETL type'
2933
required: true
@@ -359,6 +363,34 @@ jobs:
359363
ADAPTER_DOCKER_OPTS="${ADAPTER_DOCKER_OPTS} -e SPICEAI_API_KEY -e SPICE_CLOUD_API_URL"
360364
ADAPTER_ARGS="${ADAPTER_DOCKER_OPTS} ghcr.io/spiceai/spidapter:latest stdio --compute cloud --storage postgres --verbose --channel nightly --ec2-associate-public-ip"
361365
ADAPTER_ENVS=""
366+
elif [ "${SYSTEM_UNDER_TEST_PREFIX}" = "mongodb" ]; then
367+
export SPICEBENCH_TARGET_BATCH_ROWS=50000
368+
export SPICEBENCH_ADBC_DELETE_BATCH_SIZE=1000
369+
export SPICEBENCH_ADBC_UPDATE_STRATEGY=statement
370+
case "${SYSTEM_UNDER_TEST}" in
371+
mongodb-cdc-duckdb) SPICE_ACCELERATION="duckdb" ;;
372+
*) SPICE_ACCELERATION="cayenne" ;;
373+
esac
374+
ADAPTER_CMD="docker"
375+
ADAPTER_DOCKER_OPTS="run -i -e SPICE_ACCELERATION=${SPICE_ACCELERATION} -e EC2_SUBNET_ID -e EC2_SECURITY_GROUP_ID -e EC2_AMI_ID -e EC2_INSTANCE_TYPE -e AWS_REGION=us-west-2 -e EC2_IAM_INSTANCE_PROFILE"
376+
ADAPTER_DOCKER_OPTS="${ADAPTER_DOCKER_OPTS} -e AWS_ACCESS_KEY_ID=${S3_AWS_ACCESS_KEY_ID} -e AWS_SECRET_ACCESS_KEY=${S3_AWS_SECRET_ACCESS_KEY}"
377+
ADAPTER_DOCKER_OPTS="${ADAPTER_DOCKER_OPTS} -e SPICEAI_API_KEY -e SPICE_CLOUD_API_URL"
378+
ADAPTER_ARGS="${ADAPTER_DOCKER_OPTS} ghcr.io/spiceai/spidapter:latest stdio --compute cloud --storage mongo --verbose --channel nightly --ec2-associate-public-ip"
379+
ADAPTER_ENVS=""
380+
elif [ "${SYSTEM_UNDER_TEST_PREFIX}" = "dynamodb" ]; then
381+
export SPICEBENCH_TARGET_BATCH_ROWS=50000
382+
export SPICEBENCH_ADBC_DELETE_BATCH_SIZE=1000
383+
export SPICEBENCH_ADBC_UPDATE_STRATEGY=bulk_ingest_upsert
384+
case "${SYSTEM_UNDER_TEST}" in
385+
dynamodb-cdc-duckdb) SPICE_ACCELERATION="duckdb" ;;
386+
*) SPICE_ACCELERATION="cayenne" ;;
387+
esac
388+
ADAPTER_CMD="docker"
389+
ADAPTER_DOCKER_OPTS="run -i -e SPICE_ACCELERATION=${SPICE_ACCELERATION}"
390+
ADAPTER_DOCKER_OPTS="${ADAPTER_DOCKER_OPTS} -e AWS_ACCESS_KEY_ID=${S3_AWS_ACCESS_KEY_ID} -e AWS_SECRET_ACCESS_KEY=${S3_AWS_SECRET_ACCESS_KEY}"
391+
ADAPTER_DOCKER_OPTS="${ADAPTER_DOCKER_OPTS} -e SPICEAI_API_KEY -e SPICE_CLOUD_API_URL"
392+
ADAPTER_ARGS="${ADAPTER_DOCKER_OPTS} ghcr.io/spiceai/spidapter:latest stdio --compute cloud --storage dynamodb --verbose --channel nightly"
393+
ADAPTER_ENVS=""
362394
else
363395
export SPICEBENCH_ADBC_UPDATE_STRATEGY=bulk_ingest_upsert
364396
export SPICEBENCH_ADBC_FLUSH_STREAM_BEFORE_UPSERT=true

0 commit comments

Comments
 (0)