File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change 5
5
from datahub .ingestion .source .sql .mysql import MySQLConfig , MySQLSource
6
6
from datahub .ingestion .source .state .checkpoint import Checkpoint
7
7
from datahub .ingestion .source .state .entity_removal_state import GenericCheckpointState
8
+ from datahub .ingestion .source .state .stale_entity_removal_handler import StaleEntityRemovalHandler
8
9
from sqlalchemy import create_engine
9
10
from sqlalchemy .sql import text
10
11
@@ -52,7 +53,9 @@ def get_current_checkpoint_from_pipeline(
52
53
# TODO: Refactor to use the helper method in the metadata-ingestion tests, instead of copying it here.
53
54
mysql_source = cast (MySQLSource , pipeline .source )
54
55
return mysql_source .state_provider .get_current_checkpoint (
55
- mysql_source .stale_entity_removal_handler .job_id
56
+ StaleEntityRemovalHandler .compute_job_id (
57
+ getattr (mysql_source , "platform" , "default" )
58
+ )
56
59
)
57
60
58
61
source_config_dict : Dict [str , Any ] = {
You can’t perform that action at this time.
0 commit comments