refactor: extract runtime state backends

crates/aether-testkit/src/bin/failure_recovery_baseline.rs

@@ -7,8 +7,10 @@ use aether_data::driver::postgres::{
     PostgresLeaseClaimOptions, PostgresLeaseClaimSpec, PostgresLeaseRunnerConfig,
     PostgresPoolConfig, PostgresTransactionOptions,
 };
-use aether_data::driver::redis::{RedisClientConfig, RedisLockRunnerConfig};
-use aether_data::{DataLayerError, PostgresBackend, RedisBackend};
+use aether_data::{DataLayerError, PostgresBackend};
+use aether_runtime_state::{
+    RedisClientConfig, RedisClientFactory, RedisLockRunner, RedisLockRunnerConfig,
+};
 use aether_testkit::{
     init_test_runtime_for, reserve_local_port, ManagedPostgresServer, ManagedRedisServer,
     TunnelHarness, TunnelHarnessConfig,
@@ -235,15 +237,20 @@ async fn benchmark_redis_restart_recovery(
     config: &FailureRecoveryBaselineConfig,
 ) -> Result<RecoverySummary, Box<dyn std::error::Error>> {
     let redis_url = redis_server.lock().await.redis_url().to_string();
-    let backend = RedisBackend::from_config(RedisClientConfig {
+    let factory = RedisClientFactory::new(RedisClientConfig {
         url: redis_url,
         key_prefix: Some(format!("aether-failure-recovery-{}", std::process::id())),
     })?;
-    let runner = backend.lock_runner(RedisLockRunnerConfig {
-        command_timeout_ms: Some(250),
-        default_ttl_ms: 1_000,
-    })?;
-    let keyspace = backend.keyspace();
+    let client = factory.connect_lazy()?;
+    let keyspace = factory.config().keyspace();
+    let runner = RedisLockRunner::new(
+        client,
+        keyspace.clone(),
+        RedisLockRunnerConfig {
+            command_timeout_ms: Some(250),
+            default_ttl_ms: 1_000,
+        },
+    )?;
     let collector = Arc::new(RecoveryCollector::default());
     let next_attempt = Arc::new(AtomicUsize::new(0));
     let phase = Arc::new(AtomicUsize::new(0));