From 9cbe78ca33c23b429d085858d186c9906f62d384 Mon Sep 17 00:00:00 2001
From: Rony Rahman <ronycsdu@yahoo.com>
Date: Tue, 19 Nov 2024 13:34:45 +1100
Subject: [PATCH] Fix redis bench

---
 .../benches/windsock/redis/bench.rs           | 20 ++++++++++---------
 shotover/benches/benches/chain.rs             |  2 +-
 2 files changed, 12 insertions(+), 10 deletions(-)

diff --git a/shotover-proxy/benches/windsock/redis/bench.rs b/shotover-proxy/benches/windsock/redis/bench.rs
index 3ff8c359a..f4bd75bf0 100644
--- 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
index 597aaea8b..9b783be8e 100644
--- 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(),