a few additional client commands

profunktor · Aug 22, 2024 · 19c3f4a · 19c3f4a
1 parent 6aca4d1
commit 19c3f4a
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 0 deletions.
diff --git a/modules/effects/src/main/scala/dev/profunktor/redis4cats/algebra/connection.scala b/modules/effects/src/main/scala/dev/profunktor/redis4cats/algebra/connection.scala
@@ -32,4 +32,7 @@ trait Client[F[_], K] {
   def setClientName(name: K): F[Boolean]
   def getClientName(): F[Option[K]]
   def getClientId(): F[Long]
+  def getClientInfo: F[Map[String, String]]
+  def setLibName(name: String): F[Boolean]
+  def setLibVersion(version: String): F[Boolean]
 }
diff --git a/modules/effects/src/main/scala/dev/profunktor/redis4cats/redis.scala b/modules/effects/src/main/scala/dev/profunktor/redis4cats/redis.scala
@@ -1281,6 +1281,29 @@ private[redis4cats] class BaseRedis[F[_]: FutureLift: MonadThrow: Log, K, V](
   override def getClientId(): F[Long] =
     async.flatMap(_.clientId().futureLift.map(Long.unbox))
 
+  override def setLibName(name: String): F[Boolean] =
+    async.flatMap(_.clientSetinfo("LIB-NAME", name).futureLift.map(_ == "OK"))
+
+  override def setLibVersion(version: String): F[Boolean] =
+    async.flatMap(_.clientSetinfo("LIB-VER", version).futureLift.map(_ == "OK"))
+
+  override def getClientInfo: F[Map[String, String]] =
+    async.flatMap(
+      _.clientInfo().futureLift
+        .flatMap(parseClientInfo)
+    )
+
+  private def parseClientInfo(info: String): F[Map[String, String]] =
+    FutureLift[F].delay(
+      info
+        .replace("\n", "")
+        .split(" ")
+        .toList
+        .map(_.split("=", 2).toList)
+        .collect { case k :: v :: Nil => (k, v) }
+        .toMap
+    )
+
   /******************************* Server API **********************************/
   override val flushAll: F[Unit] =
     async.flatMap(_.flushall().futureLift.void)

diff --git a/modules/tests/src/test/scala/dev/profunktor/redis4cats/TestScenarios.scala b/modules/tests/src/test/scala/dev/profunktor/redis4cats/TestScenarios.scala
@@ -501,6 +501,15 @@ trait TestScenarios { self: FunSuite =>
       newClientName <- redis.getClientName()
       _ <- IO(assertEquals(newClientName, Some(clientName)))
       _ <- redis.getClientId()
+      info <- redis.getClientInfo
+      _ <- IO(assert(info.nonEmpty))
+      success <- redis.setLibName("redis4cats")
+      _ <- IO(assert(success))
+      success <- redis.setLibVersion("0.10.0")
+      _ <- IO(assert(success))
+      info <- redis.getClientInfo
+      _ <- IO(assert(info.get("lib-name").contains("redis4cats")))
+      _ <- IO(assert(info.get("lib-ver").contains("0.10.0")))
     } yield ()
   }