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

Go to download
package otoroshi.storage.stores

import akka.util.ByteString
import otoroshi.env.Env
import otoroshi.events.{AuditDataStore, AuditEvent}
import play.api.libs.json.{JsValue, Json}
import otoroshi.storage.RedisLike

import scala.concurrent.{ExecutionContext, Future}

class KvAuditDataStore(redisCli: RedisLike) extends AuditDataStore {

  override def count()(implicit ec: ExecutionContext, env: Env): Future[Long] =
    redisCli.llen(s"${env.storageRoot}:events:audit")

  override def findAllRaw(from: Long = 0, to: Long = 1000)(implicit
      ec: ExecutionContext,
      env: Env
  ): Future[Seq[ByteString]] =
    redisCli.lrange(s"${env.storageRoot}:events:audit", from, to)

  override def push(event: JsValue)(implicit ec: ExecutionContext, env: Env): Future[Long] =
    for {
      config <- env.datastores.globalConfigDataStore.singleton()
      n      <- redisCli.lpush(s"${env.storageRoot}:events:audit", Json.stringify(event))
      -      <- redisCli.ltrim(s"${env.storageRoot}:events:audit", 0, config.maxLogsSize)
    } yield n
}