main.misk.redis.JedisExtensions.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of misk-redis Show documentation

Open source application container in Kotlin

There is a newer version: 2024.11.14.031606-bddeae6

package misk.redis

import mu.KLogger
import redis.clients.jedis.Jedis
import redis.clients.jedis.JedisCluster
import redis.clients.jedis.JedisPooled
import redis.clients.jedis.UnifiedJedis

internal fun UnifiedJedis.flushAllWithClusterSupport(logger: KLogger) {
  when (this) {
    is JedisPooled -> this.flushAll()
    is JedisCluster -> {
      // Note: flushAll cannot be broadcast to all nodes in a cluster. We need to flush each node individually.
      this.clusterNodes.forEach { (node, pool) ->
        pool.resource.use { conn ->
          try {
            Jedis(conn).use { jedis -> jedis.flushAll() }
          } catch (e : Exception) {
            logger.error(e) { "Error flushing node $node: + ${e.message}" }
          }
        }
      }
    }
    else -> error("flushAll is not supported for UnifiedJedis implementation ${this.javaClass}")
  }
}