Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

DMS/DynamoDB/MongoDB: Use SQL with parameters instead of inlining values #243

Merged
merged 1 commit into from
Aug 26, 2024

Conversation

amotl
Copy link
Member

@amotl amotl commented Aug 26, 2024

About

This patch adjusts for an improvement to commons-codec where the SQLOperation class has been introduced to bundle together an SQL statement and its parameters.

References

Base automatically changed from amo/zyp-int64-timestamp to main August 26, 2024 11:27
@amotl amotl force-pushed the amo/io-sql-with-params branch from bdb0428 to ed5a465 Compare August 26, 2024 11:28
Comment on lines 135 to 153
# Define two CDC events: INSERT and UPDATE.
# They have to be conveyed separately because CrateDB needs a
# `REFRESH TABLE` operation between them.
event1 = {
"Records": [
wrap_kinesis(DYNAMODB_CDC_INSERT_NESTED),
]
}
event2 = {
"Records": [
wrap_kinesis(DYNAMODB_CDC_MODIFY_NESTED),
]
}

# Run transfer command.
handler(event1, None)
cratedb.database.refresh_table(table_name)
handler(event2, None)
cratedb.database.refresh_table(table_name)
Copy link
Member Author

@amotl amotl Aug 26, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

By adding and exercising this test case, I observed that an INSERT operation followed by an UPDATE operation on the same document/record indeed requires a REFRESH operation in between.

Before, the test case processed both CDC events in direct succession, and the outcome in CrateDB lacks corresponding fields added by the second operation (UPDATE), which adjusts data in the same record (CrateDB's data column).

Copy link
Member Author

@amotl amotl Aug 26, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This section has been amended in favor of adding the REFRESH TABLE operation to the handler itself. See below.

Comment on lines 125 to +132
# Process record.
sql = cdc.to_sql(record_data)
connection.execute(sa.text(sql))
operation = cdc.to_sql(record_data)
connection.execute(sa.text(operation.statement), parameters=operation.parameters)

# Processing alternating CDC events requires write synchronization.
# FIXME: Needs proper table name quoting.
connection.execute(sa.text(f"REFRESH TABLE {CRATEDB_TABLE}"))

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hereby, I am adding a REFRESH TABLE operation again. Please share your objections if you have any.

/cc @wierdvanderhaar, @hlcianfagna, @hammerhead, @proddata, @widmogrod

@amotl amotl requested review from seut and wierdvanderhaar August 26, 2024 14:58
@amotl amotl marked this pull request as ready for review August 26, 2024 14:58
pyproject.toml Outdated
@@ -139,7 +139,7 @@ docs = [
]
dynamodb = [
"boto3",
"commons-codec",
"commons-codec @ git+https://github.com/crate/commons-codec.git@sql-with-params",
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sure enough, this and other relevant spots need to be amended after running another release of commons-codec.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

commons-codec v0.0.12 has been released.

@amotl amotl force-pushed the amo/io-sql-with-params branch from 16df9f6 to 52bcc50 Compare August 26, 2024 15:32
- `SQLOperation` bundles data about an SQL operation, including
  statement and parameters.

- Also, refactor `DynamoDBFullLoadTranslator` to `commons-codec`.
@amotl amotl force-pushed the amo/io-sql-with-params branch from 52bcc50 to 8436e6b Compare August 26, 2024 17:19
@amotl amotl merged commit ebe0165 into main Aug 26, 2024
29 checks passed
@amotl amotl deleted the amo/io-sql-with-params branch August 26, 2024 17:23
@amotl amotl mentioned this pull request Sep 10, 2024
8 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant