storage.stores.KvHealthCheckDataStore.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of otoroshi_2.12 Show documentation

Lightweight api management on top of a modern http reverse proxy

The newest version!

package otoroshi.storage.stores

import otoroshi.env.Env
import otoroshi.events.{HealthCheckDataStore, HealthCheckEvent}
import otoroshi.models.ServiceDescriptor
import play.api.libs.json.{JsValue, Json}
import otoroshi.storage.RedisLike

import scala.concurrent.{ExecutionContext, Future}

class KvHealthCheckDataStore(redisCli: RedisLike, _env: Env) extends HealthCheckDataStore {

  val collectionSize = 30

  def key(name: String) = s"${_env.storageRoot}:deschealthcheck:$name"

  override def push(evt: JsValue)(implicit ec: ExecutionContext, env: Env): Future[Long] =
    for {
      n <- redisCli.lpush(key((evt \ "@serviceId").as[String]), Json.stringify(evt))
      _ <- redisCli.ltrim(key((evt \ "@serviceId").as[String]), 0, collectionSize)
    } yield n

  override def findAll(
      serviceDescriptor: ServiceDescriptor
  )(implicit ec: ExecutionContext, env: Env): Future[Seq[HealthCheckEvent]] =
    redisCli
      .lrange(key(serviceDescriptor.id), 0, collectionSize - 1)
      .map(seq => seq.map(i => Json.parse(i.utf8String).as(HealthCheckEvent.format)))

  override def findLast(
      serviceDescriptor: ServiceDescriptor
  )(implicit ec: ExecutionContext, env: Env): Future[Option[HealthCheckEvent]] =
    redisCli
      .lrange(key(serviceDescriptor.id), 0, 1)
      .map(seq => seq.map(i => Json.parse(i.utf8String).as(HealthCheckEvent.format)).headOption)
}