akka.persistence.r2dbc.query.ReactiveReadJournal.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of akka-persistence-r2dbc_2.13 Show documentation
Reactive Akka Journal, Snapshot Store, and Persistence Query plugin
The newest version!
/*
 * Copyright 2020-2021 Borislav Borisov.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package akka.persistence.r2dbc.query

import akka.NotUsed
import akka.actor.ExtendedActorSystem
import akka.persistence.query.scaladsl._
import akka.persistence.query.{EventEnvelope, NoOffset, Offset, Sequence}
import akka.persistence.r2dbc.journal.{PersistenceReprSerDe, JournalEntry}
import akka.serialization.SerializationExtension
import akka.stream.scaladsl.Source
import scala.concurrent.duration.{FiniteDuration, _}
import scala.concurrent.{ExecutionContextExecutor, Future}

private[akka] trait ReactiveReadJournal
    extends ReadJournal
        with CurrentPersistenceIdsQuery
        with PersistenceIdsQuery
        with CurrentEventsByPersistenceIdQuery
        with EventsByPersistenceIdQuery
        with CurrentEventsByTagQuery
        with EventsByTagQuery {

  protected val system: ExtendedActorSystem
  private lazy implicit val ec: ExecutionContextExecutor = system.dispatcher

  private lazy val reprSerDe = new PersistenceReprSerDe(SerializationExtension(system))
  protected val dao: QueryDao

  override def currentPersistenceIds(): Source[String, NotUsed] =
    Source.fromGraph(AllPersistentIdStage(dao))

  override def persistenceIds(): Source[String, NotUsed] =
    Source.fromGraph(AllPersistentIdStage(dao, Some(100.millis)))

  override def currentEventsByPersistenceId(
      persistenceId: String,
      fromSequenceNr: Long,
      toSequenceNr: Long
  ): Source[EventEnvelope, NotUsed] =
    eventsByPersistenceIdInternal(persistenceId, fromSequenceNr, toSequenceNr)

  override def eventsByPersistenceId(persistenceId: String,
      fromSequenceNr: Long,
      toSequenceNr: Long
  ): Source[EventEnvelope, NotUsed] =
    eventsByPersistenceIdInternal(persistenceId, fromSequenceNr, toSequenceNr, Some(100.millis))

  override def currentEventsByTag(tag: String, offset: Offset): Source[EventEnvelope, NotUsed] =
    eventsByTagInternal(tag, offset, None)

  override def eventsByTag(tag: String, offset: Offset): Source[EventEnvelope, NotUsed] =
    eventsByTagInternal(tag, offset, Some(100.millis))

  private[this] def eventsByPersistenceIdInternal(
      persistenceId: String,
      fromSequenceNr: Long,
      toSequenceNr: Long,
      refreshInterval: Option[FiniteDuration] = None
  ): Source[EventEnvelope, NotUsed] = mapEntries(
    Source.fromGraph(EventsByPersistenceIdStage(dao, persistenceId, fromSequenceNr, toSequenceNr, refreshInterval))
  )

  private[this] def eventsByTagInternal(
      tag: String,
      offset: Offset,
      refreshInterval: Option[FiniteDuration]
  ): Source[EventEnvelope, NotUsed] = mapEntries(
    offset match {
      case Sequence(value) => Source.fromGraph(EventsByTagStage(dao, tag, value, refreshInterval))
      case NoOffset => Source.fromGraph(EventsByTagStage(dao, tag, 0, refreshInterval))
      case _ => Source.failed(new IllegalArgumentException("Only Sequence is supported"))
    }
  )

  private[this] def mapEntries(source: Source[JournalEntry, NotUsed]): Source[EventEnvelope, NotUsed] = {
    source.mapAsync(1) { entry =>
      reprSerDe.deserialize(entry).flatMap(Future.fromTry).map((entry.id, _))
    } map { case (index, repr) =>
      EventEnvelope(Sequence(index), repr.persistenceId, repr.sequenceNr, repr.payload, repr.timestamp)
    }
  }

}