Merge pull request #10 from dimacurrentai/redis

Redis.

Author: dkorolev

step06_redis/code/Cargo.toml

@@ -0,0 +1,8 @@
+[package]
+name = "redis_example"
+edition = "2024"
+
+[dependencies]
+clap = { version = "4.0", features = ["derive"] }
+redis = { version = "0.25", features = ["tokio-comp"] }
+tokio = { version = "1", features = ["full"] }

step06_redis/code/src/main.rs

@@ -0,0 +1,59 @@
+use clap::Parser;
+use redis::AsyncCommands;
+use std::future::Future;
+use std::process::ExitCode;
+
+#[derive(Parser)]
+struct Args {
+  #[arg(long, default_value = "redis://127.0.0.1")]
+  redis: String,
+
+  #[arg(long)]
+  mode: String,
+}
+
+async fn run_check(args: &Args) -> Result<u8, redis::RedisError> {
+  let _ = redis::Client::open(args.redis.to_string())?.get_multiplexed_async_connection().await?;
+  println!("Redis is available.");
+  Ok(0)
+}
+
+async fn run_test(args: &Args) -> Result<u8, redis::RedisError> {
+  let client = redis::Client::open(args.redis.to_string())?;
+  let mut con = client.get_multiplexed_async_connection().await?;
+
+  let _: () = con.set("key", "value").await.expect("failed");
+  let val: String = con.get("key").await?;
+
+  println!("Got value: {}", val);
+
+  if val == "value" { Ok(0) } else { Ok(1) }
+}
+
+async fn run_arm<'a, F, RETVAL>(f: F, args: &'a Args) -> ExitCode
+where
+  F: FnOnce(&'a Args) -> RETVAL,
+  RETVAL: Future<Output = Result<u8, redis::RedisError>>,
+{
+  let result = f(&args).await;
+  ExitCode::from(match result {
+    Ok(code) => code,
+    _ => {
+      println!("Something failed, or Redis is not available.");
+      1
+    }
+  })
+}
+
+#[tokio::main]
+async fn main() -> ExitCode {
+  let args = Args::parse();
+  match args.mode.as_str() {
+    "check" => run_arm(&run_check, &args).await,
+    "test" => run_arm(&run_test, &args).await,
+    _ => {
+      println!("This `--mode` value is not supported.");
+      ExitCode::from(1)
+    }
+  }
+}

step06_redis/run.sh

@@ -0,0 +1,31 @@
+#!/bin/bash
+
+set -e
+
+docker pull redis
+
+docker build -f ../Dockerfile.template . -t demo
+
+docker stop redis-for-rust >/dev/null 2>&1 || true
+docker rm redis-for-rust >/dev/null 2>&1 || true
+
+docker run --rm --name redis-for-rust -p 6379:6379 -d redis
+
+echo 'Waiting for Redis ...'
+
+for i in $(seq 20); do
+  if docker run --add-host=host.docker.internal:host-gateway --rm -t --network bridge demo --mode check --redis redis://host.docker.internal ; then
+    echo 'Redis is up.'
+    break
+  else
+    echo 'Need to wait more.'
+    sleep 0.5
+  fi
+done
+
+echo 'Running the test.'
+docker run --add-host=host.docker.internal:host-gateway --rm -t demo --mode test --redis redis://host.docker.internal
+echo 'Test run successfully.'
+
+docker stop redis-for-rust
+echo 'Redis stopped.'