TeamSearchApi.scala

package lila.teamSearch

import akka.stream.scaladsl._
import play.api.libs.json._

import lila.search._
import lila.team.{ Team, TeamRepo }

final class TeamSearchApi(
    client: ESClient,
    teamRepo: TeamRepo
)(implicit
    ec: scala.concurrent.ExecutionContext,
    mat: akka.stream.Materializer
) extends SearchReadApi[Team, Query] {

  def search(query: Query, from: From, size: Size) =
    client.search(query, from, size) flatMap { res =>
      teamRepo byOrderedIds res.ids
    }

  def count(query: Query) = client.count(query) dmap (_.count)

  def store(team: Team) = client.store(Id(team.id), toDoc(team))

  private def toDoc(team: Team) =
    Json.obj(
      Fields.name        -> team.name,
      Fields.description -> team.description.take(10000),
      Fields.nbMembers   -> team.nbMembers
    )

  def reset =
    client match {
      case c: ESClientHttp =>
        c.putMapping >> {

          logger.info(s"Index to ${c.index.name}")

          teamRepo.cursor
            .documentSource()
            .via(lila.common.LilaStream.logRate[Team]("team index")(logger))
            .map(t => Id(t.id) -> toDoc(t))
            .grouped(200)
            .mapAsync(1)(c.storeBulk)
            .toMat(Sink.ignore)(Keep.right)
            .run()
        } >> client.refresh
      case _ => funit
    }
}