Add KeyDB to crud-bench benchmark

Cargo.toml

@@ -6,7 +6,8 @@ license = "Apache-2.0"
 readme = "README.md"
 
 [features]
-default = ["redis", "redb", "rocksdb", "postgres", "mongodb", "surrealkv", "surrealdb"]
+default = ["keydb", "redis", "redb", "rocksdb", "postgres", "mongodb", "surrealkv", "surrealdb"]
+keydb = ["dep:redis"]
 mongodb = ["dep:mongodb"]
 postgres = ["dep:tokio-postgres"]
 redb = ["dep:redb"]

src/keydb.rs

@@ -0,0 +1,60 @@
+#![cfg(feature = "keydb")]
+
+use anyhow::Result;
+use redis::aio::Connection;
+use redis::{AsyncCommands, Client};
+
+use crate::benchmark::{BenchmarkClient, BenchmarkEngine, Record};
+use crate::docker::DockerParams;
+
+pub(crate) const KEYDB_DOCKER_PARAMS: DockerParams = DockerParams {
+	image: "eqalpha/keydb",
+	pre_args: "",
+	post_args: "keydb-server --requirepass root",
+};
+
+#[derive(Default)]
+pub(crate) struct KeydbClientProvider {}
+
+impl BenchmarkEngine<KeydbClient> for KeydbClientProvider {
+	async fn create_client(&self, endpoint: Option<String>) -> Result<KeydbClient> {
+		let url = endpoint.unwrap_or("redis://:[email protected]:6379/".to_owned());
+		let client = Client::open(url)?;
+		let conn = client.get_async_connection().await?;
+		Ok(KeydbClient {
+			conn,
+		})
+	}
+}
+
+pub(crate) struct KeydbClient {
+	conn: Connection,
+}
+
+impl BenchmarkClient for KeydbClient {
+	#[allow(dependency_on_unit_never_type_fallback)]
+	async fn create(&mut self, key: i32, record: &Record) -> Result<()> {
+		let val = bincode::serialize(record)?;
+		self.conn.set(key, val).await?;
+		Ok(())
+	}
+
+	async fn read(&mut self, key: i32) -> Result<()> {
+		let val: Vec<u8> = self.conn.get(key).await?;
+		assert!(!val.is_empty());
+		Ok(())
+	}
+
+	#[allow(dependency_on_unit_never_type_fallback)]
+	async fn update(&mut self, key: i32, record: &Record) -> Result<()> {
+		let val = bincode::serialize(record)?;
+		self.conn.set(key, val).await?;
+		Ok(())
+	}
+
+	#[allow(dependency_on_unit_never_type_fallback)]
+	async fn delete(&mut self, key: i32) -> Result<()> {
+		self.conn.del(key).await?;
+		Ok(())
+	}
+}

src/main.rs

@@ -2,6 +2,8 @@ use crate::benchmark::{Benchmark, BenchmarkResult};
 use crate::docker::DockerContainer;
 use crate::docker::DockerParams;
 use crate::dry::DryClientProvider;
+#[cfg(feature = "keydb")]
+use crate::keydb::KeydbClientProvider;
 #[cfg(feature = "mongodb")]
 use crate::mongodb::MongoDBClientProvider;
 #[cfg(feature = "postgres")]
@@ -26,6 +28,7 @@ use tokio::runtime::Builder;
 mod benchmark;
 mod docker;
 mod dry;
+mod keydb;
 mod mongodb;
 mod postgres;
 mod redb;
@@ -88,6 +91,8 @@ pub(crate) enum Database {
 	Postgres,
 	#[cfg(feature = "redis")]
 	Redis,
+	#[cfg(feature = "keydb")]
+	Keydb,
 }
 
 impl Database {
@@ -116,6 +121,8 @@ impl Database {
 			Database::Postgres => postgres::POSTGRES_DOCKER_PARAMS,
 			#[cfg(feature = "redis")]
 			Database::Redis => redis::REDIS_DOCKER_PARAMS,
+			#[cfg(feature = "keydb")]
+			Database::Keydb => keydb::KEYDB_DOCKER_PARAMS,
 		};
 		let image = image.unwrap_or(params.image.to_string());
 		let container = DockerContainer::start(image, params.pre_args, params.post_args);
@@ -147,6 +154,8 @@ impl Database {
 			Database::Postgres => benchmark.run(PostgresClientProvider::default()).await,
 			#[cfg(feature = "redis")]
 			Database::Redis => benchmark.run(RedisClientProvider::default()).await,
+			#[cfg(feature = "keydb")]
+			Database::Keydb => benchmark.run(KeydbClientProvider::default()).await,
 		}
 	}
 }