com.convergencelabs.convergence.server.api.rest.domain.DomainStatsService.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2019 - Convergence Labs, Inc.
 *
 * This file is part of the Convergence Server, which is released under
 * the terms of the GNU General Public License version 3 (GPLv3). A copy
 * of the GPLv3 should have been provided along with this file, typically
 * located in the "LICENSE" file, which is part of this source code package.
 * Alternatively, see  for the
 * full text of the GPLv3 license, if it was not provided.
 */

package com.convergencelabs.convergence.server.api.rest.domain


import akka.actor.typed.scaladsl.AskPattern._
import akka.actor.typed.{ActorRef, Scheduler}
import akka.http.scaladsl.marshalling.ToResponseMarshallable.apply
import akka.http.scaladsl.server.Directive.addByNameNullaryApply
import akka.http.scaladsl.server.Directives.{_segmentStringToPathMatcher, complete, get, path}
import akka.http.scaladsl.server.Route
import akka.util.Timeout
import com.convergencelabs.convergence.server.api.rest._
import com.convergencelabs.convergence.server.backend.services.domain.rest.DomainRestActor
import com.convergencelabs.convergence.server.backend.services.domain.rest.DomainRestActor.DomainRestMessage
import com.convergencelabs.convergence.server.backend.services.domain.stats.DomainStatsActor._
import com.convergencelabs.convergence.server.model.DomainId
import com.convergencelabs.convergence.server.security.AuthorizationProfile

import scala.concurrent.{ExecutionContext, Future}

private[domain] final class DomainStatsService(domainRestActor: ActorRef[DomainRestActor.Message],
                                               scheduler: Scheduler,
                                               executionContext: ExecutionContext,
                                               timeout: Timeout)
  extends AbstractDomainRestService(scheduler, executionContext, timeout) {

  import DomainStatsService._

  def route(authProfile: AuthorizationProfile, domain: DomainId): Route =
    (path("stats") & get) {
      complete(getStats(domain))
    }

  private[this] def getStats(domain: DomainId): Future[RestResponse] = {
    domainRestActor
      .ask[GetStatsResponse](r => DomainRestMessage(domain, GetStatsRequest(r)))
      .map(_.stats.fold(
        {
          case UnknownError() =>
            InternalServerError
        },
        { stats =>
          val DomainStats(activeSessionCount, userCount, modelCount, dbSize) = stats
          val response = DomainStatsRestData(activeSessionCount, userCount, modelCount, dbSize)
          okResponse(response)
        }
      ))
  }
}

object DomainStatsService {

  case class DomainStatsRestData(activeSessionCount: Long, userCount: Long, modelCount: Long, dbSize: Long)

}