AWS Database Migration Service Operators¶
To use these operators, you must do a few things:
Airflow to AWS Database Migration Service (DMS) integration provides several operators to create and interact with DMS replication tasks.
One example_dag is provided which showcases some of these operators in action.
This example dag
DmsDeleteTaskOperator to create replication task, start it, wait for it
to be completed, and then delete it.
In the following code we create a new replication task, start it, wait for it to be completed and then delete it.
create_task = DmsCreateTaskOperator( task_id='create_task', replication_task_id=REPLICATION_TASK_ID, source_endpoint_arn=SOURCE_ENDPOINT_ARN, target_endpoint_arn=TARGET_ENDPOINT_ARN, replication_instance_arn=REPLICATION_INSTANCE_ARN, table_mappings=TABLE_MAPPINGS, )
start_task = DmsStartTaskOperator( task_id='start_task', replication_task_arn=create_task.output, )
wait_for_completion = DmsTaskCompletedSensor( task_id='wait_for_completion', replication_task_arn=create_task.output, )
delete_task = DmsDeleteTaskOperator( task_id='delete_task', replication_task_arn=create_task.output, )