com.soundcloud.twinagle.ProtoService.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of twinagle-runtime_2.13 Show documentation

An implementation of the Twirp protocol on top of Finagle

The newest version!

package com.soundcloud.twinagle

import com.twitter.finagle.Service
import com.twitter.finagle.http.{Request, Response}
import com.twitter.util.Future
import scalapb.{GeneratedMessage, GeneratedMessageCompanion}

case class ProtoService(rpcs: Seq[ProtoRpcBuilder]) {
  assert(rpcs.map(_.metadata.service).toSet.size == 1, "inconsistent services in metadata")
}

object ProtoService {
  implicit val asProtoService: AsProtoService[ProtoService] = (t: ProtoService) => t
}

case class ProtoRpc(metadata: EndpointMetadata, svc: Service[Request, Response])

object ProtoRpc {
  def apply[
      Req <: GeneratedMessage: GeneratedMessageCompanion,
      Resp <: GeneratedMessage: GeneratedMessageCompanion
  ](endpointMetadata: EndpointMetadata, rpc: Req => Future[Resp]): ProtoRpc = {
    ProtoRpcBuilder(endpointMetadata, rpc).build(MessageFilter.Identity)
  }
}

trait ProtoRpcBuilder {
  val metadata: EndpointMetadata

  def build(messageFilter: MessageFilter): ProtoRpc
}

object ProtoRpcBuilder {
  def apply[
      Req <: GeneratedMessage: GeneratedMessageCompanion,
      Resp <: GeneratedMessage: GeneratedMessageCompanion
  ](endpointMetadata: EndpointMetadata, rpc: Req => Future[Resp]): ProtoRpcBuilder = new ProtoRpcBuilder {
    override val metadata: EndpointMetadata = endpointMetadata

    override def build(messageFilter: MessageFilter): ProtoRpc = {
      val svc = new TwirpEndpointFilter[Req, Resp] andThen
        messageFilter.toFilter[Req, Resp] andThen
        Service.mk(rpc)
      ProtoRpc(endpointMetadata, svc)
    }
  }
}