58 lines
1.9 KiB
Scala
58 lines
1.9 KiB
Scala
package lila.relay
|
|
|
|
import akka.stream.scaladsl._
|
|
import org.joda.time.format.DateTimeFormat
|
|
import scala.concurrent.duration._
|
|
import scala.concurrent.ExecutionContext
|
|
|
|
import lila.study.{ Chapter, ChapterRepo, PgnDump, Study, StudyRepo }
|
|
import lila.common.Bus
|
|
|
|
final class RelayPgnStream(
|
|
roundRepo: RelayRoundRepo,
|
|
studyRepo: StudyRepo,
|
|
studyChapterRepo: ChapterRepo,
|
|
studyPgnDump: PgnDump
|
|
)(implicit ec: ExecutionContext) {
|
|
|
|
def exportFullTour(tour: RelayTour): Source[String, _] =
|
|
Source futureSource {
|
|
roundRepo.idsByTourOrdered(tour) flatMap { ids =>
|
|
studyRepo.byOrderedIds(ids.map(_.value).map(Study.Id)) map { studies =>
|
|
Source(studies).flatMapConcat { studyPgnDump(_, flags) }
|
|
}
|
|
}
|
|
}
|
|
|
|
private val flags = PgnDump.WithFlags(comments = false, variations = false, clocks = true)
|
|
private val fileR = """[\s,]""".r
|
|
private val dateFormat = DateTimeFormat forPattern "yyyy.MM.dd"
|
|
|
|
def filename(tour: RelayTour): String = {
|
|
val date = dateFormat.print(tour.syncedAt | tour.createdAt)
|
|
fileR.replaceAllIn(s"lichess_broadcast_${tour.slug}_${tour.id}_$date", "")
|
|
}
|
|
|
|
def streamRoundGames(rt: RelayRound.WithTourAndStudy): Source[String, _] = {
|
|
if (rt.relay.hasStarted) studyPgnDump(rt.study, flags)
|
|
else Source.empty[String]
|
|
} concat Source
|
|
.queue[Set[Chapter.Id]](8, akka.stream.OverflowStrategy.dropHead)
|
|
.mapMaterializedValue { queue =>
|
|
val chan = SyncResult busChannel rt.relay.id
|
|
val sub = Bus.subscribeFun(chan) { case SyncResult.Ok(moves, _) =>
|
|
queue
|
|
.offer(moves.collect {
|
|
case (id, nb) if nb > 0 => id
|
|
}.toSet)
|
|
.unit
|
|
}
|
|
queue.watchCompletion().foreach { _ =>
|
|
Bus.unsubscribe(sub, chan)
|
|
}
|
|
}
|
|
.flatMapConcat(studyChapterRepo.byIdsSource)
|
|
.throttle(16, 1 second)
|
|
.mapAsync(1)(studyPgnDump.ofChapter(rt.study, flags))
|
|
}
|