Fix redis bench

shotover · Nov 19, 2024 · 9cbe78c · 9cbe78c
1 parent ced3d04
commit 9cbe78c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/shotover-proxy/benches/windsock/redis/bench.rs b/shotover-proxy/benches/windsock/redis/bench.rs
@@ -130,15 +130,17 @@ impl RedisBench {
             }
         }
 
-        common::generate_topology(SourceConfig::Redis(shotover::sources::redis::RedisConfig {
-            name: "redis".to_owned(),
-            listen_addr: host_address,
-            connection_limit: None,
-            hard_connection_limit: None,
-            tls: tls_acceptor,
-            timeout: None,
-            chain: TransformChainConfig(transforms),
-        }))
+        common::generate_topology(SourceConfig::Valkey(
+            shotover::sources::redis::ValkeyConfig {
+                name: "redis".to_owned(),
+                listen_addr: host_address,
+                connection_limit: None,
+                hard_connection_limit: None,
+                tls: tls_acceptor,
+                timeout: None,
+                chain: TransformChainConfig(transforms),
+            },
+        ))
     }
 
     async fn run_aws_shotover(

diff --git a/shotover/benches/benches/chain.rs b/shotover/benches/benches/chain.rs
@@ -241,7 +241,7 @@ fn criterion_benchmark(c: &mut Criterion) {
                     }
                     .get_builder(TransformContextConfig {
                         chain_name: "".into(),
-                        up_chain_protocol: MessageType::Redis,
+                        up_chain_protocol: MessageType::Valkey,
                     }),
                 )
                 .unwrap(),