com.sksamuel.cohort.redis.RedisHealthCheck.kt Maven / Gradle / Ivy

Go to download

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

Ktor Actuator

There is a newer version: 2.1.4

package com.sksamuel.cohort.redis

import com.sksamuel.cohort.HealthCheck
import com.sksamuel.cohort.HealthCheckResult
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.withContext
import redis.clients.jedis.DefaultJedisClientConfig
import redis.clients.jedis.HostAndPort
import redis.clients.jedis.Jedis

/**
 * A [HealthCheck] that checks that a connection can be made to a redis instance.
 */
class RedisHealthCheck(
  private val hostAndPort: HostPort,
  private val username: String?,
  private val password: String?,
  private val tls: Boolean,
) : HealthCheck {

  override suspend fun check(): HealthCheckResult {
    return withContext(Dispatchers.IO) {
      runCatching {
        val config = DefaultJedisClientConfig.builder().password(password).user(username).ssl(tls).build()
        val jedis = Jedis(HostAndPort(hostAndPort.host, hostAndPort.port), config)
        jedis.connection.use { it.ping() }
        HealthCheckResult.Healthy("Connected to redis cluster")
      }.getOrElse {
        HealthCheckResult.Unhealthy("Could not connect to redis at $hostAndPort", it)
      }
    }
  }
}