整理了初始化过程，现在更容易被继承了。

tests/unit/test_factory_endpoint_initialization.py

@@ -7,12 +7,23 @@ from pydantic import BaseModel
 
 from sync_state_machine.common.registry import DomainRegistry
 from sync_state_machine.common.sync_node import SyncNode
-from sync_state_machine.config import DbDataSourceConfig, LoggingConfig, PersistConfig, PipelineRunConfig
+from sync_state_machine.config import (
+    BaseDataSourceConfig,
+    LoggingConfig,
+    PersistConfig,
+    PipelineRunConfig,
+    register_datasource_type,
+)
 from sync_state_machine.datasource.db import BaseDbHandler, DbDataSource
 from sync_state_machine.pipeline.factory import create_pipeline_from_config
 from sync_state_machine.sync_system.strategy import DefaultSyncStrategy
 
 
+class EndpointInitDbConfig(BaseDataSourceConfig):
+    __test__ = False
+    type: str = "endpoint_init_db"
+
+
 class EndpointInitSchema(BaseModel):
     __test__ = False
     id: str
@@ -57,6 +68,7 @@ def _ensure_registered() -> None:
         reg.node_class = EndpointInitNode
         reg.strategy_class = EndpointInitStrategy
         reg.db_handler_class = EndpointInitDbHandler
+        reg.handler_classes["endpoint_init_db"] = EndpointInitDbHandler
         return
 
     DomainRegistry.register(
@@ -66,6 +78,16 @@ def _ensure_registered() -> None:
         strategy_class=EndpointInitStrategy,
         db_handler_class=EndpointInitDbHandler,
     )
+    DomainRegistry.register_handler(node_type, "endpoint_init_db", EndpointInitDbHandler)
+
+
+def _register_endpoint_init_datasource_type() -> None:
+    register_datasource_type(
+        "endpoint_init_db",
+        config_model=EndpointInitDbConfig,
+        factory=lambda config, endpoint_name: DbDataSource(),
+        replace=True,
+    )
 
 
 @pytest.mark.asyncio
@@ -74,16 +96,17 @@ async def test_factory_initializes_same_datasource_type_with_distinct_handler_co
     monkeypatch: pytest.MonkeyPatch,
 ) -> None:
     _ensure_registered()
+    _register_endpoint_init_datasource_type()
 
     config = PipelineRunConfig(
         node_types=["factory_endpoint_demo"],
-        local_datasource=DbDataSourceConfig(
-            type="db",
+        local_datasource=EndpointInitDbConfig(
+            type="endpoint_init_db",
             handler_configs={"factory_endpoint_demo": {"side": "local", "batch_size": 10}},
             target_project_ids=["project_local"],
         ),
-        remote_datasource=DbDataSourceConfig(
-            type="db",
+        remote_datasource=EndpointInitDbConfig(
+            type="endpoint_init_db",
             handler_configs={"factory_endpoint_demo": {"side": "remote", "batch_size": 20}},
             target_project_ids=["project_remote"],
         ),